def init():
bot_id = '1437569240:AAEd2sZ0faC1EwPvQGJPPW4xf7ohP1hTzV8'
updater = Updater(bot_id)
updater.setPhotoHandler(imageHandler)
QualityChecker.init()
ShoeDetector.init()
FeatureExtractor.init()
data_structure = Indexer.build_data_structure(config.DATASET_PATH)
Matcher.init(data_structure)
print("Bot is running...")
updater.start()
def main(argv):
def __call__(self):
pass
stdscr = curses.initscr()
curses.noecho() # @UndefinedVariable
curses.cbreak() # @UndefinedVariable
stdscr.keypad(1)
curses.start_color()
try:
wl = WindowLogic(stdscr)
#wl.start()
except Exception as e:
raise
try:
dlog = wl.dlog
dlog.msg("Logging debug output to " + str(dlog.outputFile))
dlog.msg("Images will be cached in " +
wl.cfg.get('file.image.directory'))
ci = CommandInterpreter(stdscr, wl)
ci.start()
updater = Updater(stdscr, wl)
updater.start()
except Exception as e:
dlog.excpt(e)
raise
ci.join()
dlog.msg("Command Interpreter joined.")
updater.stop()
updater.join()
dlog.msg("Updater joined.")
#wl.stop()
#wl.join()
dlog.msg("Thread Fetcher joined.")
curses.nocbreak() # @UndefinedVariable
stdscr.keypad(0)
curses.echo() # @UndefinedVariable
curses.endwin() # @UndefinedVariable
curses.resetty() # @UndefinedVariable
dlog.msg("Terminal restored.")
class MainApplication(tk.Frame):
def __init__(self, parent, *args, **kwargs):
tk.Frame.__init__(self, parent, *args, **kwargs)
self.parent = parent
self.uiMenu = UIMenu(self.parent)
self.parent.config(menu=self.uiMenu.menu)
# self.toolbar = ttk.Frame(self.parent)
# self.toolbarButton1 = ttk.Button(self.toolbar, text='Button')
# self.toolbarButton1.pack(side=LEFT, padx=2, pady=2)
# self.toolbar.pack(side=TOP, fill=X)
self.statusbar = Label(self.parent, text='Ready', bd=1, relief=SUNKEN, anchor=W)
self.statusbar.pack(side=BOTTOM, fill=X)
self.notebook = ttk.Notebook(self.parent)
self.frameSniff = ttk.Frame(self.notebook)
self.frameArp = ttk.Frame(self.notebook)
self.frameDhcp = ttk.Frame(self.notebook)
self.frameSysInfo = ttk.Frame(self.notebook)
self.notebook.add(self.frameSniff, text='Sniffer')
self.notebook.add(self.frameArp, text='ARP')
self.notebook.add(self.frameDhcp, text='DHCP Servers')
self.notebook.add(self.frameSysInfo, text='System Info')
self.notebook.pack(fill=BOTH)
self.sniffLabelFrame = ttk.LabelFrame(self.frameSniff, text="Sniffer")
self.sniffLabelFrame.pack(padx=10, pady=10)
self.sniffButton = ttk.Button(self.sniffLabelFrame, text="Sniff", command=self.beginSniff)
self.sniffButton.pack(side=LEFT)
self.stopSniffButton = ttk.Button(self.sniffLabelFrame, text="Stop sniffing", command=self.stopSniff, state="disabled")
self.stopSniffButton.pack(side=LEFT)
self.sniffTv = ttk.Treeview(self.frameSniff)
ysb = ttk.Scrollbar(self, orient='vertical', command=self.sniffTv.yview)
xsb = ttk.Scrollbar(self, orient='horizontal', command=self.sniffTv.xview)
self.sniffTv.configure(yscroll=ysb.set, xscroll=xsb.set)
self.sniffTv['columns'] = ('senderip', 'sendermac', 'received')
self.sniffTv.heading('#0', text='Description', anchor='w')
self.sniffTv.column('#0', anchor='w')
self.sniffTv.heading('senderip', text='Sender IP')
self.sniffTv.column('senderip', width=100)
self.sniffTv.heading('sendermac', text='Sender MAC')
self.sniffTv.column('sendermac', width=100)
self.sniffTv.heading('received', text='Received at')
self.sniffTv.column('received', width=100)
self.sniffTv.pack(fill=BOTH)
self.arpLabel = ttk.Label(self.frameArp, text="ARP cache")
self.arpLabel.pack()
self.arpTv = ttk.Treeview(self.frameArp)
self.arpTv['columns'] = ('ip', 'status')
self.arpTv.heading('#0', text='MAC address', anchor='w')
self.arpTv.column('#0', anchor='w')
self.arpTv.heading('ip', text='IP address')
self.arpTv.column('ip', width=100)
self.arpTv.heading('status', text='Status')
self.arpTv.column('status', width=100)
self.arpTv.pack(fill=X)
self.addDhcpLabelFrame = ttk.LabelFrame(self.frameDhcp, text="Add trusted server")
self.addDhcpLabelFrame.pack(padx=10, pady=10)
self.addDhcpNameLabel = ttk.Label(self.addDhcpLabelFrame, text="Server name")
self.addDhcpNameLabel.pack()
self.addDhcpNameEntry = ttk.Entry(self.addDhcpLabelFrame)
self.addDhcpNameEntry.pack()
self.addDhcpIpLabel = ttk.Label(self.addDhcpLabelFrame, text="Server IP address")
self.addDhcpIpLabel.pack()
self.addDhcpIpEntry = ttk.Entry(self.addDhcpLabelFrame)
self.addDhcpIpEntry.pack()
self.addDhcpMacLabel = ttk.Label(self.addDhcpLabelFrame, text="Server Mac address")
self.addDhcpMacLabel.pack()
self.addDhcpMacEntry = ttk.Entry(self.addDhcpLabelFrame)
self.addDhcpMacEntry.pack()
self.addDhcpButton = ttk.Button(self.addDhcpLabelFrame, text="Add", command=self.addDhcpButtonPressed)
self.addDhcpButton.pack()
self.clrDhcpButton = ttk.Button(self.addDhcpLabelFrame, text="Clear", command=self.clrDhcpButtonPressed)
self.clrDhcpButton.pack()
self.dhcpTv = ttk.Treeview(self.frameDhcp)
self.dhcpTv['columns'] = ('ip', 'mac', 'date')
self.dhcpTv.heading('#0', text='Server name', anchor='w')
self.dhcpTv.column('#0', anchor='w')
self.dhcpTv.heading('ip', text='IP address')
self.dhcpTv.column('ip', width=100)
self.dhcpTv.heading('mac', text='MAC address')
self.dhcpTv.column('mac', width=100)
self.dhcpTv.heading('date', text='Date added')
self.dhcpTv.column('date', width=100)
self.dhcpTv.pack(fill=X)
strVersion = 'Python ' + platform.python_version()
self.versionLabel = ttk.Label(self.frameSysInfo, text=strVersion)
self.versionLabel.pack()
strPlatform = 'Platform: ' + platform.platform()
self.platformLabel = ttk.Label(self.frameSysInfo, text=strPlatform)
self.platformLabel.pack()
self.dhcpDefender = DhcpDefender(self)
self.arpDefender = ArpDefender(self)
self.sniffer = Sniffer(1, 'Sniffer-1', 1, self)
self.sniffer.start()
self.updater = Updater(1, 'Updater-1', 1, self)
self.updater.start()
def addDhcpButtonPressed(self):
name = self.addDhcpNameEntry.get()
ip = self.addDhcpIpEntry.get()
mac = self.addDhcpMacEntry.get().lower()
isValidIp =re.match(r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$",ip)
macRe = re.compile(r'^([0-9A-F]{1,2})' + '\:([0-9A-F]{1,2})'*5 + '$', re.IGNORECASE)
isValidMac = macRe.match(mac)
if isValidIp and isValidMac:
self.dhcpDefender.add_trusted_server(name, ip, mac)
def clrDhcpButtonPressed(self):
self.dhcpDefender.clear_db()
def beginSniff(self):
self.sniffButton.config(state="disabled")
self.stopSniffButton.config(state="normal")
self.sniffer.resume()
def stopSniff(self):
self.stopSniffButton.config(state="disabled")
self.sniffButton.config(state="normal")
self.sniffer.pause()
def updateSniffTv(self, packet):
if packet[ARP].op == 1:
response = 'Request: ' + packet[ARP].psrc + ' is asking about ' + packet[ARP].pdst
elif packet[ARP].op == 2:
response = 'Response: ' + packet[ARP].hwsrc + ' has address ' + packet[ARP].psrc
timeStr = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
self.sniffTv.insert("", 0, text=response, values=(packet[ARP].psrc, packet[ARP].hwsrc, timeStr))
def updateDhcpTv(self, trusted_servers):
self.dhcpTv.delete(*self.dhcpTv.get_children())
for server in trusted_servers:
self.dhcpTv.insert("", 0, text=server['name'], values=(server['ip'], server['mac'], server['date']))
def import_arp_cache(self):
self.arpTv.delete(*self.arpTv.get_children())
out, err = Popen(['arp', '-na'], stdout=PIPE, stderr=PIPE).communicate()
out = out.splitlines()
for line in out:
ip = self.find_between(line, '(', ')')
mac = self.find_between(line, 'at ', ' on')
self.arpTv.insert("", 0, text=mac, values=(ip, ""))
def find_between(self, s, first, last):
try:
start = s.index(first) + len(first)
end = s.index(last, start)
return s[start:end]
except ValueError:
return ""
def process_arp(self, packet):
result = self.arpDefender.arp_pkt_callback(packet)
if result == -1: # inconsistent headers
print("inconsistent headers")
self.statusbar.config(text="Warning: Inconsistent header packet detected.")
pass
elif result == 0: # failed active check
print("failed active check")
self.statusbar.config(text="Warning: You might be under ARP spoof attack.")
pass
elif result == 1: # passed active check
print("passed active check")
pass
def process_dhcp(self, packet):
result = self.dhcpDefender.dhcp_pkt_callback(packet)
if result == 0: # unknown dhcp server
print("Unknown dhcp server")
self.statusbar.config(text="Warning: Untrusted DHCP server present in network.")
pass
elif result == 1:
print("Trusted dhcp server")
pass
model_rpn.load_weights(detector_model_path, by_name=True)
model_classifier.load_weights(detector_model_path, by_name=True)
# COMPILE MODEL
model_rpn.compile(optimizer='sgd', loss='mse')
model_classifier.compile(optimizer='sgd', loss='mse')
# LOAD PREDICTION MODEL
classifier_model_path = 'models/MobileNetV2_192_NoCar.h5'
classifier = load_model(classifier_model_path)
# SETUP BOT AND FOLDERS
alphabeth = [
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'L', 'M', 'N', 'O', 'P', 'Q',
'R', 'S', 'T', 'U', 'V', 'Z'
]
if not os.path.exists("res"):
os.makedirs("res")
classif_input = 192
marker_true = cv2.imread("V_tick.png", -1)
marker_false = cv2.imread("X_tick.png", -1)
bot_id = '*********'
updater = Updater(bot_id)
updater.setPhotoHandler(imageHandler)
print('---------- BOT IS READY')
updater.start()
