tox_profile/tox_savefile.py

886 lines
33 KiB
Python
Raw Normal View History

# -*- mode: python; indent-tabs-mode: nil; py-indent-offset: 4; coding: utf-8 -*-
2022-09-30 22:09:14 +02:00
"""
2022-10-01 20:18:56 +02:00
Reads a tox profile and prints out information on what's in there to stderr.
Call it with one argument, the filename of the profile for the decrypt or info
commands, or the filename of the nodes file for the nodes command.
2022-10-04 13:53:29 +02:00
4 commands are supported:
2022-10-03 00:29:36 +02:00
--command decrypt
2022-10-01 20:18:56 +02:00
decrypts the profile and writes to the result to stdout
--command info
prints info about what's in the Tox profile to stderr
--command nodes
assumes you are reading a json nodes file instead of a profile
2022-10-04 13:53:29 +02:00
--command edits
edits fields in a Tox profile with --output to a file
2022-10-01 20:18:56 +02:00
"""
"""
2022-10-04 13:53:29 +02:00
--output Destination for info/decrypt/edit/nodes/download
2022-10-01 20:18:56 +02:00
--info default='info',
2022-10-04 13:53:29 +02:00
choices=[info, save, repr, yaml,json, pprint]
2022-10-01 20:18:56 +02:00
with --info=info prints info about the profile to stderr
2022-10-04 13:53:29 +02:00
yaml,json, pprint, repr - output format
2022-10-02 07:24:58 +02:00
nmap_udp - test DHT nodes with nmap
nmap_tcp - test TCP_RELAY nodes with nmap
nmap_onion - test PATH_NODE nodes with nmap
--indent for pprint/yaml/json default=2
2022-10-01 20:18:56 +02:00
2022-10-03 00:29:36 +02:00
--nodes
2022-10-04 13:53:29 +02:00
choices=[select_tcp, select_udp, nmap_tcp, select_version, nmap_udp]
2022-10-01 20:18:56 +02:00
select_udp - select udp nodes
select_tcp - select tcp nodes
2022-10-02 05:36:34 +02:00
nmap_udp - test UDP nodes with nmap
nmap_tcp - test TCP nodes with nmap
2022-10-01 20:18:56 +02:00
select_version - select nodes that are the latest version
download - download nodes from --download_nodes_url
--download_nodes_url https://nodes.tox.chat/json
2022-10-04 13:53:29 +02:00
--edit
help - print a summary of what fields can be edited
section,num,key,val - edit the field section,num,key with val
2022-09-30 22:09:14 +02:00
"""
2022-09-24 04:55:48 +02:00
# originally from:
# https://stackoverflow.com/questions/30901873/what-format-are-tox-files-stored-in
2022-09-24 04:55:48 +02:00
import sys
import os
2022-10-03 07:35:26 +02:00
import time
import struct
from socket import inet_ntop, AF_INET6, AF_INET
2022-09-24 04:55:48 +02:00
import logging
2022-10-01 20:18:56 +02:00
import argparse
from pprint import pprint
import shutil
2022-09-24 04:14:56 +02:00
try:
2022-09-24 04:55:48 +02:00
# https://pypi.org/project/msgpack/
2022-09-24 04:14:56 +02:00
import msgpack
except ImportError as e:
msgpack = None
2022-09-30 22:09:14 +02:00
try:
import yaml
except ImportError as e:
yaml = None
try:
2022-10-01 20:18:56 +02:00
import json
except ImportError as e:
2022-10-01 20:18:56 +02:00
json = None
try:
# https://pypi.org/project/coloredlogs/
import coloredlogs
if 'COLOREDLOGS_LEVEL_STYLES' not in os.environ:
os.environ['COLOREDLOGS_LEVEL_STYLES'] = 'spam=22;debug=28;verbose=34;notice=220;warning=202;success=118,bold;error=124;critical=background=red'
except ImportError as e:
coloredlogs = False
2022-10-01 20:18:56 +02:00
try:
# https://git.plastiras.org/emdee/toxygen_wrapper
from wrapper.toxencryptsave import ToxEncryptSave
2022-10-03 07:35:26 +02:00
from wrapper_tests.support_http import download_url, bAreWeConnected
2022-10-01 20:18:56 +02:00
except ImportError as e:
print(f"Import Error {e}")
print("Download toxygen_wrapper to deal with encrypted tox files, from:")
print("https://git.plastiras.org/emdee/toxygen_wrapper")
print("Just put the parent of the wrapper directory on your PYTHONPATH")
print("You also need to link your libtoxcore.so and libtoxav.so")
print("and libtoxencryptsave.so into wrapper/../libs/")
print("Link all 3 from libtoxcore.so if you have only libtoxcore.so")
ToxEncryptSave = None
2022-10-02 10:44:33 +02:00
download_url = None
2022-10-03 00:29:36 +02:00
2022-10-01 20:18:56 +02:00
LOG = logging.getLogger('TSF')
2022-10-02 07:24:58 +02:00
2022-10-03 00:29:36 +02:00
# Fix for Windows
2022-10-01 20:18:56 +02:00
sDIR = os.environ.get('TMPDIR', '/tmp')
sTOX_VERSION = "1000002018"
2022-10-03 00:29:36 +02:00
bHAVE_NMAP = shutil.which('nmap')
2022-10-01 20:18:56 +02:00
bHAVE_JQ = shutil.which('jq')
2022-10-03 07:35:26 +02:00
bHAVE_BASH = shutil.which('bash')
2022-10-03 00:29:36 +02:00
bMARK = b'\x00\x00\x00\x00\x1f\x1b\xed\x15'
bDEBUG = 'DEBUG' in os.environ and os.environ['DEBUG'] != 0
def trace(s): LOG.log(LOG.level, '+ ' +s)
LOG.trace = trace
2022-10-03 07:35:26 +02:00
global bOUT, aOUT, sENC
2022-10-03 03:27:38 +02:00
aOUT = {}
bOUT = b''
sENC = 'utf-8'
# grep '#''#' logging_tox_savefile.py|sed -e 's/.* //'
sEDIT_HELP = """
NAME,.,Nick_name,str
STATUSMESSAGE,.,Status_message,str
STATUS,.,Online_status,int
"""
#messenger.c
MESSENGER_STATE_TYPE_NOSPAMKEYS = 1
MESSENGER_STATE_TYPE_DHT = 2
MESSENGER_STATE_TYPE_FRIENDS = 3
MESSENGER_STATE_TYPE_NAME = 4
MESSENGER_STATE_TYPE_STATUSMESSAGE = 5
MESSENGER_STATE_TYPE_STATUS = 6
MESSENGER_STATE_TYPE_GROUPS = 7
MESSENGER_STATE_TYPE_TCP_RELAY = 10
MESSENGER_STATE_TYPE_PATH_NODE = 11
MESSENGER_STATE_TYPE_CONFERENCES = 20
MESSENGER_STATE_TYPE_END = 255
dSTATE_TYPE = {
MESSENGER_STATE_TYPE_NOSPAMKEYS: "NOSPAMKEYS",
MESSENGER_STATE_TYPE_DHT: "DHT",
MESSENGER_STATE_TYPE_FRIENDS: "FRIENDS",
MESSENGER_STATE_TYPE_NAME: "NAME",
MESSENGER_STATE_TYPE_STATUSMESSAGE: "STATUSMESSAGE",
MESSENGER_STATE_TYPE_STATUS: "STATUS",
MESSENGER_STATE_TYPE_GROUPS: "GROUPS",
MESSENGER_STATE_TYPE_TCP_RELAY: "TCP_RELAY",
MESSENGER_STATE_TYPE_PATH_NODE: "PATH_NODE",
MESSENGER_STATE_TYPE_CONFERENCES: "CONFERENCES",
MESSENGER_STATE_TYPE_END: "END",
}
def decrypt_data(data):
from getpass import getpass
if not ToxEncryptSave: return data
oToxES = ToxEncryptSave()
if not oToxES.is_data_encrypted(data):
2022-09-24 04:14:56 +02:00
LOG.debug('Not encrypted')
return data
assert data[:8] == b'toxEsave', data[:8]
sys.stdout.flush()
password = getpass('Password: ')
assert password
newData = oToxES.pass_decrypt(data, password)
LOG.debug('Decrypted: ' +str(len(newData)) +' bytes')
return newData
2022-09-24 04:14:56 +02:00
def str_to_hex(raw_id, length=None):
if length is None: length = len(raw_id)
res = ''.join('{:02x}'.format(ord(raw_id[i])) for i in range(length))
return res.upper()
2022-09-24 04:14:56 +02:00
def bin_to_hex(raw_id, length=None):
if length is None: length = len(raw_id)
res = ''.join('{:02x}'.format(raw_id[i]) for i in range(length))
return res.upper()
2022-10-01 20:18:56 +02:00
def lProcessFriends(state, index, length, result):
2022-09-30 22:09:14 +02:00
"""Friend:
The integers in this structure are stored in Big Endian format.
Length Contents
1 uint8_t Status
32 Long term public key
1024 Friend request message as a byte string
1 PADDING
2 uint16_t Size of the friend request message
128 Name as a byte string
2 uint16_t Size of the name
1007 Status message as a byte string
1 PADDING
2 uint16_t Size of the status message
1 uint8_t User status (see also: USERSTATUS)
3 PADDING
4 uint32_t Nospam (only used for sending a friend request)
8 uint64_t Last seen time
"""
dStatus = { # Status Meaning
0: 'Not a friend',
1: 'Friend added',
2: 'Friend request sent',
3: 'Confirmed friend',
4: 'Friend online'
}
slen = 1+32+1024+1+2+128+2+1007+1+2+1+3+4+8 # 2216
assert length % slen == 0
lIN = []
for i in range(length // slen):
delta = i*slen
status = struct.unpack_from(">b", result, delta)[0]
o = delta+1; l = 32
pk = bin_to_hex(result[o:o+l], l)
o = delta+1+32+1024+1+2+128; l = 2
nsize = struct.unpack_from(">H", result, o)[0]
o = delta+1+32+1024+1+2; l = 128
2022-10-03 00:29:36 +02:00
name = str(result[o:o+nsize], sENC)
2022-09-30 22:09:14 +02:00
o = delta+1+32+1024+1+2+128+2+1007; l = 2
msize = struct.unpack_from(">H", result, o)[0]
o = delta+1+32+1024+1+2+128+2; l = 1007
2022-10-03 00:29:36 +02:00
mame = str(result[o:o+msize], sENC)
2022-09-30 22:09:14 +02:00
LOG.info(f"Friend #{i} {dStatus[status]} {name} {pk}")
lIN += [{"Status": dStatus[status],
"Name": name,
"Pk": pk}]
return lIN
2022-10-02 10:44:33 +02:00
def lProcessGroups(state, index, length, result, label="GROUPS"):
"""
No GROUPS description in spec.html
"""
2022-10-03 00:29:36 +02:00
global sENC
2022-09-30 22:09:14 +02:00
lIN = []
i = 0
2022-10-01 20:18:56 +02:00
if not msgpack:
2022-10-02 08:12:29 +02:00
LOG.warn(f"process_chunk Groups = NO msgpack bytes={length}")
2022-10-01 20:18:56 +02:00
return []
try:
groups = msgpack.loads(result, raw=True)
2022-10-02 10:44:33 +02:00
LOG.info(f"{label} {len(groups)} groups")
2022-10-01 20:18:56 +02:00
for group in groups:
assert len(group) == 7, group
i += 1
2022-10-03 00:29:36 +02:00
2022-10-01 20:18:56 +02:00
state_values, \
state_bin, \
topic_info, \
mod_list, \
keys, \
self_info, \
saved_peers, = group
assert len(state_values) == 8, state_values
manually_disconnected, \
group_name_len, \
privacy_state, \
maxpeers, \
password_length, \
version, \
topic_lock, \
voice_state = state_values
LOG.info(f"lProcessGroups #{i} version={version}")
2022-10-03 00:29:36 +02:00
dBINS = {"Version": version,
"Privacy_state": privacy_state}
2022-10-01 20:18:56 +02:00
lIN += [{"State_values": dBINS}]
assert len(state_bin) == 5, state_bin
shared_state_sig, \
founder_public_key, \
group_name_len, \
password_length, \
mod_list_hash = state_bin
LOG.info(f"lProcessGroups #{i} founder_public_key={bin_to_hex(founder_public_key)}")
dBINS = {"Founder_public_key": bin_to_hex(founder_public_key)}
lIN += [{"State_bin": dBINS}]
assert len(topic_info) == 6, topic_info
2022-10-03 00:29:36 +02:00
topic_info_topic = str(topic_info[3], sENC)
2022-10-01 20:18:56 +02:00
LOG.info(f"lProcessGroups #{i} topic_info_topic={topic_info_topic}")
dBINS = {"topic_info_topic": topic_info_topic}
lIN += [{"Topic_info": dBINS}]
assert len(mod_list) == 2, mod_list
num_moderators = mod_list[0]
2022-10-02 10:44:33 +02:00
LOG.info(f"lProcessGroups #{i} num moderators={mod_list[0]}")
2022-10-01 20:18:56 +02:00
#define CRYPTO_SIGN_PUBLIC_KEY_SIZE 32
mods = mod_list[1]
assert len(mods) % 32 == 0, len(mods)
assert len(mods) == num_moderators * 32, len(mods)
lMODS = []
for j in range(num_moderators):
mod = mods[j*32:j*32 + 32]
LOG.info(f"lProcessGroups group#{i} mod#{j} sig_pk={bin_to_hex(mod)}")
lMODS += [{"Sig_pk": bin_to_hex(mod)}]
2022-10-02 10:44:33 +02:00
lIN += [{"Moderators": lMODS}]
2022-10-01 20:18:56 +02:00
assert len(keys) == 4, keys
LOG.debug(f"lProcessGroups #{i} {repr(list(map(len, keys)))}")
chat_public_key, \
chat_secret_key, \
self_public_key, \
self_secret_key = keys
LOG.info(f"lProcessGroups #{i} chat_public_key={bin_to_hex(chat_public_key)}")
lIN[0].update({"Chat_public_key": bin_to_hex(chat_public_key)})
if int(bin_to_hex(chat_secret_key), 16) != 0:
# 192 * b'0'
LOG.info(f"lProcessGroups #{i} chat_secret_key={bin_to_hex(chat_secret_key)}")
lIN[0].update({"Chat_secret_key": bin_to_hex(chat_secret_key)})
LOG.info(f"lProcessGroups #{i} self_public_key={bin_to_hex(self_public_key)}")
lIN[0].update({"Self_public_key": bin_to_hex(self_public_key)})
LOG.info(f"lProcessGroups #{i} self_secret_key={bin_to_hex(self_secret_key)}")
lIN[0].update({"Self_secret_key": bin_to_hex(self_secret_key)})
assert len(self_info) == 4, self_info
self_nick_len, self_role, self_status, self_nick = self_info
2022-10-03 00:29:36 +02:00
self_nick = str(self_nick, sENC)
2022-10-01 20:18:56 +02:00
LOG.info(f"lProcessGroups #{i} self_nick={self_nick}")
dBINS = {"Self_nick": self_nick}
lIN += [{"Self_info": dBINS}]
assert len(saved_peers) == 2, saved_peers
except Exception as e:
LOG.warn(f"process_chunk Groups #{i} error={e}")
2022-09-30 22:09:14 +02:00
return lIN
2022-10-02 07:24:58 +02:00
def lProcessNodeInfo(state, index, length, result, label="DHTnode"):
2022-09-30 22:09:14 +02:00
"""Node Info (packed node format)
The Node Info data structure contains a Transport Protocol, a Socket
Address, and a Public Key. This is sufficient information to start
communicating with that node. The binary representation of a Node Info is
called the packed node format.
Length Type Contents
1 bit Transport Protocol UDP = 0, TCP = 1
7 bit Address Family 2 = IPv4, 10 = IPv6
4 | 16 IP address 4 bytes for IPv4, 16 bytes for IPv6
2 Port Number Port number
32 Public Key Node ID
"""
delta = 0
relay = 0
lIN = []
while length > 0:
status = struct.unpack_from(">B", result, delta)[0]
if status >= 128:
2022-10-02 08:12:29 +02:00
prot = 'TCP'
2022-09-30 22:09:14 +02:00
af = status - 128
else:
2022-10-02 08:12:29 +02:00
prot = 'UDP'
2022-09-30 22:09:14 +02:00
af = status
if af == 2:
af = 'IPv4'
alen = 4
ipaddr = inet_ntop(AF_INET, result[delta+1:delta+1+alen])
else:
af = 'IPv6'
alen = 16
ipaddr = inet_ntop(AF_INET6, result[delta+1:delta+1+alen])
2022-10-02 05:36:34 +02:00
total = 1 + alen + 2 + 32
port = int(struct.unpack_from(">H", result, delta+1+alen)[0])
pk = bin_to_hex(result[delta+1+alen+2:delta+1+alen+2+32], 32)
2022-10-02 08:12:29 +02:00
LOG.info(f"{label} #{relay} bytes={length} status={status} prot={prot} af={af} ip={ipaddr} port={port} pk={pk}")
2022-10-02 05:36:34 +02:00
lIN += [{"Bytes": length,
"Status": status,
2022-10-02 08:12:29 +02:00
"Prot": prot,
2022-10-02 05:36:34 +02:00
"Af": af,
"Ip": ipaddr,
"Port": port,
"Pk": pk}]
2022-09-30 22:09:14 +02:00
delta += total
length -= total
relay += 1
return lIN
2022-10-02 07:24:58 +02:00
def lProcessDHTnodes(state, index, length, result, label="DHTnode"):
2022-09-30 22:09:14 +02:00
relay = 0
status = struct.unpack_from("<L", result, 0)[0]
# 4 uint32_t (0x159000D)
assert status == 0x159000D
length -= 4
delta = 4
lIN = []
while length > 0:
slen = struct.unpack_from("<L", result, delta)[0]
stype = struct.unpack_from("<H", result, delta+4)[0]
smark = struct.unpack_from("<H", result, delta+6)[0]
assert smark == 0x11CE
total = slen + 4 + 2 + 2
subtotal = 0
offset = delta
while offset < slen: #loop over nodes
status = struct.unpack_from(">B", result, offset+8)[0]
assert status < 12
2022-10-02 08:12:29 +02:00
prot = 'UDP'
2022-09-30 22:09:14 +02:00
if status == 2:
af = 'IPv4'
alen = 4
ipaddr = inet_ntop(AF_INET, result[offset+8+1:offset+8+1+alen])
else:
af = 'IPv6'
alen = 16
ipaddr = inet_ntop(AF_INET6, result[offset+8+1:offset+8+1+alen])
subtotal = 1 + alen + 2 + 32
port = int(struct.unpack_from(">H", result, offset+8+1+alen)[0])
pk = bin_to_hex(result[offset+8+1+alen+2:offset+8+1+alen+2+32], 32)
2022-10-02 07:24:58 +02:00
LOG.info(f"{label} #{relay} status={status} ipaddr={ipaddr} port={port} {pk}")
2022-10-02 08:12:29 +02:00
lIN += [{
"Status": status,
"Prot": prot,
"Af": af,
"Ip": ipaddr,
"Port": port,
"Pk": pk}]
2022-09-30 22:09:14 +02:00
offset += subtotal
delta += total
length -= total
relay += 1
return lIN
2022-10-03 00:29:36 +02:00
def process_chunk(index, state, oArgs=None):
2022-10-03 07:35:26 +02:00
global bOUT, aOUT
2022-10-03 00:29:36 +02:00
global sENC
2022-09-30 22:09:14 +02:00
2022-10-02 10:44:33 +02:00
length = struct.unpack_from("<I", state, index)[0]
data_type = struct.unpack_from("<H", state, index + 4)[0]
2022-10-02 10:44:33 +02:00
check = struct.unpack_from("<H", state, index + 6)[0]
assert check == 0x01CE, check
2022-10-01 20:18:56 +02:00
new_index = index + length + 8
result = state[index + 8:index + 8 + length]
2022-10-03 00:29:36 +02:00
2022-10-02 10:44:33 +02:00
label = dSTATE_TYPE[data_type]
2022-10-03 00:29:36 +02:00
if oArgs.command == 'edit' and oArgs.edit:
section,num,key,val = oArgs.edit.split(',',3)
2022-10-03 03:27:38 +02:00
2022-10-03 00:29:36 +02:00
diff = index - len(bOUT)
2022-10-03 03:13:26 +02:00
if bDEBUG and diff > 0:
2022-10-03 00:29:36 +02:00
LOG.warn(f"PROCESS_CHUNK {label} index={index} bOUT={len(bOUT)} delta={diff} length={length}")
elif bDEBUG:
LOG.trace(f"PROCESS_CHUNK {label} index={index} bOUT={len(bOUT)} delta={diff} length={length}")
if data_type == MESSENGER_STATE_TYPE_NOSPAMKEYS:
2022-09-24 04:14:56 +02:00
nospam = bin_to_hex(result[0:4])
public_key = bin_to_hex(result[4:36])
private_key = bin_to_hex(result[36:68])
LOG.info(f"nospam = {nospam}")
LOG.info(f"public_key = {public_key}")
LOG.info(f"private_key = {private_key}")
2022-10-01 20:18:56 +02:00
aIN = {"Nospam": f"{nospam}",
"Public_key": f"{public_key}",
"Private_key": f"{private_key}"}
2022-10-03 07:35:26 +02:00
aOUT.update({label: aIN})
elif data_type == MESSENGER_STATE_TYPE_DHT:
2022-10-02 08:12:29 +02:00
LOG.debug(f"process_chunk {label} length={length}")
2022-10-01 20:18:56 +02:00
lIN = lProcessDHTnodes(state, index, length, result)
2022-10-03 07:35:26 +02:00
aOUT.update({label: lIN})
elif data_type == MESSENGER_STATE_TYPE_FRIENDS:
2022-10-02 10:44:33 +02:00
LOG.info(f"{label} {length // 2216} FRIENDS {length % 2216}")
2022-10-01 20:18:56 +02:00
lIN = lProcessFriends(state, index, length, result)
2022-10-03 07:35:26 +02:00
aOUT.update({label: lIN})
elif data_type == MESSENGER_STATE_TYPE_NAME:
2022-10-03 00:29:36 +02:00
name = str(result, sENC)
2022-10-02 10:44:33 +02:00
LOG.info(f"{label} Nick_name = " +name)
aIN = {"Nick_name": name}
2022-10-03 07:35:26 +02:00
aOUT.update({label: aIN})
2022-10-03 00:29:36 +02:00
if oArgs.command == 'edit' and section == label:
2022-10-03 00:40:43 +02:00
## NAME,.,Nick_name,str
2022-10-03 00:29:36 +02:00
if key == "Nick_name":
result = bytes(val, sENC)
length = len(result)
LOG.info(f"{label} {key} EDITED to {val}")
2022-10-03 03:27:38 +02:00
elif data_type == MESSENGER_STATE_TYPE_STATUSMESSAGE:
2022-10-03 00:29:36 +02:00
mess = str(result, sENC)
2022-10-02 10:44:33 +02:00
LOG.info(f"{label} StatusMessage = " +mess)
2022-10-01 20:18:56 +02:00
aIN = {"Status_message": mess}
2022-10-03 07:35:26 +02:00
aOUT.update({label: aIN})
2022-10-03 00:29:36 +02:00
if oArgs.command == 'edit' and section == label:
2022-10-03 00:40:43 +02:00
## STATUSMESSAGE,.,Status_message,str
2022-10-03 00:29:36 +02:00
if key == "Status_message":
result = bytes(val, sENC)
length = len(result)
LOG.info(f"{label} {key} EDITED to {val}")
2022-10-03 03:27:38 +02:00
elif data_type == MESSENGER_STATE_TYPE_STATUS:
# 1 uint8_t status (0 = online, 1 = away, 2 = busy)
dStatus = {0: 'online', 1: 'away', 2: 'busy'}
status = struct.unpack_from(">b", state, index)[0]
2022-09-30 22:09:14 +02:00
status = dStatus[status]
2022-10-02 08:12:29 +02:00
LOG.info(f"{label} = " +status)
2022-10-01 20:18:56 +02:00
aIN = {f"Online_status": status}
2022-10-03 07:35:26 +02:00
aOUT.update({label: aIN})
2022-10-03 00:29:36 +02:00
if oArgs.command == 'edit' and section == label:
2022-10-03 00:40:43 +02:00
## STATUS,.,Online_status,int
2022-10-03 00:29:36 +02:00
if key == "Online_status":
result = struct.pack(">b", int(val))
length = len(result)
LOG.info(f"{label} {key} EDITED to {val}")
2022-09-24 04:55:48 +02:00
elif data_type == MESSENGER_STATE_TYPE_GROUPS:
2022-10-02 10:44:33 +02:00
if length > 0:
lIN = lProcessGroups(state, index, length, result, label)
else:
lIN = []
LOG.info(f"NO {label}")
2022-10-03 07:35:26 +02:00
aOUT.update({label: lIN})
2022-09-30 22:09:14 +02:00
elif data_type == MESSENGER_STATE_TYPE_TCP_RELAY:
2022-10-02 10:44:33 +02:00
if length > 0:
lIN = lProcessNodeInfo(state, index, length, result, "TCPnode")
else:
lIN = []
LOG.info(f"NO {label}")
2022-10-03 07:35:26 +02:00
aOUT.update({label: lIN})
2022-09-24 04:55:48 +02:00
elif data_type == MESSENGER_STATE_TYPE_PATH_NODE:
2022-10-02 07:24:58 +02:00
#define NUM_SAVED_PATH_NODES 8
2022-10-03 07:35:26 +02:00
if not length % 8 == 0:
# this should be an assert?
LOG.warn(f"process_chunk {label} mod={length % 8}")
else:
LOG.debug(f"process_chunk {label} bytes={length}")
2022-10-02 07:24:58 +02:00
lIN = lProcessNodeInfo(state, index, length, result, "PATHnode")
2022-10-03 07:35:26 +02:00
aOUT.update({label: lIN})
2022-10-03 00:29:36 +02:00
elif data_type == MESSENGER_STATE_TYPE_CONFERENCES:
2022-10-02 08:12:29 +02:00
lIN = []
if length > 0:
2022-10-02 08:12:29 +02:00
LOG.debug(f"TODO process_chunk {label} bytes={length}")
else:
2022-10-02 08:12:29 +02:00
LOG.info(f"NO {label}")
2022-10-03 07:35:26 +02:00
aOUT.update({label: []})
2022-09-30 22:09:14 +02:00
2022-10-01 20:18:56 +02:00
elif data_type != MESSENGER_STATE_TYPE_END:
2022-10-03 00:29:36 +02:00
LOG.error("UNRECOGNIZED datatype={datatype}")
sys.exit(1)
2022-10-03 03:27:38 +02:00
2022-10-01 20:18:56 +02:00
else:
2022-10-03 00:29:36 +02:00
LOG.info("END") # That's all folks...
# drop through
# We repack as we read: or edit as we parse; simply edit result and length.
# We'll add the results back to bOUT to see if we get what we started with.
# Then will will be able to selectively null sections or selectively edit.
assert length == len(result), length
bOUT += struct.pack("<I", length) + \
struct.pack("<H", data_type) + \
struct.pack("<H", check) + \
result
2022-10-03 07:35:26 +02:00
if data_type == MESSENGER_STATE_TYPE_END or index + 8 >= len(state):
2022-10-02 10:44:33 +02:00
diff = len(bSAVE) - len(bOUT)
2022-10-03 00:29:36 +02:00
if oArgs.command != 'edit' and diff > 0:
2022-10-02 10:44:33 +02:00
# if short repacking as we read - tox_profile is padded with nulls
2022-10-03 00:29:36 +02:00
LOG.warn(f"PROCESS_CHUNK bSAVE={len(bSAVE)} bOUT={len(bOUT)} delta={diff}")
2022-10-01 20:18:56 +02:00
return
2022-10-03 00:29:36 +02:00
process_chunk(new_index, state, oArgs)
2022-10-01 20:18:56 +02:00
sNMAP_TCP = """#!/bin/bash
ip=""
declare -a ports
jq '.|with_entries(select(.key|match("nodes"))).nodes[]|select(.status_tcp)|select(.ipv4|match("."))|.ipv4,.tcp_ports' | while read line ; do
if [ -z "$ip" ] ; then
ip=`echo $line|sed -e 's/"//g'`
ports=()
continue
elif [ "$line" = '[' ] ; then
continue
elif [ "$line" = ']' ] ; then
if ! route | grep -q ^def ; then
echo ERROR no route
exit 3
fi
if [ "$ip" = '"NONE"' -o "$ip" = 'NONE' ] ; then
:
elif ping -c 1 $ip | grep '100% packet loss' ; then
echo WARN failed ping $ip
else
echo INFO $ip "${ports[*]}"
cmd="nmap -Pn -n -sT -p T:"`echo "${ports[*]}" |sed -e 's/ /,/g'`
2022-10-02 05:36:34 +02:00
echo DBUG $cmd $ip
2022-10-01 20:18:56 +02:00
$cmd $ip | grep /tcp
fi
ip=""
continue
else
port=`echo $line|sed -e 's/,//'`
ports+=($port)
fi
done"""
def vBashFileNmapTcp():
2022-10-03 07:35:26 +02:00
assert bHAVE_JQ, "jq is required for this command"
assert bHAVE_NMAP, "nmap is required for this command"
assert bHAVE_BASH, "bash is required for this command"
2022-10-01 20:18:56 +02:00
f = "NmapTcp.bash"
sFile = os.path.join(sDIR, f)
if not os.path.exists(sFile):
with open(sFile, 'wt') as iFd:
iFd.write(sNMAP_TCP)
os.chmod(sFile, 0o0775)
return sFile
def vBashFileNmapUdp():
2022-10-03 07:35:26 +02:00
assert bHAVE_JQ, "jq is required for this command"
assert bHAVE_NMAP, "nmap is required for this command"
assert bHAVE_BASH, "bash is required for this command"
2022-10-01 20:18:56 +02:00
f = "NmapUdp.bash"
sFile = os.path.join(sDIR, f)
if not os.path.exists(sFile):
with open(sFile, 'wt') as iFd:
iFd.write(sNMAP_TCP.
replace('nmap -Pn -n -sT -p T',
'nmap -Pn -n -sU -p U').
replace('tcp_ports','udp_ports').
replace('status_tcp','status_udp'))
os.chmod(sFile, 0o0775)
return sFile
2022-10-02 05:36:34 +02:00
def vOsSystemNmapUdp(l, oArgs):
iErrs = 0
for elt in aOUT["DHT"]:
2022-10-02 08:12:29 +02:00
cmd = f"sudo nmap -Pn -n -sU -p U:{elt['Port']} {elt['Ip']}"
2022-10-02 05:36:34 +02:00
iErrs += os.system(cmd +f" >> {oArgs.output} 2>&1")
if iErrs:
LOG.warn(f"{oArgs.info} {iErrs} ERRORs to {oArgs.output}")
print(f"{oArgs.info} {iErrs} ERRORs to {oArgs.output}")
else:
LOG.info(f"{oArgs.info} NO errors to {oArgs.output}")
print(f"{oArgs.info} NO errors to {oArgs.output}")
def vOsSystemNmapTcp(l, oArgs):
iErrs = 0
for elt in l:
2022-10-02 08:12:29 +02:00
cmd = f"sudo nmap -Pn -n -sT -p T:{elt['Port']} {elt['Ip']}"
2022-10-02 05:36:34 +02:00
print(f"{oArgs.info} NO errors to {oArgs.output}")
iErrs += os.system(cmd +f" >> {oArgs.output} 2>&1")
if iErrs:
LOG.warn(f"{oArgs.info} {iErrs} ERRORs to {oArgs.output}")
print(f"{oArgs.info} {iErrs} ERRORs to {oArgs.output}")
else:
LOG.info(f"{oArgs.info} NO errors to {oArgs.output}")
print(f"{oArgs.info} NO errors to {oArgs.output}")
def vSetupLogging(loglevel=logging.DEBUG):
global LOG
if coloredlogs:
aKw = dict(level=loglevel,
logger=LOG,
fmt='%(name)s %(levelname)s %(message)s')
coloredlogs.install(**aKw)
else:
aKw = dict(level=loglevel,
format='%(name)s %(levelname)-4s %(message)s')
logging.basicConfig(**aKw)
logging._defaultFormatter = logging.Formatter(datefmt='%m-%d %H:%M:%S')
logging._defaultFormatter.default_time_format = '%m-%d %H:%M:%S'
logging._defaultFormatter.default_msec_format = ''
2022-10-03 03:27:38 +02:00
2022-10-01 20:18:56 +02:00
def oMainArgparser(_=None):
if not os.path.exists('/proc/sys/net/ipv6'):
bIpV6 = 'False'
else:
2022-10-01 20:18:56 +02:00
bIpV6 = 'True'
lIpV6Choices=[bIpV6, 'False']
parser = argparse.ArgumentParser(epilog=__doc__)
# list(dSTATE_TYPE.values())
# ['nospamkeys', 'dht', 'friends', 'name', 'statusmessage', 'status', 'groups', 'tcp_relay', 'path_node', 'conferences']
2022-10-03 00:29:36 +02:00
2022-10-01 20:18:56 +02:00
parser.add_argument('--output', type=str, default='',
help='Destination for info/decrypt - defaults to stderr')
parser.add_argument('--command', type=str, default='info',
2022-10-03 00:29:36 +02:00
choices=['info', 'decrypt', 'nodes', 'edit'],
required=True,
2022-10-01 20:18:56 +02:00
help='Action command - default: info')
2022-10-03 00:29:36 +02:00
parser.add_argument('--edit', type=str, default='',
help='comma seperated SECTION,key,value - unfinished')
2022-10-01 20:18:56 +02:00
parser.add_argument('--indent', type=int, default=2,
help='Indent for yaml/json/pprint')
2022-10-03 00:29:36 +02:00
choices=['info', 'save', 'repr', 'yaml','json', 'pprint']
2022-10-02 07:24:58 +02:00
if bHAVE_NMAP: choices += ['nmap_tcp', 'nmap_udp', 'nmap_onion']
2022-10-01 20:18:56 +02:00
parser.add_argument('--info', type=str, default='info',
2022-10-02 07:24:58 +02:00
choices=choices,
2022-10-01 20:18:56 +02:00
help='Format for info command')
choices = []
if bHAVE_JQ:
2022-10-02 07:24:58 +02:00
choices += ['select_tcp', 'select_udp', 'select_version']
if bHAVE_NMAP: choices += ['nmap_tcp', 'nmap_udp']
2022-10-01 20:18:56 +02:00
if download_url:
choices += ['download']
parser.add_argument('--nodes', type=str, default='',
choices=choices,
help='Action for nodes command (requires jq)')
parser.add_argument('--download_nodes_url', type=str,
default='https://nodes.tox.chat/json')
2022-10-03 00:29:36 +02:00
parser.add_argument('--encoding', type=str, default=sENC)
2022-10-01 20:18:56 +02:00
parser.add_argument('profile', type=str, nargs='?', default=None,
help='tox profile file - may be encrypted')
return parser
if __name__ == '__main__':
lArgv = sys.argv[1:]
parser = oMainArgparser()
oArgs = parser.parse_args(lArgv)
2022-10-03 00:29:36 +02:00
if oArgs.command in ['edit'] and oArgs.edit == 'help':
l = list(dSTATE_TYPE.values())
l.remove('END')
print('Available Sections: ' +repr(l))
print('Supported Quads: section,num,key,type ' +sEDIT_HELP)
sys.exit(0)
2022-10-01 20:18:56 +02:00
sFile = oArgs.profile
assert os.path.isfile(sFile), sFile
2022-10-03 00:29:36 +02:00
sENC = oArgs.encoding
vSetupLogging()
2022-10-01 20:18:56 +02:00
bSAVE = open(sFile, 'rb').read()
if ToxEncryptSave and bSAVE[:8] == b'toxEsave':
try:
2022-10-01 20:18:56 +02:00
bSAVE = decrypt_data(bSAVE)
except Exception as e:
2022-10-01 20:18:56 +02:00
LOG.error(f"decrypting {sFile} - {e}")
sys.exit(1)
2022-10-01 20:18:56 +02:00
assert bSAVE
2022-10-01 20:18:56 +02:00
oStream = None
if oArgs.command == 'decrypt':
2022-10-02 08:12:29 +02:00
assert oArgs.output, "--output required for this command"
oStream = open(oArgs.output, 'wb')
iRet = oStream.write(bSAVE)
LOG.info(f"Wrote {iRet} to {oArgs.output}")
iRet = 0
2022-10-01 20:18:56 +02:00
elif oArgs.command == 'nodes':
iRet = -1
2022-10-03 07:35:26 +02:00
ep_sec = str(int(time.time()))
json_head = '{"last_scan":' +ep_sec \
+',"last_refresh":' +ep_sec \
+',"nodes":['
2022-10-01 20:18:56 +02:00
if oArgs.nodes == 'select_tcp':
assert oArgs.output, "--output required for this command"
assert bHAVE_JQ, "jq is required for this command"
2022-10-03 07:35:26 +02:00
with open(oArgs.output, 'wt') as oFd:
oFd.write(json_head)
2022-10-01 20:18:56 +02:00
cmd = f"cat '{sFile}' | jq '.|with_entries(select(.key|match(\"nodes\"))).nodes[]|select(.status_tcp)|select(.ipv4|match(\".\"))' "
2022-10-03 08:44:02 +02:00
iRet = os.system(cmd +"| sed -e '2,$s/^{/,{/'" +f" >>{oArgs.output}")
with open(oArgs.output, 'at') as oFd: oFd.write(']}\n')
2022-10-01 20:18:56 +02:00
elif oArgs.nodes == 'select_udp':
assert oArgs.output, "--output required for this command"
assert bHAVE_JQ, "jq is required for this command"
2022-10-03 07:35:26 +02:00
with open(oArgs.output, 'wt') as oFd:
oFd.write(json_head)
2022-10-01 20:18:56 +02:00
cmd = f"cat '{sFile}' | jq '.|with_entries(select(.key|match(\"nodes\"))).nodes[]|select(.status_udp)|select(.ipv4|match(\".\"))' "
2022-10-03 08:44:02 +02:00
iRet = os.system(cmd +"| sed -e '2,$s/^{/,{/'" +f" >>{oArgs.output}")
with open(oArgs.output, 'at') as oFd: oFd.write(']}\n')
2022-10-01 20:18:56 +02:00
elif oArgs.nodes == 'select_version':
assert bHAVE_JQ, "jq is required for this command"
assert oArgs.output, "--output required for this command"
2022-10-03 07:35:26 +02:00
with open(oArgs.output, 'wt') as oFd:
oFd.write(json_head)
2022-10-03 08:44:02 +02:00
cmd = f"cat '{sFile}' | jq '.|with_entries(select(.key|match(\"nodes\"))).nodes[]|select(.status_udp)|select(.version|match(\"{sTOX_VERSION}\"))'"
iRet = os.system(cmd +"| sed -e '2,$s/^{/,{/'" +f" >>{oArgs.output}")
2022-10-03 07:35:26 +02:00
with open(oArgs.output, 'at') as oFd:
2022-10-03 08:44:02 +02:00
oFd.write(']}\n')
2022-10-01 20:18:56 +02:00
2022-10-03 07:35:26 +02:00
elif oArgs.nodes == 'nmap_tcp':
2022-10-01 20:18:56 +02:00
assert oArgs.output, "--output required for this command"
if not bAreWeConnected():
2022-10-03 07:35:26 +02:00
LOG.warn(f"{oArgs.nodes} we are not connected")
cmd = vBashFileNmapTcp()
iRet = os.system(f"bash {cmd} < '{sFile}'" +f" >'{oArgs.output}'")
2022-10-03 00:29:36 +02:00
2022-10-03 07:35:26 +02:00
elif oArgs.nodes == 'nmap_udp':
2022-10-01 20:18:56 +02:00
assert oArgs.output, "--output required for this command"
if not bAreWeConnected():
2022-10-03 07:35:26 +02:00
LOG.warn(f"{oArgs.nodes} we are not connected")
cmd = vBashFileNmapUdp()
iRet = os.system(f"bash {cmd} < '{sFile}'" +f" >'{oArgs.output}'")
2022-10-03 00:29:36 +02:00
2022-10-01 20:18:56 +02:00
elif oArgs.nodes == 'download' and download_url:
if not bAreWeConnected():
2022-10-03 07:35:26 +02:00
LOG.warn(f"{oArgs.nodes} we are not connected")
url = oArgs.download_nodes_url
b = download_url(url)
if not bSAVE:
LOG.warn("failed downloading list of nodes")
2022-10-01 20:18:56 +02:00
iRet = -1
else:
2022-10-03 07:35:26 +02:00
if oArgs.output:
oStream = open(oArgs.output, 'rb')
oStream.write(b)
2022-10-01 20:18:56 +02:00
else:
2022-10-03 07:35:26 +02:00
oStream = sys.stdout
oStream.write(str(b, sENC))
iRet = -1
LOG.info(f"downloaded list of nodes to {oStream}")
2022-10-01 20:18:56 +02:00
if iRet > 0:
LOG.warn(f"{oArgs.nodes} iRet={iRet} to {oArgs.output}")
elif iRet == 0:
LOG.info(f"{oArgs.nodes} iRet={iRet} to {oArgs.output}")
2022-10-03 00:29:36 +02:00
elif oArgs.command in ['info', 'edit']:
if oArgs.command in ['edit']:
2022-10-02 10:44:33 +02:00
assert oArgs.output, "--output required for this command"
2022-10-03 00:29:36 +02:00
assert oArgs.edit != '', "--edit required for this command"
elif oArgs.command == 'info':
# assert oArgs.info != '', "--info required for this command"
if oArgs.info in ['save', 'yaml', 'json', 'repr', 'pprint']:
assert oArgs.output, "--output required for this command"
2022-10-02 10:44:33 +02:00
2022-10-01 20:18:56 +02:00
# toxEsave
2022-10-03 00:29:36 +02:00
assert bSAVE[:8] == bMARK, "Not a Tox profile"
bOUT = bMARK
2022-10-01 20:18:56 +02:00
2022-10-02 05:36:34 +02:00
iErrs = 0
2022-10-03 00:29:36 +02:00
process_chunk(len(bOUT), bSAVE, oArgs)
if not bOUT:
LOG.error(f"{oArgs.command} NO bOUT results")
else:
oStream = None
LOG.debug(f"command={oArgs.command} len bOUT={len(bOUT)} results")
if oArgs.command in ['edit'] or oArgs.info in ['save']:
LOG.debug(f"{oArgs.command} saving to {oArgs.output}")
oStream = open(oArgs.output, 'wb', encoding=None)
if oStream.write(bOUT) > 0: iRet = 0
LOG.info(f"{oArgs.info}ed iRet={iRet} to {oArgs.output}")
2022-10-02 10:44:33 +02:00
elif oArgs.info == 'info':
pass
elif oArgs.info == 'yaml' and yaml:
2022-10-03 00:29:36 +02:00
LOG.debug(f"{oArgs.command} saving to {oArgs.output}")
oStream = open(oArgs.output, 'wt', encoding=sENC)
2022-10-02 05:36:34 +02:00
yaml.dump(aOUT, stream=oStream, indent=oArgs.indent)
2022-10-03 00:29:36 +02:00
if oStream.write('\n') > 0: iRet = 0
LOG.info(f"{oArgs.info}ing iRet={iRet} to {oArgs.output}")
2022-10-01 20:18:56 +02:00
elif oArgs.info == 'json' and json:
2022-10-03 00:29:36 +02:00
LOG.debug(f"{oArgs.command} saving to {oArgs.output}")
2022-10-03 03:13:26 +02:00
oStream = open(oArgs.output, 'wt', encoding=sENC)
2022-10-02 05:36:34 +02:00
json.dump(aOUT, oStream, indent=oArgs.indent)
2022-10-03 00:29:36 +02:00
if oStream.write('\n') > 0: iRet = 0
LOG.info(f"{oArgs.info}ing iRet={iRet} to {oArgs.output}")
2022-10-01 20:18:56 +02:00
elif oArgs.info == 'repr':
2022-10-03 00:29:36 +02:00
LOG.debug(f"{oArgs.command} saving to {oArgs.output}")
oStream = open(oArgs.output, 'wt', encoding=sENC)
if oStream.write(repr(bOUT)) > 0: iRet = 0
if oStream.write('\n') > 0: iRet = 0
LOG.info(f"{oArgs.info}ing iRet={iRet} to {oArgs.output}")
2022-10-01 20:18:56 +02:00
elif oArgs.info == 'pprint':
2022-10-03 00:29:36 +02:00
LOG.debug(f"{oArgs.command} saving to {oArgs.output}")
oStream = open(oArgs.output, 'wt', encoding=sENC)
2022-10-02 05:36:34 +02:00
pprint(aOUT, stream=oStream, indent=oArgs.indent, width=80)
2022-10-03 00:29:36 +02:00
iRet = 0
LOG.info(f"{oArgs.info}ing iRet={iRet} to {oArgs.output}")
2022-10-02 07:24:58 +02:00
elif oArgs.info == 'nmap_tcp' and bHAVE_NMAP:
2022-10-02 05:36:34 +02:00
assert oArgs.output, "--output required for this command"
vOsSystemNmapTcp(aOUT["TCP_RELAY"], oArgs)
2022-10-02 07:24:58 +02:00
elif oArgs.info == 'nmap_udp' and bHAVE_NMAP:
2022-10-02 05:36:34 +02:00
assert oArgs.output, "--output required for this command"
vOsSystemNmapUdp(aOUT["DHT"], oArgs)
2022-10-02 07:24:58 +02:00
elif oArgs.info == 'nmap_onion' and bHAVE_NMAP:
assert oArgs.output, "--output required for this command"
vOsSystemNmapUdp(aOUT["PATH_NODE"], oArgs)
2022-10-02 05:36:34 +02:00
2022-10-01 20:18:56 +02:00
if oStream and oStream != sys.stdout and oStream != sys.stderr:
oStream.close()
sys.exit(0)