D4
/
d4-core
mirror of https://github.com/D4-project/d4-core
2
0
Fork 0
Code Issues Releases Wiki Activity

chg: [server] use redis stream

pull/23/head
Terrtia 2018-12-13 16:03:05 +01:00
parent 7a2d04cfe1
commit 48fc60edeb
No known key found for this signature in database
GPG Key ID: 1E1B1F50D84613D0
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
server/server.py
View File

@ -4,6 +4,7 @@ import os
import sys import sys
import hmac import hmac
import stat import stat
import redis
from twisted.internet import ssl, task, protocol, endpoints, defer from twisted.internet import ssl, task, protocol, endpoints, defer
from twisted.python import log from twisted.python import log
@ -15,11 +16,15 @@ from twisted.internet.protocol import Protocol
from ctypes import * from ctypes import *
from uuid import UUID from uuid import UUID
path = '/mypath'
hmac_reset = '0000000000000000000000000000000000000000000000000000000000000000' hmac_reset = '0000000000000000000000000000000000000000000000000000000000000000'
hmac_key = b'private key to change\n' hmac_key = b'private key to change\n'
redis_server = redis.StrictRedis(
host="localhost",
port=6379,
db=0,
decode_responses=True)
class Echo(Protocol): class Echo(Protocol):
#def __init__(self, factory): #def __init__(self, factory):
@ -27,6 +32,7 @@ class Echo(Protocol):
def dataReceived(self, data): def dataReceived(self, data):
process_header(data) process_header(data)
print(self.transport.client)
#print(data[72:]) #print(data[72:])
class D4Header(Structure): class D4Header(Structure):
@ -48,6 +54,7 @@ def process_header(data):
if len(data) > 73: if len(data) > 73:
d4_header = data[:72].hex() d4_header = data[:72].hex()
print(d4_header)
uuid_header = d4_header[4:36] uuid_header = d4_header[4:36]
hmac_header = d4_header[64:128] hmac_header = d4_header[64:128]
@ -58,6 +65,17 @@ def process_header(data):
data_header = unpack(D4Header, data) data_header = unpack(D4Header, data)
### Debug ###
print('version: {}'.format( data_header['version'] ))
print('type: {}'.format( data_header['type'] ))
print('uuid: {}'.format(uuid_header))
print('timestamp: {}'.format( data_header['timestamp'] ))
print('hmac: {}'.format( hmac_header ))
print('size: {}'.format( data_header['size'] ))
print('size: {}'.format( len(data) - data_header['struct_size']))
#print(d4_header)
### ###
# check if is valid uuid v4 # check if is valid uuid v4
if not is_valid_uuid_v4(uuid_header): if not is_valid_uuid_v4(uuid_header):
print('not uuid v4') print('not uuid v4')
@ -69,25 +87,16 @@ def process_header(data):
print('size: {}'.format(data_header['size'])) print('size: {}'.format(data_header['size']))
print(len(data) - data_header['struct_size']) # sizeof(d)=72 print(len(data) - data_header['struct_size']) # sizeof(d)=72
else: else:
### Debug ###
print('version: {}'.format( data_header['version'] ))
print('type: {}'.format( data_header['type'] ))
print('uuid: {}'.format(uuid_header))
print('timestamp: {}'.format( data_header['timestamp'] ))
print('hmac: {}'.format( hmac_header ))
print('size: {}'.format( data_header['size'] ))
#print(d4_header)
### ###
# verify hmac sha256 # verify hmac sha256
HMAC = hmac.new(hmac_key, msg=data, digestmod='sha256') HMAC = hmac.new(hmac_key, msg=data, digestmod='sha256')
print('hexdigest: {}'.format( HMAC.hexdigest() ))
print(data)
if hmac_header == HMAC.hexdigest(): if hmac_header == HMAC.hexdigest():
print('hmac match') print('hmac match')
path_file = os.path.join(path,uuid_header)
if not os.path.isdir(path_file):
os.mkdir(path_file) redis_server.xadd('stream:{}'.format(data_header['type']), {'message': data[72:], 'uuid':uuid_header, 'timestamp': data_header['timestamp'], 'version': data_header['version']})
with open(os.path.join(path_file, 'out'), 'ab')as f:
f.write(data[72:])
print('END') print('END')
print() print()