Changeset 19 in chevmsgr

Timestamp:

11/23/15 23:39:17 (8 years ago)

Author:

cheese

Message:

init -> setSocket / setKey 로 나눔
msg.hpp 에서 SecureSocket::close 함수 수정, 기본 생성자 수정

Location:

trunk

Files:

• msg.hpp (modified) (2 diffs)
• msgsrv.cpp (modified) (1 diff)

trunk/msg.hpp

@@ -178 +178 @@
 
 public :
-    SecureSocket() {}
+    SecureSocket()
+    {
+        sock = NULL;
+    }
     SecureSocket(cf::network::tcp * sock, const std::string & ip, const std::string & sms)
     {
-        init(sock, ip, sms);
-    }
-
-    void init(cf::network::tcp * sock, const std::string & ip, const std::string & sms)
+        setSocket(sock);
+        setKey(ip, sms);
+    }
+
+    void setSocket(cf::network::tcp * sock)
     {
         this->sock = sock;
-
+    }
+
+    void setKey(const std::string & ip, const std::string & sms)
+    {
         cf::bin seed = sms + DELIMITER + ip;
         cf::bin key = crypto().sha256(seed);
@@ -206 +213 @@
     void close()
     {
-        sock->close();
+        if (sock)
+            sock->close();
     }
 };

trunk/msgsrv.cpp

@@ -590 +590 @@
 int main(int argc, char ** argv)
 {
-    if (argc != 2)
+    if (argc != 3)
     {
         std::cerr << "-_-^" << std::endl;