forked from p34709852/monkey
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
import socket
|
|
import sys
|
|
|
|
import array
|
|
|
|
import struct
|
|
from cc.database import mongo
|
|
|
|
__author__ = 'Barak'
|
|
|
|
|
|
# data structures
|
|
|
|
def creds_add_username(username):
|
|
mongo.db.usernames.update(
|
|
{'username': username},
|
|
{'$inc': {'count': 1}},
|
|
upsert=True
|
|
)
|
|
|
|
|
|
def creds_add_password(password):
|
|
mongo.db.passwords.update(
|
|
{'password': password},
|
|
{'$inc': {'count': 1}},
|
|
upsert=True
|
|
)
|
|
|
|
|
|
def init_collections(usernames, passwords):
|
|
if "usernames" not in mongo.db.collection_names():
|
|
mongo.db.usernames.create_index([("username", 1)], unique=True)
|
|
for username in usernames:
|
|
creds_add_username(username)
|
|
|
|
if "passwords" not in mongo.db.collection_names():
|
|
mongo.db.passwords.create_index([("password", 1)], unique=True)
|
|
for password in passwords:
|
|
creds_add_password(password)
|
|
|
|
|
|
# Local ips function
|
|
if sys.platform == "win32":
|
|
def local_ips():
|
|
local_hostname = socket.gethostname()
|
|
return socket.gethostbyname_ex(local_hostname)[2]
|
|
else:
|
|
import fcntl
|
|
def local_ips():
|
|
result = []
|
|
try:
|
|
is_64bits = sys.maxsize > 2 ** 32
|
|
struct_size = 40 if is_64bits else 32
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
max_possible = 8 # initial value
|
|
while True:
|
|
struct_bytes = max_possible * struct_size
|
|
names = array.array('B', '\0' * struct_bytes)
|
|
outbytes = struct.unpack('iL', fcntl.ioctl(
|
|
s.fileno(),
|
|
0x8912, # SIOCGIFCONF
|
|
struct.pack('iL', struct_bytes, names.buffer_info()[0])
|
|
))[0]
|
|
if outbytes == struct_bytes:
|
|
max_possible *= 2
|
|
else:
|
|
break
|
|
namestr = names.tostring()
|
|
|
|
for i in range(0, outbytes, struct_size):
|
|
addr = socket.inet_ntoa(namestr[i + 20:i + 24])
|
|
if not addr.startswith('127'):
|
|
result.append(addr)
|
|
# name of interface is (namestr[i:i+16].split('\0', 1)[0]
|
|
finally:
|
|
return result
|
|
# End of local ips function
|