2023-05-02 07:39:11 -05:00
|
|
|
import json
|
2023-07-26 06:50:48 -05:00
|
|
|
import logging
|
|
|
|
import threading
|
|
|
|
import time
|
|
|
|
from base64 import b64decode, b64encode
|
2023-05-09 16:03:27 -05:00
|
|
|
from getpass import getpass
|
2023-05-02 07:39:11 -05:00
|
|
|
|
2023-07-23 17:55:13 -05:00
|
|
|
from rich.logging import RichHandler
|
|
|
|
|
2023-07-26 06:50:48 -05:00
|
|
|
import apns
|
|
|
|
import ids
|
2023-07-28 10:53:13 -05:00
|
|
|
import imessage
|
2023-07-26 06:50:48 -05:00
|
|
|
|
2023-07-23 17:55:13 -05:00
|
|
|
logging.basicConfig(
|
2023-07-26 17:49:41 -05:00
|
|
|
level=logging.NOTSET, format="%(message)s", datefmt="[%X]", handlers=[RichHandler()]
|
2023-07-23 17:55:13 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
# Set sane log levels
|
|
|
|
logging.getLogger("urllib3").setLevel(logging.WARNING)
|
2023-07-31 12:25:19 -05:00
|
|
|
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
2023-07-23 17:55:13 -05:00
|
|
|
logging.getLogger("jelly").setLevel(logging.INFO)
|
|
|
|
logging.getLogger("nac").setLevel(logging.INFO)
|
2023-07-31 12:08:57 -05:00
|
|
|
logging.getLogger("apns").setLevel(logging.INFO)
|
2023-07-24 08:18:21 -05:00
|
|
|
logging.getLogger("albert").setLevel(logging.INFO)
|
|
|
|
logging.getLogger("ids").setLevel(logging.DEBUG)
|
2023-07-31 12:08:57 -05:00
|
|
|
logging.getLogger("bags").setLevel(logging.INFO)
|
|
|
|
logging.getLogger("imessage").setLevel(logging.DEBUG)
|
2023-05-02 19:53:18 -05:00
|
|
|
|
2023-05-09 16:03:27 -05:00
|
|
|
# Try and load config.json
|
|
|
|
try:
|
|
|
|
with open("config.json", "r") as f:
|
|
|
|
CONFIG = json.load(f)
|
|
|
|
except FileNotFoundError:
|
|
|
|
CONFIG = {}
|
2023-05-02 07:39:11 -05:00
|
|
|
|
2023-05-09 16:03:27 -05:00
|
|
|
conn = apns.APNSConnection(
|
|
|
|
CONFIG.get("push", {}).get("key"), CONFIG.get("push", {}).get("cert")
|
|
|
|
)
|
2023-05-02 12:40:06 -05:00
|
|
|
|
2023-07-26 06:50:48 -05:00
|
|
|
|
2023-05-09 17:48:44 -05:00
|
|
|
def safe_b64decode(s):
|
|
|
|
try:
|
|
|
|
return b64decode(s)
|
|
|
|
except:
|
|
|
|
return None
|
2023-07-26 06:50:48 -05:00
|
|
|
|
|
|
|
|
2023-05-09 17:48:44 -05:00
|
|
|
conn.connect(token=safe_b64decode(CONFIG.get("push", {}).get("token")))
|
2023-07-24 15:37:53 -05:00
|
|
|
conn.set_state(1)
|
2023-07-24 15:43:11 -05:00
|
|
|
conn.filter(["com.apple.madrid"])
|
2023-07-26 06:50:48 -05:00
|
|
|
|
2023-05-09 16:03:27 -05:00
|
|
|
user = ids.IDSUser(conn)
|
2023-05-02 12:40:06 -05:00
|
|
|
|
2023-05-09 16:03:27 -05:00
|
|
|
if CONFIG.get("auth", {}).get("cert") is not None:
|
|
|
|
auth_keypair = ids._helpers.KeyPair(CONFIG["auth"]["key"], CONFIG["auth"]["cert"])
|
|
|
|
user_id = CONFIG["auth"]["user_id"]
|
|
|
|
handles = CONFIG["auth"]["handles"]
|
|
|
|
user.restore_authentication(auth_keypair, user_id, handles)
|
2023-05-02 13:10:13 -05:00
|
|
|
else:
|
2023-05-09 16:03:27 -05:00
|
|
|
username = input("Username: ")
|
|
|
|
password = getpass("Password: ")
|
2023-05-02 19:53:18 -05:00
|
|
|
|
2023-05-09 16:03:27 -05:00
|
|
|
user.authenticate(username, password)
|
2023-05-02 17:11:14 -05:00
|
|
|
|
2023-07-28 10:53:13 -05:00
|
|
|
user.encryption_identity = ids.identity.IDSIdentity(
|
|
|
|
encryption_key=CONFIG.get("encryption", {}).get("rsa_key"),
|
|
|
|
signing_key=CONFIG.get("encryption", {}).get("ec_key"),
|
|
|
|
)
|
2023-07-25 17:46:50 -05:00
|
|
|
|
2023-07-26 19:21:58 -05:00
|
|
|
if (
|
|
|
|
CONFIG.get("id", {}).get("cert") is not None
|
2023-07-27 10:04:57 -05:00
|
|
|
and user.encryption_identity is not None
|
2023-07-26 19:21:58 -05:00
|
|
|
):
|
2023-05-09 16:03:27 -05:00
|
|
|
id_keypair = ids._helpers.KeyPair(CONFIG["id"]["key"], CONFIG["id"]["cert"])
|
|
|
|
user.restore_identity(id_keypair)
|
|
|
|
else:
|
2023-07-26 17:49:41 -05:00
|
|
|
logging.info("Registering new identity...")
|
2023-07-23 14:47:35 -05:00
|
|
|
import emulated.nac
|
2023-07-26 06:50:48 -05:00
|
|
|
|
2023-07-23 14:47:35 -05:00
|
|
|
vd = emulated.nac.generate_validation_data()
|
|
|
|
vd = b64encode(vd).decode()
|
2023-07-25 17:46:50 -05:00
|
|
|
|
2023-07-26 17:49:41 -05:00
|
|
|
user.register(vd)
|
2023-05-09 16:03:27 -05:00
|
|
|
|
2023-07-26 17:49:41 -05:00
|
|
|
logging.info("Waiting for incoming messages...")
|
2023-07-25 17:46:50 -05:00
|
|
|
|
2023-05-09 16:03:27 -05:00
|
|
|
# Write config.json
|
2023-07-26 17:49:41 -05:00
|
|
|
CONFIG["encryption"] = {
|
2023-07-27 10:04:57 -05:00
|
|
|
"rsa_key": user.encryption_identity.encryption_key,
|
|
|
|
"ec_key": user.encryption_identity.signing_key,
|
2023-07-26 17:49:41 -05:00
|
|
|
}
|
2023-07-25 17:46:50 -05:00
|
|
|
CONFIG["id"] = {
|
2023-07-26 06:50:48 -05:00
|
|
|
"key": user._id_keypair.key,
|
|
|
|
"cert": user._id_keypair.cert,
|
2023-07-25 17:46:50 -05:00
|
|
|
}
|
2023-05-09 16:03:27 -05:00
|
|
|
CONFIG["auth"] = {
|
|
|
|
"key": user._auth_keypair.key,
|
|
|
|
"cert": user._auth_keypair.cert,
|
|
|
|
"user_id": user.user_id,
|
|
|
|
"handles": user.handles,
|
|
|
|
}
|
|
|
|
CONFIG["push"] = {
|
|
|
|
"token": b64encode(user.push_connection.token).decode(),
|
|
|
|
"key": user.push_connection.private_key,
|
|
|
|
"cert": user.push_connection.cert,
|
|
|
|
}
|
2023-07-26 06:50:48 -05:00
|
|
|
|
2023-05-09 17:48:44 -05:00
|
|
|
with open("config.json", "w") as f:
|
2023-05-02 19:53:18 -05:00
|
|
|
json.dump(CONFIG, f, indent=4)
|
2023-07-25 17:46:50 -05:00
|
|
|
|
2023-07-27 10:52:20 -05:00
|
|
|
im = imessage.iMessageUser(conn, user)
|
2023-07-25 17:46:50 -05:00
|
|
|
|
2023-07-31 12:08:57 -05:00
|
|
|
INPUT_QUEUE = apns.IncomingQueue()
|
|
|
|
|
|
|
|
def input_thread():
|
2023-07-31 12:25:19 -05:00
|
|
|
from prompt_toolkit import prompt
|
2023-07-31 12:08:57 -05:00
|
|
|
while True:
|
|
|
|
|
|
|
|
try:
|
|
|
|
msg = prompt('>> ')
|
|
|
|
except:
|
|
|
|
msg = 'quit'
|
|
|
|
INPUT_QUEUE.append(msg)
|
|
|
|
|
|
|
|
threading.Thread(target=input_thread, daemon=True).start()
|
2023-07-31 12:25:19 -05:00
|
|
|
|
|
|
|
print("Type 'help' for help")
|
|
|
|
|
2023-07-31 12:57:50 -05:00
|
|
|
current_participants = []
|
2023-07-31 12:08:57 -05:00
|
|
|
while True:
|
|
|
|
msg = im.receive()
|
2023-07-31 12:57:50 -05:00
|
|
|
if msg is not None:
|
2023-07-31 12:25:19 -05:00
|
|
|
print(f'[{msg.sender}] {msg.text}')
|
2023-07-28 16:31:27 -05:00
|
|
|
|
2023-07-31 12:08:57 -05:00
|
|
|
if len(INPUT_QUEUE) > 0:
|
|
|
|
msg = INPUT_QUEUE.pop()
|
2023-07-31 12:25:19 -05:00
|
|
|
if msg == '': continue
|
2023-07-31 12:08:57 -05:00
|
|
|
if msg == 'help' or msg == 'h':
|
|
|
|
print('help (h): show this message')
|
|
|
|
print('quit (q): quit')
|
2023-07-31 12:25:19 -05:00
|
|
|
#print('send (s) [recipient] [message]: send a message')
|
|
|
|
print('filter (f) [recipient]: set the current chat')
|
|
|
|
print('note: recipient must start with tel: or mailto: and include the country code')
|
|
|
|
print('\\: escape commands (will be removed from message)')
|
2023-07-31 12:08:57 -05:00
|
|
|
elif msg == 'quit' or msg == 'q':
|
|
|
|
break
|
2023-07-31 12:57:50 -05:00
|
|
|
elif msg.startswith('filter ') or msg.startswith('f '):
|
2023-07-31 12:25:19 -05:00
|
|
|
# Set the curernt chat
|
2023-07-31 12:08:57 -05:00
|
|
|
msg = msg.split(' ')
|
2023-07-31 12:57:50 -05:00
|
|
|
if len(msg) < 2 or msg[1] == '':
|
|
|
|
print('filter [recipients]')
|
2023-07-31 12:08:57 -05:00
|
|
|
else:
|
2023-07-31 12:57:50 -05:00
|
|
|
print(f'Filtering to {msg[1:]}')
|
|
|
|
current_participants = msg[1:]
|
|
|
|
elif current_participants != []:
|
2023-07-31 12:25:19 -05:00
|
|
|
if msg.startswith('\\'):
|
|
|
|
msg = msg[1:]
|
|
|
|
im.send(imessage.iMessage(
|
|
|
|
text=msg,
|
2023-07-31 12:57:50 -05:00
|
|
|
participants=current_participants,
|
|
|
|
sender=user.handles[0]
|
2023-07-31 12:25:19 -05:00
|
|
|
))
|
|
|
|
else:
|
|
|
|
print('No chat selected, use help for help')
|
|
|
|
|
|
|
|
# elif msg.startswith('send') or msg.startswith('s'):
|
|
|
|
# msg = msg.split(' ')
|
|
|
|
# if len(msg) < 3:
|
|
|
|
# print('send [recipient] [message]')
|
|
|
|
# else:
|
|
|
|
# im.send(imessage.iMessage(
|
|
|
|
# text=' '.join(msg[2:]),
|
|
|
|
# participants=[msg[1], user.handles[0]],
|
|
|
|
# #sender=user.handles[0]
|
|
|
|
# ))
|
2023-07-28 16:31:27 -05:00
|
|
|
|