def handle(self):
def md5_file(filename):
if os.path.isfile(filename):
m = md5()
with open(filename, 'rb') as f:
m.update(f.read())
return m.hexdigest()
else:
return None
def receive_data(sock):
return_data =''
while True:
data = sock.recv(4096)
#print '\033[36;1m%s\033[0m' %data
if data == 'EndOfDataConfirmationMark':
print 'file transfer done==='
break
return_data += data
#print '\033[34;1m=====\033[0m',return_data
return return_data
# self.request is the TCP socket connected to the client
def receive_data_by_size(sock,size):
return_data = ''
filename = time.time()
fp = open('recv/'+str(filename)+'.txt','w+')
restsize = size
print "recving..."
while 1:
if(restsize > 8096):
data = sock.recv(8096)
return_data += data
else:
data = sock.recv(restsize)
return_data += data
if restsize ==0:
break
fp.write(data)
restsize = restsize - len(data)
print restsize
#if restsize <= 0:
# break
fp.close()
print "receving is done...............",restsize
return return_data
def RSA_verifyication(sock):
raw_rsa_data = sock.recv(396) #fixed key length
#print '-------->',raw_rsa_data, len(raw_rsa_data)
try:
RSA_signal, encrypted_data,random_num_from_client = json.loads(raw_rsa_data)
except ValueError:
print '\033[41;1m.Wrong data format from %s,close connection!\033[0m' % self.client_address[0]
return 0
if RSA_signal == 'RSA_KEY_Virification':
#encrypted_data = "iwTgqSzMcNOHauWdXXc+rgfbWt6IUXmdIXUqNUJ2U7FZKISc2WR2yAJrq7ldR3TxQEppWgIo/Ycj\nA5gl0fGDVvAEvV02CKZ3gZEI6fWpiMoy6ucpFFDyVAWUrpiXdUOVKxOsDXGgeOObgvd1jsEQCo4i\ncLCBTWDn0HfyQic+Btm1txXc7Nw9jknUCZx6Y8I+6JaIYjNRLwJ6kSMwpTsfP37lvrQfdUkWu3bX\npV9z3hHOQ6+A8rlK7fmL1zk75TXDCmnrLY88UIv6BL4zPXtim4BCD7PlOvDG296br0VIcvF5uhqr\ntj7zOcbA81P1JBFm1nMJqLv+SB5sit923v05XA==\n"
try:
decrpted_data = key_gen.RSA_key.decrypt_RSA(key_gen.private_file,encrypted_data)
#print decrpted_data,'+++++++', random_num_from_client
except ValueError:
decrpted_data = 'Incorrect decryption'
if decrpted_data != random_num_from_client:
return 0
else:
return 0
if RSA_verifyication(self.request) == 0: #didn't pass the RSA virification
#if self.client_address[0] != server_address:
err_msg= "\033[31;1mIP %s didn't pass the RSA virification,drop it!\033[0m" % self.client_address[0]
pickle_data = '', err_msg
self.request.sendall( pickle.dumps(pickle_data) )
print err_msg
block_list.append(self.client_address[0])
if block_list.count(self.client_address[0]) > 5:
print "Alert::malicious attack from:" , self.client_address[0]
else:
print "RSA virification passed!"
self.request.sendall('RSA_OK')
#开始接收服务端发来的通信消息
self.data_type = self.request.recv(1024).strip()
if self.data_type.startswith( 'MonitorDataChange'):
#向proxy服务端发送开始准备接受数据
self.request.send('ReadyToReceiveData')
#接收改变的监控数据本地保存
data_size=self.data_type.split("|")[1]
#data_size=self.request.recv(1024)
if data_size<=8096:
monitor_data_new=self.request.recv(8096)
else:
monitor_data_new=receive_data_by_size(self.request,int(data_size))
monitor_dic_new=json.loads(monitor_data_new)
print monitor_dic_new
#filename='../recv/monitor_data_change.json'
filename='../recv/monitor_date_frist.json'
with open(filename, 'w+') as f:
f.write(monitor_data_new)
print 'write data into file....'
#新的监控数据如何放到监控中
elif self.data_type.startswith('RUN_Script|'):
recv_data= self.data
filename = "%s%s" %(recv_dir,recv_data.split('|')[1].strip())
print '+++++|receiving file from server:',filename
md5_key_from_client = self.data.split('|')[2]
file_content = receive_data(self.request)
print 'write data into file....'
with open(filename, 'wb') as f:
f.write(file_content)
md5_key= md5_file(filename)
print '+++++|verfiying md5key---'
if md5_key == md5_key_from_client:
print 'file transfer verification ok'
self.request.send('FileTransferComplete')
os.system('chmod +rx %s ' % filename)
#os.chmod(filename,stat.S_IREAD+stat.S_IWOTH+stat.S_IXUSR+stat.S_IRWXO)
cmd = "nohup %s > %s.log &" % (filename, filename)
print '\033[32;1mgoing to run script:\033[0m',cmd
result = os.system(cmd )
#print result,'||||+|||'
else:
print "file not complete"
self.request.send('FileTransferNotComplete')
elif self.data_type.startswith('SEND_FILE|'):
file_info = self.data.split('|')
md5_key_from_client = file_info[1]
compress_mark = file_info[4]
print compress_mark
source_file = file_info[2].split('/')[-1]
if file_info[3].endswith('/'):
dst_filename = '%s%s' %(file_info[3], source_file)
else:
dst_filename = '%s/%s' %(file_info[3], source_file)
print dst_filename
print '+++++|receiving file from server:',dst_filename
try:
with open(dst_filename, 'wb') as f:
self.request.send("FileOk2Send")
file_content = receive_data(self.request)
print 'write data into file....'
f.write(file_content)
except IOError:
print dst_filename,'not exist'
self.request.send("FileTransferError:cannot save file to %s" % dst_filename )
return "FileTransferError..."
md5_key= md5_file(dst_filename )
print '+++++|verfiying md5key---'
if md5_key == md5_key_from_client:
print 'file transfer verification ok'
self.request.send('FileTransferComplete')
if compress_mark == 'tarAndCompressed':
temp_tar_file = 'temp/%s.tar' % source_file
compress.decompress(dst_filename, temp_tar_file)
compress.untar(temp_tar_file,file_info[3] )
os.remove(temp_tar_file)
os.remove(dst_filename)
elif compress_mark == 'compressed':
target_path = dst_filename.split('/%s' %source_file)[0]
print target_path
target_file = '%s/%s' %(target_path, source_file[:-2])
compress.decompress(dst_filename, target_file)
os.remove(dst_filename)
else:
print "file not complete"
self.request.send('FileTransferNotComplete')
def handle(self):
def md5_file(filename):
if os.path.isfile(filename):
m = md5()
with open(filename, 'rb') as f:
m.update(f.read())
return m.hexdigest()
else:
return None
def receive_data(sock):
return_data =''
while True:
data = sock.recv(4096)
#print '\033[36;1m%s\033[0m' %data
if data == 'EndOfDataConfirmationMark':
print 'file transfer done==='
break
return_data += data
#print '\033[34;1m=====\033[0m',return_data
return return_data
# self.request is the TCP socket connected to the client
print "Got a connection from",self.client_address[0]
RSA_signal =self.request.recv(1024).strip()
random_num,dcrypted_data_from_server = str(random.random()),None
#print random_num, dcrypted_data_from_server
if RSA_signal == 'RSA_KEY_Virification':
encrypted_data = key_gen.RSA_key.encrypt_RSA(key_gen.public_file,random_num)
print encrypted_data
self.request.send(encrypted_data)
dcrypted_data_from_server = self.request.recv(1024)
if random_num != dcrypted_data_from_server:
print random_num, dcrypted_data_from_server
self.request.send('WrongRSAkey')
self.finish()
"""if self.client_address[0] != server_address:
err_msg= "\033[31;1mIP %s not allowed to connect this server!\033[0m" % self.client_address[0]
pickle_data = '', err_msg
self.request.sendall( pickle.dumps(pickle_data) )
print err_msg
block_list.append(self.client_address[0])
if block_list.count(self.client_address[0]) > 5:
print "Alert::malicious attack from:" , self.client_address[0]
"""
else:
self.request.send('CorrectRSAkey')
print "RSA virification passed!"
self.data = self.request.recv(1024).strip()
if self.data.startswith('CMD_Excution|'):
cmd= self.data.split('CMD_Excution|')[1]
cmd_status,result = commands.getstatusoutput(cmd)
print 'host:%s \tcmd:%s \tresult:%s' %(self.client_address[0], cmd, cmd_status)
self.request.sendall(pickle.dumps( (cmd_status,result) ))
if self.data.startswith('RUN_Script|'):
recv_data= self.data
filename = "%s%s" %(recv_dir,recv_data.split('|')[1].strip())
print '+++++|receiving file from server:',filename
md5_key_from_client = self.data.split('|')[2]
file_content = receive_data(self.request)
print 'write data into file....'
with open(filename, 'wb') as f:
f.write(file_content)
md5_key= md5_file(filename )
print '+++++|verfiying md5key---'
if md5_key == md5_key_from_client:
print 'file transfer verification ok'
self.request.send('FileTransferComplete')
os.system('chmod +rx %s ' % filename)
#os.chmod(filename,stat.S_IREAD+stat.S_IWOTH+stat.S_IXUSR+stat.S_IRWXO)
cmd = "nohup %s > %s.log &" % (filename, filename)
print '\033[32;1mgoing to run script:\033[0m',cmd
result = os.system(cmd )
#print result,'||||+|||'
else:
print "file not complete"
self.request.send('FileTransferNotComplete')
if self.data.startswith('SEND_FILE|'):
file_info = self.data.split('|')
md5_key_from_client = file_info[1]
compress_mark = file_info[4]
print compress_mark
source_file = file_info[2].split('/')[-1]
if file_info[3].endswith('/'):
dst_filename = '%s%s' %(file_info[3], source_file)
else:
dst_filename = '%s/%s' %(file_info[3], source_file)
print dst_filename
print '+++++|receiving file from server:',dst_filename
try:
with open(dst_filename, 'wb') as f:
self.request.send("FileOk2Send")
file_content = receive_data(self.request)
print 'write data into file....'
f.write(file_content)
except IOError:
print dst_filename,'not exist'
self.request.send("FileTransferError:cannot save file to %s" % dst_filename )
return "FileTransferError..."
md5_key= md5_file(dst_filename )
print '+++++|verfiying md5key---'
if md5_key == md5_key_from_client:
print 'file transfer verification ok'
self.request.send('FileTransferComplete')
if compress_mark == 'tarAndCompressed':
temp_tar_file = 'temp/%s.tar' % source_file
compress.decompress(dst_filename, temp_tar_file)
compress.untar(temp_tar_file,file_info[3] )
os.remove(temp_tar_file)
os.remove(dst_filename)
elif compress_mark == 'compressed':
target_path = dst_filename.split('/%s' %source_file)[0]
print target_path
target_file = '%s/%s' %(target_path, source_file[:-2])
compress.decompress(dst_filename, target_file)
os.remove(dst_filename)
else:
print "file not complete"
self.request.send('FileTransferNotComplete')
if self.data == 'getHardwareInfo':
import Hardware_Collect_Script
hardware_data = Hardware_Collect_Script.collectAsset()
self.request.sendall(hardware_data )
def handle(self):
def md5_file(filename):
if os.path.isfile(filename):
m = md5()
with open(filename, 'rb') as f:
m.update(f.read())
return m.hexdigest()
else:
return None
def receive_data(sock):
return_data = ''
while True:
data = sock.recv(4096)
#print '\033[36;1m%s\033[0m' %data
if data == 'EndOfDataConfirmationMark':
print 'file transfer done==='
break
return_data += data
#print '\033[34;1m=====\033[0m',return_data
return return_data
# self.request is the TCP socket connected to the client
def receive_data_by_size(sock, size):
return_data = ''
filename = time.time()
fp = open('recv/' + str(filename) + '.txt', 'w+')
restsize = size
print "recving..."
while 1:
if (restsize > 8096):
data = sock.recv(8096)
return_data += data
else:
data = sock.recv(restsize)
return_data += data
if restsize == 0:
break
fp.write(data)
restsize = restsize - len(data)
print restsize
#if restsize <= 0:
# break
fp.close()
print "receving is done...............", restsize
return return_data
def RSA_verifyication(sock):
raw_rsa_data = sock.recv(396) #fixed key length
#print '-------->',raw_rsa_data, len(raw_rsa_data)
try:
RSA_signal, encrypted_data, random_num_from_client = json.loads(
raw_rsa_data)
except ValueError:
print '\033[41;1m.Wrong data format from %s,close connection!\033[0m' % self.client_address[
0]
return 0
if RSA_signal == 'RSA_KEY_Virification':
#encrypted_data = "iwTgqSzMcNOHauWdXXc+rgfbWt6IUXmdIXUqNUJ2U7FZKISc2WR2yAJrq7ldR3TxQEppWgIo/Ycj\nA5gl0fGDVvAEvV02CKZ3gZEI6fWpiMoy6ucpFFDyVAWUrpiXdUOVKxOsDXGgeOObgvd1jsEQCo4i\ncLCBTWDn0HfyQic+Btm1txXc7Nw9jknUCZx6Y8I+6JaIYjNRLwJ6kSMwpTsfP37lvrQfdUkWu3bX\npV9z3hHOQ6+A8rlK7fmL1zk75TXDCmnrLY88UIv6BL4zPXtim4BCD7PlOvDG296br0VIcvF5uhqr\ntj7zOcbA81P1JBFm1nMJqLv+SB5sit923v05XA==\n"
try:
decrpted_data = key_gen.RSA_key.decrypt_RSA(
key_gen.private_file, encrypted_data)
#print decrpted_data,'+++++++', random_num_from_client
except ValueError:
decrpted_data = 'Incorrect decryption'
if decrpted_data != random_num_from_client:
return 0
else:
return 0
if RSA_verifyication(
self.request) == 0: #didn't pass the RSA virification
#if self.client_address[0] != server_address:
err_msg = "\033[31;1mIP %s didn't pass the RSA virification,drop it!\033[0m" % self.client_address[
0]
pickle_data = '', err_msg
self.request.sendall(pickle.dumps(pickle_data))
print err_msg
block_list.append(self.client_address[0])
if block_list.count(self.client_address[0]) > 5:
print "Alert::malicious attack from:", self.client_address[0]
else:
print "RSA virification passed!"
self.request.sendall('RSA_OK')
#开始接收服务端发来的通信消息
self.data_type = self.request.recv(1024).strip()
if self.data_type.startswith('MonitorDataChange'):
#向proxy服务端发送开始准备接受数据
self.request.send('ReadyToReceiveData')
#接收改变的监控数据本地保存
data_size = self.data_type.split("|")[1]
#data_size=self.request.recv(1024)
if data_size <= 8096:
monitor_data_new = self.request.recv(8096)
else:
monitor_data_new = receive_data_by_size(
self.request, int(data_size))
monitor_dic_new = json.loads(monitor_data_new)
print monitor_dic_new
#filename='../recv/monitor_data_change.json'
filename = '../recv/monitor_date_frist.json'
with open(filename, 'w+') as f:
f.write(monitor_data_new)
print 'write data into file....'
#新的监控数据如何放到监控中
elif self.data_type.startswith('RUN_Script|'):
recv_data = self.data
filename = "%s%s" % (recv_dir, recv_data.split('|')[1].strip())
print '+++++|receiving file from server:', filename
md5_key_from_client = self.data.split('|')[2]
file_content = receive_data(self.request)
print 'write data into file....'
with open(filename, 'wb') as f:
f.write(file_content)
md5_key = md5_file(filename)
print '+++++|verfiying md5key---'
if md5_key == md5_key_from_client:
print 'file transfer verification ok'
self.request.send('FileTransferComplete')
os.system('chmod +rx %s ' % filename)
#os.chmod(filename,stat.S_IREAD+stat.S_IWOTH+stat.S_IXUSR+stat.S_IRWXO)
cmd = "nohup %s > %s.log &" % (filename, filename)
print '\033[32;1mgoing to run script:\033[0m', cmd
result = os.system(cmd)
#print result,'||||+|||'
else:
print "file not complete"
self.request.send('FileTransferNotComplete')
elif self.data_type.startswith('SEND_FILE|'):
file_info = self.data.split('|')
md5_key_from_client = file_info[1]
compress_mark = file_info[4]
print compress_mark
source_file = file_info[2].split('/')[-1]
if file_info[3].endswith('/'):
dst_filename = '%s%s' % (file_info[3], source_file)
else:
dst_filename = '%s/%s' % (file_info[3], source_file)
print dst_filename
print '+++++|receiving file from server:', dst_filename
try:
with open(dst_filename, 'wb') as f:
self.request.send("FileOk2Send")
file_content = receive_data(self.request)
print 'write data into file....'
f.write(file_content)
except IOError:
print dst_filename, 'not exist'
self.request.send(
"FileTransferError:cannot save file to %s" %
dst_filename)
return "FileTransferError..."
md5_key = md5_file(dst_filename)
print '+++++|verfiying md5key---'
if md5_key == md5_key_from_client:
print 'file transfer verification ok'
self.request.send('FileTransferComplete')
if compress_mark == 'tarAndCompressed':
temp_tar_file = 'temp/%s.tar' % source_file
compress.decompress(dst_filename, temp_tar_file)
compress.untar(temp_tar_file, file_info[3])
os.remove(temp_tar_file)
os.remove(dst_filename)
elif compress_mark == 'compressed':
target_path = dst_filename.split('/%s' %
source_file)[0]
print target_path
target_file = '%s/%s' % (target_path, source_file[:-2])
compress.decompress(dst_filename, target_file)
os.remove(dst_filename)
else:
print "file not complete"
self.request.send('FileTransferNotComplete')
