about summary refs log tree commit diff stats
diff options
context:
space:
mode:
-rw-r--r--src/event/client_events.c36
-rw-r--r--src/event/server_events.c41
-rw-r--r--src/otr/otr.c81
-rw-r--r--src/otr/otr.h4
-rw-r--r--tests/otr/stub_otr.c4
5 files changed, 92 insertions, 74 deletions
diff --git a/src/event/client_events.c b/src/event/client_events.c
index 2a1d349a..2aa66d08 100644
--- a/src/event/client_events.c
+++ b/src/event/client_events.c
@@ -62,44 +62,14 @@ client_connect_account(ProfAccount *account)
 void
 client_send_msg(const char * const barejid, const char * const msg)
 {
-    char *id = NULL;
-
 #ifdef HAVE_LIBOTR
-    prof_otrpolicy_t policy = otr_get_policy(barejid);
-
-    if (otr_is_secure(barejid)) {
-        char *encrypted = otr_encrypt_message(barejid, msg);
-        if (encrypted != NULL) {
-            id = message_send_chat_encrypted(barejid, encrypted);
-            chat_log_otr_msg_out(barejid, msg);
-            ui_outgoing_chat_msg(barejid, msg, id);
-            otr_free_message(encrypted);
-        } else {
-            cons_show_error("Failed to encrypt and send message.");
-        }
-
-    } else if (policy == PROF_OTRPOLICY_ALWAYS) {
-        cons_show_error("Failed to send message. Please check OTR policy");
-
-    } else if (policy == PROF_OTRPOLICY_OPPORTUNISTIC) {
-        char *otr_tagged_msg = otr_tag_message(msg);
-        id = message_send_chat_encrypted(barejid, otr_tagged_msg);
-        ui_outgoing_chat_msg(barejid, msg, id);
-        chat_log_msg_out(barejid, msg);
-        free(otr_tagged_msg);
-
-    } else {
-        id = message_send_chat(barejid, msg);
-        ui_outgoing_chat_msg(barejid, msg, id);
-        chat_log_msg_out(barejid, msg);
-    }
+    otr_on_message_send(barejid, msg);
 #else
-    id = message_send_chat(barejid, msg);
+    char *id = message_send_chat(barejid, msg);
     chat_log_msg_out(barejid, msg);
     ui_outgoing_chat_msg(barejid, msg, id);
-#endif
-
     free(id);
+#endif
 }
 
 void
diff --git a/src/event/server_events.c b/src/event/server_events.c
index bd4634e5..514835ba 100644
--- a/src/event/server_events.c
+++ b/src/event/server_events.c
@@ -46,7 +46,6 @@
 
 #ifdef HAVE_LIBOTR
 #include "otr/otr.h"
-#include <libotr/proto.h>
 #endif
 
 #include "ui/ui.h"
@@ -321,45 +320,7 @@ void
 srv_incoming_message(char *barejid, char *resource, char *message)
 {
 #ifdef HAVE_LIBOTR
-    gboolean was_decrypted = FALSE;
-    char *newmessage;
-
-    prof_otrpolicy_t policy = otr_get_policy(barejid);
-    char *whitespace_base = strstr(message,OTRL_MESSAGE_TAG_BASE);
-
-    //check for OTR whitespace (opportunistic or always)
-    if (policy == PROF_OTRPOLICY_OPPORTUNISTIC || policy == PROF_OTRPOLICY_ALWAYS) {
-        if (whitespace_base) {
-            if (strstr(message, OTRL_MESSAGE_TAG_V2) || strstr(message, OTRL_MESSAGE_TAG_V1)) {
-                // Remove whitespace pattern for proper display in UI
-                // Handle both BASE+TAGV1/2(16+8) and BASE+TAGV1+TAGV2(16+8+8)
-                int tag_length	=	24;
-                if (strstr(message, OTRL_MESSAGE_TAG_V2) && strstr(message, OTRL_MESSAGE_TAG_V1)) {
-                    tag_length = 32;
-                }
-                memmove(whitespace_base, whitespace_base+tag_length, tag_length);
-                char *otr_query_message = otr_start_query();
-                cons_show("OTR Whitespace pattern detected. Attempting to start OTR session...");
-                message_send_chat_encrypted(barejid, otr_query_message);
-            }
-        }
-    }
-    newmessage = otr_decrypt_message(barejid, message, &was_decrypted);
-
-    // internal OTR message
-    if (newmessage == NULL) {
-        return;
-    }
-
-    if (policy == PROF_OTRPOLICY_ALWAYS && !was_decrypted && !whitespace_base) {
-        char *otr_query_message = otr_start_query();
-        cons_show("Attempting to start OTR session...");
-        message_send_chat_encrypted(barejid, otr_query_message);
-    }
-
-    ui_incoming_msg(barejid, resource, newmessage, NULL);
-    chat_log_otr_msg_in(barejid, newmessage, was_decrypted);
-    otr_free_message(newmessage);
+    otr_on_message_recv(barejid, resource, message);
 #else
     ui_incoming_msg(barejid, resource, message, NULL);
     chat_log_msg_in(barejid, message);
diff --git a/src/otr/otr.c b/src/otr/otr.c
index 9dcff1f9..c8518c70 100644
--- a/src/otr/otr.c
+++ b/src/otr/otr.c
@@ -270,6 +270,87 @@ otr_on_connect(ProfAccount *account)
 }
 
 void
