class procComm:
HOST = 0
#ocket.gethostbyname(socket.gethostname())
port = 0
server = 0
IS_DATA = 0
IS_CMD = 1
def __init__(self, token, port):
if (IS_SLACK == True):
self.server = SlackClient(token)
else:
self.port = port
self.server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM,
socket.IPPROTO_UDP)
self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.server.bind((self.HOST, self.port))
sys.stderr.write('Msg Bind: ' + str(self.port) + '\n')
def procMsgEx(self, data):
for AT_BOT in BOT_MSG_HEAD:
if data.startswith(AT_BOT):
sys.stderr.write('[*] msgEx:' + str(BOT_MAP[AT_BOT][0]))
self.procMsgSend(channel=BOT_MAP[AT_BOT][0],
text=data,
dataType=0)
ret = self.procMsgRecv()
ret[0]['channel'] = BOT_MAP[AT_BOT][0]
return ret[0]['text'], ret[0]['channel']
return '', 0
def procMsgConn(self):
if (IS_SLACK == True):
return self.server.rtm_connect()
else:
return True
def procMsgRecv(self):
#print('Msg Recv')
if (IS_SLACK == True):
return self.server.rtm_read()
else:
output_list = []
# self.request is the UDP socket connected to the client
data = self.server.recv(1024).strip()
sys.stderr.write('Msg Recv Data on ' +
str(self.server.getsockname()) + ' data: ' +
data + '\n')
output_list.append({'text': data, 'channel': IS_MSG_HANDLER_PORT})
return output_list
def procMsgSend(self, channel, text, dataType):
port = channel
if (IS_SLACK == True):
self.server.api_call("chat.postMessage",
channel=channel,
text=text,
as_user=True)
else:
sys.stderr.write('Msg Send on ' + str(self.server.getsockname()) +
' to ' + str(port) + '\n')
server_address = (self.HOST, port)
self.server.sendto(bytes(text + '\n'), server_address)
class procComm:
HOST = socket.gethostbyname(socket.gethostname())
port = 0
server = 0
IS_DATA = 0
IS_HINT = 1
def __init__(self,token,port):
if(IS_SLACK == True):
self.server = SlackClient(token)
else:
self.port = port
self.server = socket.socket(socket.AF_INET,socket.SOCK_DGRAM, socket.IPPROTO_UDP)
self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.server.bind((self.HOST, self.port))
sys.stderr.write('[*]Msg Bind: ' + str(self.port) + '\n')
def procMsgEx(self,data):
botmsg = BotMsgNode();
for AT_BOT in BOT_MSG_HEAD:
if data.startswith(AT_BOT):
sys.stderr.write('[*] msgEx:'+str(BOT_MAP[AT_BOT][0]) + '\n')
msg, msgType, port, chapter, milestone, msgFrom = botmsg.parseBotMsg(data)
if int(msgType) == BotMsgNode.BOT_MSG_TYPE_HINT:
self.procHintReqSend(BOT_MAP[AT_BOT][0] - self.HINT_SHIFT)
ret = self.procHintRecv()
return ret, BOT_MAP[AT_BOT][0]
else:
self.procMsgSend(
channel=BOT_MAP[AT_BOT][0], text=msg,
chapter=0, milestone=milestone,
msgFrom=msgFrom, dataType=msgType
)
ret = self.procMsgRecv()
retStr = botmsg.formatBotMsg(
msg = ret[0]['text'],msgTo = ret[0]['channel'],
msgType = ret[1]['type'],chapter = ret[1]['chapter'],
milestone = ret[2]['MS'],msgFrom = ret[2]['msgFrom']
)
ret[0]['channel'] = BOT_MAP[AT_BOT][0]
return retStr,ret[0]['channel']
return '',0
def procMsgConn(self):
if(IS_SLACK == True):
return self.server.rtm_connect()
else:
return True
def procMsgRecv(self):
#print('Msg Recv')
botmsg = BotMsgNode();
if(IS_SLACK == True):
return self.server.rtm_read()
else:
output_list = []
strPool = []
# self.request is the UDP socket connected to the client
# Data Format:
data = self.server.recv(1024).strip()
strPool = botmsg.parseBotMsg(data)
sys.stderr.write('[*] procMsgRecv '+str(self.server.getsockname())+' data: '+data +'\n')
output_list.append({'text':strPool[botmsg.BOT_MSG_IDX_MSG],'channel':strPool[botmsg.BOT_MSG_IDX_MSGTO]})
output_list.append({'type':strPool[botmsg.BOT_MSG_IDX_MSG_TYPE],'chapter':strPool[botmsg.BOT_MSG_IDX_CHAP]})
output_list.append({'MS': strPool[botmsg.BOT_MSG_IDX_MS], 'msgFrom' : strPool[botmsg.BOT_MSG_IDX_MSGFROM]})
#output_list.append({'type':strPool[botmsg.BOT_MSG_IDX_MSG_TYPE],'chapter':strPool[botmsg.BOT_MSG_IDX_CHAP]})
#output_list.append({'MS':strPool[botmsg.BOT_MSG_IDX_MS]})
return output_list
def procMsgSend(self, channel, text, chapter, milestone, msgFrom, dataType):
port = channel
botmsg = BotMsgNode();
if(IS_SLACK == True):
if(dataType == botmsg.BOT_MSG_TYPE_TXT):
self.server.api_call("chat.postMessage", channel=channel,text=text, as_user=True)
else:
msgTx = botmsg.formatBotMsg(
text, dataType, channel, chapter, milestone, msgFrom)
sys.stderr.write('[*] procMsgSend ' + str(self.server.getsockname()
) + ' to ' + str(port) + ' msg:' + msgTx + '\n')
server_address = (self.HOST, port)
self.server.sendto(bytes(msgTx + '\n'),server_address)
#print("Sent: {}".format(data))
def procHintRecv(self):
if(IS_SLACK == True):
return None
else:
data = self.server.recv(1024).strip()
sys.stderr.write('[*] Hint Recv ' + data + '\n')
return data
def procHintSend(self, msgList,channel):
if(IS_SLACK == True):
return False
else:
retStr = ''+str(msgList[0])+'!@#'+str(msgList[1])+'!$#'+str(msgList[2])
sys.stderr.write('[*]Hint Send ' + retStr + '\n')
server_address = (self.HOST, channel)
self.hintServer.sendto(bytes(retStr + '\n'), server_address)
return True
def procHintReqRecv(self):
if(IS_SLACK == True):
return False
else:
data = self.hintServer.recv(1024).strip()
sys.stderr.write('[*]Hint Req Recv ' + data + '\n')
if data == 'Hint Request':
return True
return False
def procHintReqSend(self,channel):
if(IS_SLACK == True):
return False
else:
sys.stderr.write('[*]Hint Req Send \n')
msgTx = 'Hint Request'
server_address = (self.HOST, channel)
self.server.sendto(bytes(msgTx + '\n'), server_address)
return True
def procMSSend(self,ms,channel):
if(IS_SLACK == True):
return False
else:
sys.stderr.write('[*] MS Send '+str(ms)+' to '+str(channel))
server_address = (self.HOST, channel)
self.server.sendto(bytes(ms + '\n'), server_address)
return True
def procMSRecv(self):
if(IS_SLACK == True):
return None
else:
data = self.msServer.recv(1024).strip()
sys.stderr.write('[*] MS Recv Update: ' + data + '\n')
return data