X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;f=client.py;h=610d1239a79ec115da6bbd5a60b4f5e00de14e92;hb=09958314bba592570a4d4b4dff44b28cee3257e6;hp=50da2402fc4a8b22c2c6b0625c6d79e8f0aaaca9;hpb=6127ae01609030b226eacc3f1aa72ee0a817e99d;p=NK2015_Client_Python_Alpha.git

diff --git a/client.py b/client.py
index 50da240..610d123 100755
--- a/client.py
+++ b/client.py
@@ -19,15 +19,26 @@ from pprint import pprint
 sys.path.append(client_config.rsa_path)
 import monRSA
 
-def full_read(sock):
-    """Lit sur la socket jusqu'à ce que l'output soit déJSON-izable"""
-    output=""
-    while True:
-        output+=sock.read()
-        try:
-            return json.loads(output)
-        except:
-            pass
+def full_read(socket):
+    """Lit un message complet sur la socket."""
+    # On récupère d'abord la taille du message
+    length_str = ''
+    char = socket.recv(1)
+    while char != '\n':
+        length_str += char
+        char = socket.recv(1)
+    total = int(length_str)
+    # On utilise une memoryview pour recevoir les données chunk par chunk efficacement
+    view = memoryview(bytearray(total))
+    next_offset = 0
+    while total - next_offset > 0:
+        recv_size = socket.recv_into(view[next_offset:], total - next_offset)
+        next_offset += recv_size
+    try:
+        msg = json.loads(view.tobytes())
+    except (TypeError, ValueError) as e:
+        raise NKNotJson("L'objet reçu n'est pas un JSON")
+    return msg
 
 def connect(sock,where,port=None):
     if port==None:
@@ -57,22 +68,22 @@ def checkidentity(a):
         raise KeyboardInterrupt("Imposteur !")
 
 def hello(socket,version):
-    socket.send('hello "%s"'%(version))
+    socket.send('["hello", "%s"]' % (version))
     return lit(a)
 
 def login(socket,user,mdp="",typ="bdd"):
-    if user in ["20-100","moibdd"]:
-        a.write('login ["20-100","plop","bdd",[["all"],["all"],true]]')
+    if user in ["20-100","moibdd","#1"]:
+        a.write('["login", ["#1","plop","bdd",[[],[],false]]]')
     elif user in ["vincent","moi"]:
-        a.write('login ["vincent","plop","special",["all"]]')
+        a.write('["login", ["vincent","plop","special",[]]]')
     elif user in ["toto"]:
-        a.write('login ["toto","plop","bdd",[["all"],["all"],true]]')
+        a.write('["login", ["toto","plop","bdd",[[],[],false]]]')
     else:
-        a.write('login ["%s","%s","%s",[["all"],["all"],true]]'%(user,mdp,typ))
+        a.write('["login", ["%s","%s","%s",[[],[],false]]]'%(user,mdp,typ))
     return lit(a)
     
 def search(socket,term):
-    a.write("search %s"%(term))
+    a.write('["search", %s]' % (term))
     return lit(a)
 
 def lit(socket):
@@ -84,9 +95,7 @@ def lit(socket):
           print obj["errmsg"]
           return obj["msg"]
     except ssl.SSLError:
-        return "Rien à lire"
-    except ValueError:
-        return "Serveur mort ?"
+        print "Rien à lire"
 
 if __name__=="__main__":
     where=raw_input("quel serveur ?")