client.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 # Définition des fonctions de base de communication du client
   5 # Pour l'instant ce script sert essentiellement à les charger en mémoire pour les utiliser à la main
   6
   7 import os, sys, string, time, random
   8 import socket, ssl
   9 import threading
  10
  11 import json
  12 import hashlib
  13
  14 import client_config
  15 reload(client_config)
  16
  17 from pprint import pprint
  18
  19 sys.path.append(client_config.rsa_path)
  20 import monRSA
  21
  22 def full_read(socket):
  23     """Lit un message complet sur la socket."""
  24     # On récupère d'abord la taille du message
  25     length_str = ''
  26     char = socket.recv(1)
  27     while char != '\n':
  28         length_str += char
  29         char = socket.recv(1)
  30     total = int(length_str)
  31     # On utilise une memoryview pour recevoir les données chunk par chunk efficacement
  32     view = memoryview(bytearray(total))
  33     next_offset = 0
  34     while total - next_offset > 0:
  35         recv_size = socket.recv_into(view[next_offset:], total - next_offset)
  36         next_offset += recv_size
  37     try:
  38         msg = json.loads(view.tobytes())
  39     except (TypeError, ValueError) as e:
  40         raise NKNotJson("L'objet reçu n'est pas un JSON")
  41     return msg
  42
  43 def connect(sock,where,port=None):
  44     if port==None:
  45         port=client_config.port
  46     try:
  47         ip={"ici":"localhost","vent":"vent.crans.org","eva":"eva.crans.org","cerebro":"cerebro.crans.org"}[where]
  48     except:
  49         ip=where
  50     sock.connect((ip,port))
  51     return ssl.wrap_socket(sock,ca_certs=client_config.ca_certfile)
  52 #              cert_reqs=ssl.CERT_NONE,ciphers="AES256-SHA")
  53
  54 def checkidentity(a):
  55     c=a.getpeercert()
  56     print c
  57     c=dict([i[0] for i in c["subject"]])
  58     awaited={'commonName': u'note.crans.org',
  59              'countryName': u'FR',
  60              'emailAddress': u'legallic@crans.org',
  61              'localityName': u'Cachan',
  62              'organizationName': u'BDE ENS Cachan',
  63              'organizationalUnitName': u'Kfet',
  64              'stateOrProvinceName': u'\xc3\x8ele de France'}
  65     if all([c[cle]==awaited[cle] for cle in awaited.keys()]):
  66         return
  67     else:
  68         raise KeyboardInterrupt("Imposteur !")
  69
  70 def hello(socket,version):
  71     socket.send('["hello", "%s"]' % (version))
  72     return lit(a)
  73
  74 def login(socket,user,mdp="",typ="bdd"):
  75     if user in ["20-100","moibdd","#1"]:
  76         a.write('["login", ["#1","plop","bdd",[[],[],false]]]')
  77     elif user in ["vincent","moi"]:
  78         a.write('["login", ["vincent","plop","special",[]]]')
  79     elif user in ["toto"]:
  80         a.write('["login", ["toto","plop","bdd",[[],[],false]]]')
  81     else:
  82         a.write('["login", ["%s","%s","%s",[[],[],false]]]'%(user,mdp,typ))
  83     return lit(a)
  84
  85 def search(socket,term):
  86     a.write('["search", %s]' % (term))
  87     return lit(a)
  88
  89 def lit(socket):
  90     try:
  91         obj=full_read(socket)
  92         if obj["retcode"]==0:
  93           return obj['msg']
  94         else:
  95           print obj["errmsg"]
  96           return obj["msg"]
  97     except ssl.SSLError:
  98         print "Rien à lire"
  99
 100 if __name__=="__main__":
 101     where=raw_input("quel serveur ?")
 102     #port=int(raw_input("port ? "))
 103     a=socket.socket()
 104     a=connect(a,where)
 105     a.settimeout(2)
 106     #checkidentity(a)
 107     print hello(a,"Python Client alpha")
 108     print login(a,"20-100","plop")
 109     print "Socket créée dans la variable \"a\""
 110