+otr_on_message_recv(const char * const barejid, const char * const resource, const char * const message)
+{
+    gboolean was_decrypted = FALSE;
+    char *decrypted;
+
+    prof_otrpolicy_t policy = otr_get_policy(barejid);
+    char *whitespace_base = strstr(message, OTRL_MESSAGE_TAG_BASE);
+
+    //check for OTR whitespace (opportunistic or always)
+    if (policy == PROF_OTRPOLICY_OPPORTUNISTIC || policy == PROF_OTRPOLICY_ALWAYS) {
+        if (whitespace_base) {
+            if (strstr(message, OTRL_MESSAGE_TAG_V2) || strstr(message, OTRL_MESSAGE_TAG_V1)) {
+                // Remove whitespace pattern for proper display in UI
+                // Handle both BASE+TAGV1/2(16+8) and BASE+TAGV1+TAGV2(16+8+8)
+                int tag_length = 24;
+                if (strstr(message, OTRL_MESSAGE_TAG_V2) && strstr(message, OTRL_MESSAGE_TAG_V1)) {
+                    tag_length = 32;
+                }
+                memmove(whitespace_base, whitespace_base+tag_length, tag_length);
+                char *otr_query_message = otr_start_query();
+                cons_show("OTR Whitespace pattern detected. Attempting to start OTR session...");
+                message_send_chat_encrypted(barejid, otr_query_message);
+            }
+        }
+    }
+    decrypted = otr_decrypt_message(barejid, message, &was_decrypted);
+
+    // internal OTR message
+    if (decrypted == NULL) {
+        return;
+    }
+
+    if (policy == PROF_OTRPOLICY_ALWAYS && !was_decrypted && !whitespace_base) {
+        char *otr_query_message = otr_start_query();
+        cons_show("Attempting to start OTR session...");
+        message_send_chat_encrypted(barejid, otr_query_message);
+    }
+
+    ui_incoming_msg(barejid, resource, decrypted, NULL);
+    chat_log_otr_msg_in(barejid, decrypted, was_decrypted);
+    otr_free_message(decrypted);
+}
+
+void
+otr_on_message_send(const char * const barejid, const char * const message)
+{
+    char *id = NULL;
+
+    prof_otrpolicy_t policy = otr_get_policy(barejid);
+
+    if (otr_is_secure(barejid)) {
+        char *encrypted = otr_encrypt_message(barejid, message);
+        if (encrypted != NULL) {
+            id = message_send_chat_encrypted(barejid, encrypted);
+            chat_log_otr_msg_out(barejid, message);
+            ui_outgoing_chat_msg(barejid, message, id);
+            otr_free_message(encrypted);
+        } else {
+            cons_show_error("Failed to encrypt and send message.");
+        }
+
+    } else if (policy == PROF_OTRPOLICY_ALWAYS) {
+        cons_show_error("Failed to send message. Please check OTR policy");
+
+    } else if (policy == PROF_OTRPOLICY_OPPORTUNISTIC) {
+        char *otr_tagged_msg = otr_tag_message(message);
+        id = message_send_chat_encrypted(barejid, otr_tagged_msg);
+        ui_outgoing_chat_msg(barejid, message, id);
+        chat_log_msg_out(barejid, message);
+        free(otr_tagged_msg);
+
+    } else {
+        id = message_send_chat(barejid, message);
+        ui_outgoing_chat_msg(barejid, message, id);
+        chat_log_msg_out(barejid, message);
+    }
+
+    free(id);
+}
+
+void
 otr_keygen(ProfAccount *account)
 {
     if (data_loaded) {
diff --git a/src/otr/otr.h b/src/otr/otr.h
index 91445a5c..8e1d22df 100644
--- a/src/otr/otr.h
+++ b/src/otr/otr.h
@@ -56,6 +56,10 @@ char* otr_libotr_version(void);
 char* otr_start_query(void);
 void otr_poll(void);
 void otr_on_connect(ProfAccount *account);
+
+void otr_on_message_recv(const char * const barejid, const char * const resource, const char * const message);
+void otr_on_message_send(const char * const barejid, const char * const message);
+
 void otr_keygen(ProfAccount *account);
 
 char* otr_tag_message(const char * const msg);
diff --git a/tests/otr/stub_otr.c b/tests/otr/stub_otr.c
index 7a0122b6..eb676877 100644
--- a/tests/otr/stub_otr.c
+++ b/tests/otr/stub_otr.c
@@ -41,6 +41,8 @@ char* otr_start_query(void)
 
 void otr_poll(void) {}
 void otr_on_connect(ProfAccount *account) {}
+void otr_on_message_recv(const char * const barejid, const char * const resource, const char * const message) {}
+void otr_on_message_send(const char * const barejid, const char * const message) {}
 
 void otr_keygen(ProfAccount *account)
 {
@@ -53,7 +55,7 @@ gboolean otr_key_loaded(void)
 }
 
 char* otr_tag_message(const char * const msg)
-{ 
+{
     return NULL;
 }