1 files changed, 10 insertions, 5 deletions
diff --git a/src/core/basetypes/MCConnectionLogger.h b/src/core/basetypes/MCConnectionLogger.h
index 8083d12d..dd596857 100644
--- a/src/core/basetypes/MCConnectionLogger.h
+++ b/src/core/basetypes/MCConnectionLogger.h
@@ -9,6 +9,8 @@
 #ifndef __MAILCORE_CONNECTION_LOGGER_H_
 #define __MAILCORE_CONNECTION_LOGGER_H_
 
+#include <stdlib.h>
+
 #ifdef __cplusplus
 
 namespace mailcore {
@@ -17,20 +19,23 @@ namespace mailcore {
     class String;
     
     enum ConnectionLogType {
-        ConnectionLogTypeGeneric,
+        // Received data
         ConnectionLogTypeReceived,
+        // Sent data
         ConnectionLogTypeSent,
+        // Sent private data
         ConnectionLogTypeSentPrivate,
-        ConnectionLogTypeErrorGeneric,
+        // Parse error
+        ConnectionLogTypeErrorParse,
+        // Error while receiving data - log() is called with a NULL buffer.
         ConnectionLogTypeErrorReceived,
+        // Error while sending data - log() is called with a NULL buffer.
         ConnectionLogTypeErrorSent,
-        ConnectionLogTypeErrorSentPrivate,
     };
     
     class ConnectionLogger {
     public:
-        virtual void logBuffer(ConnectionLogType logType, Data * buffer) {}
-        virtual void logString(ConnectionLogType logType, String * log) {}
+        virtual void log(void * sender, ConnectionLogType logType, Data * buffer) {}
     };
     
 }