pypush-plus-plus/demo.py

282 lines
8.7 KiB
Python
Raw Permalink Normal View History

2023-05-02 12:39:11 +00:00
import json
2023-07-26 11:50:48 +00:00
import logging
2023-07-31 23:38:28 +00:00
import os
2023-07-26 11:50:48 +00:00
import threading
import time
from base64 import b64decode, b64encode
2023-05-09 21:03:27 +00:00
from getpass import getpass
2023-08-15 13:05:08 +00:00
from subprocess import PIPE, Popen
2023-05-02 12:39:11 +00:00
2023-07-23 22:55:13 +00:00
from rich.logging import RichHandler
2023-07-26 11:50:48 +00:00
import apns
import ids
2023-07-28 15:53:13 +00:00
import imessage
2023-07-26 11:50:48 +00:00
2023-07-23 22:55:13 +00:00
logging.basicConfig(
2023-07-26 22:49:41 +00:00
level=logging.NOTSET, format="%(message)s", datefmt="[%X]", handlers=[RichHandler()]
2023-07-23 22:55:13 +00:00
)
# Set sane log levels
logging.getLogger("urllib3").setLevel(logging.WARNING)
2023-08-15 13:05:08 +00:00
logging.getLogger("py.warnings").setLevel(logging.ERROR) # Ignore warnings from urllib3
2023-07-31 17:25:19 +00:00
logging.getLogger("asyncio").setLevel(logging.WARNING)
2023-07-23 22:55:13 +00:00
logging.getLogger("jelly").setLevel(logging.INFO)
logging.getLogger("nac").setLevel(logging.INFO)
2023-08-13 23:26:57 +00:00
logging.getLogger("apns").setLevel(logging.INFO)
2023-07-24 13:18:21 +00:00
logging.getLogger("albert").setLevel(logging.INFO)
logging.getLogger("ids").setLevel(logging.DEBUG)
2023-07-31 17:08:57 +00:00
logging.getLogger("bags").setLevel(logging.INFO)
2023-08-13 23:26:57 +00:00
logging.getLogger("imessage").setLevel(logging.INFO)
2023-05-03 00:53:18 +00:00
logging.captureWarnings(True)
2023-08-15 13:05:08 +00:00
process = Popen(["git", "rev-parse", "HEAD"], stdout=PIPE)
(commit_hash, err) = process.communicate()
exit_code = process.wait()
commit_hash = commit_hash.decode().strip()
2023-05-09 21:03:27 +00:00
# Try and load config.json
try:
with open("config.json", "r") as f:
CONFIG = json.load(f)
except FileNotFoundError:
CONFIG = {}
2023-05-02 12:39:11 +00:00
2023-08-15 13:05:08 +00:00
# Re-register if the commit hash has changed
if CONFIG.get("commit_hash") != commit_hash:
logging.warning("pypush commit is different, forcing re-registration...")
CONFIG["commit_hash"] = commit_hash
if "id" in CONFIG:
del CONFIG["id"]
2023-05-09 21:03:27 +00:00
conn = apns.APNSConnection(
CONFIG.get("push", {}).get("key"), CONFIG.get("push", {}).get("cert")
)
2023-05-02 17:40:06 +00:00
2023-07-26 11:50:48 +00:00
2023-05-09 22:48:44 +00:00
def safe_b64decode(s):
try:
return b64decode(s)
except:
return None
2023-07-26 11:50:48 +00:00
2023-05-09 22:48:44 +00:00
conn.connect(token=safe_b64decode(CONFIG.get("push", {}).get("token")))
2023-07-24 20:37:53 +00:00
conn.set_state(1)
2023-08-11 19:45:17 +00:00
conn.filter(["com.apple.madrid", "com.apple.private.alloy.sms"])
2023-07-26 11:50:48 +00:00
2023-05-09 21:03:27 +00:00
user = ids.IDSUser(conn)
2023-05-02 17:40:06 +00:00
2023-05-09 21:03:27 +00: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 18:10:13 +00:00
else:
2023-05-09 21:03:27 +00:00
username = input("Username: ")
password = getpass("Password: ")
2023-05-03 00:53:18 +00:00
2023-05-09 21:03:27 +00:00
user.authenticate(username, password)
2023-07-28 15:53:13 +00: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 22:46:50 +00:00
if (
CONFIG.get("id", {}).get("cert") is not None
2023-07-27 15:04:57 +00:00
and user.encryption_identity is not None
):
2023-05-09 21:03:27 +00:00
id_keypair = ids._helpers.KeyPair(CONFIG["id"]["key"], CONFIG["id"]["cert"])
user.restore_identity(id_keypair)
else:
2023-07-26 22:49:41 +00:00
logging.info("Registering new identity...")
2023-07-23 19:47:35 +00:00
import emulated.nac
2023-07-26 11:50:48 +00:00
2023-07-23 19:47:35 +00:00
vd = emulated.nac.generate_validation_data()
vd = b64encode(vd).decode()
2023-07-25 22:46:50 +00:00
2023-07-26 22:49:41 +00:00
user.register(vd)
2023-05-09 21:03:27 +00:00
2023-07-26 22:49:41 +00:00
logging.info("Waiting for incoming messages...")
2023-07-25 22:46:50 +00:00
2023-05-09 21:03:27 +00:00
# Write config.json
2023-07-26 22:49:41 +00:00
CONFIG["encryption"] = {
2023-07-27 15:04:57 +00:00
"rsa_key": user.encryption_identity.encryption_key,
"ec_key": user.encryption_identity.signing_key,
2023-07-26 22:49:41 +00:00
}
2023-07-25 22:46:50 +00:00
CONFIG["id"] = {
2023-07-26 11:50:48 +00:00
"key": user._id_keypair.key,
"cert": user._id_keypair.cert,
2023-07-25 22:46:50 +00:00
}
2023-05-09 21:03:27 +00: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 11:50:48 +00:00
2023-05-09 22:48:44 +00:00
with open("config.json", "w") as f:
2023-05-03 00:53:18 +00:00
json.dump(CONFIG, f, indent=4)
2023-07-25 22:46:50 +00:00
2023-07-27 15:52:20 +00:00
im = imessage.iMessageUser(conn, user)
2023-07-25 22:46:50 +00:00
2023-07-31 17:08:57 +00:00
INPUT_QUEUE = apns.IncomingQueue()
2023-08-15 13:05:08 +00:00
2023-07-31 17:08:57 +00:00
def input_thread():
2023-07-31 17:25:19 +00:00
from prompt_toolkit import prompt
2023-07-31 17:08:57 +00:00
2023-08-15 13:05:08 +00:00
while True:
2023-07-31 17:08:57 +00:00
try:
2023-08-15 13:05:08 +00:00
msg = prompt(">> ")
2023-07-31 17:08:57 +00:00
except:
2023-08-15 13:05:08 +00:00
msg = "quit"
2023-07-31 17:08:57 +00:00
INPUT_QUEUE.append(msg)
2023-08-15 13:05:08 +00:00
2023-07-31 17:08:57 +00:00
threading.Thread(target=input_thread, daemon=True).start()
2023-07-31 17:25:19 +00:00
2023-08-15 13:05:08 +00:00
print("Type 'help' for help")
2023-07-31 17:25:19 +00:00
2023-07-31 23:19:16 +00:00
def fixup_handle(handle):
2023-08-15 13:05:08 +00:00
if handle.startswith("tel:+"):
2023-07-31 23:19:16 +00:00
return handle
2023-08-15 13:05:08 +00:00
elif handle.startswith("mailto:"):
2023-07-31 23:19:16 +00:00
return handle
2023-08-15 13:05:08 +00:00
elif handle.startswith("tel:"):
return "tel:+" + handle[4:]
elif handle.startswith("+"):
return "tel:" + handle
2023-07-31 23:19:16 +00:00
# If the handle starts with a number
elif handle[0].isdigit():
# If the handle is 10 digits, assume it's a US number
if len(handle) == 10:
2023-08-15 13:05:08 +00:00
return "tel:+1" + handle
2023-07-31 23:19:16 +00:00
# If the handle is 11 digits, assume it's a US number with country code
elif len(handle) == 11:
2023-08-15 13:05:08 +00:00
return "tel:+" + handle
else: # Assume it's an email
return "mailto:" + handle
2023-07-31 23:19:16 +00:00
2023-07-31 17:57:50 +00:00
current_participants = []
2023-08-16 17:31:15 +00:00
sms = False
2023-07-31 18:47:27 +00:00
current_effect = None
2023-07-31 17:08:57 +00:00
while True:
2023-08-15 13:05:08 +00:00
im.activate_sms() # We must call this always since SMS could be turned off and on again, and it might have been on before this.
2023-07-31 17:08:57 +00:00
msg = im.receive()
2023-07-31 17:57:50 +00:00
if msg is not None:
2023-07-31 23:46:38 +00:00
# print(f'[{msg.sender}] {msg.text}')
2023-08-14 13:40:33 +00:00
print(str(msg))
2023-07-31 23:38:28 +00:00
2023-08-14 13:40:33 +00:00
# attachments = msg.attachments()
# if len(attachments) > 0:
# attachments_path = f"attachments/{msg.id}/"
# os.makedirs(attachments_path, exist_ok=True)
2023-07-31 23:38:28 +00:00
2023-08-14 13:40:33 +00:00
# for attachment in attachments:
# with open(attachments_path + attachment.name, "wb") as attachment_file:
# attachment_file.write(attachment.versions[0].data())
2023-07-31 23:38:28 +00:00
2023-08-14 13:40:33 +00:00
# print(f"({len(attachments)} attachment{'s have' if len(attachments) != 1 else ' has'} been downloaded and put "
# f"in {attachments_path})")
2023-08-15 13:05:08 +00:00
2023-07-31 17:08:57 +00:00
if len(INPUT_QUEUE) > 0:
msg = INPUT_QUEUE.pop()
2023-08-15 13:05:08 +00:00
if msg == "":
continue
if msg == "help" or msg == "h":
print("help (h): show this message")
print("quit (q): quit")
# print('send (s) [recipient] [message]: send a message')
print("filter (f) [recipient]: set the current chat")
print("effect (e): adds an iMessage effect to the next sent message")
print(
"note: recipient must start with tel: or mailto: and include the country code"
)
print("handle <handle>: set the current handle (for sending messages)")
print("\\: escape commands (will be removed from message)")
elif msg == "quit" or msg == "q":
2023-07-31 17:08:57 +00:00
break
2023-08-15 13:05:08 +00:00
elif (
msg == "effect"
or msg == "e"
or msg.startswith("effect ")
or msg.startswith("e ")
):
2023-07-31 18:47:27 +00:00
msg = msg.split(" ")
if len(msg) < 2 or msg[1] == "":
print("effect [effect namespace]")
else:
print(f"next message will be sent with [{msg[1]}]")
current_effect = msg[1]
2023-08-15 13:05:08 +00:00
elif (
msg == "filter"
or msg == "f"
or msg.startswith("filter ")
or msg.startswith("f ")
):
2023-07-31 17:25:19 +00:00
# Set the curernt chat
2023-08-15 13:05:08 +00:00
msg = msg.split(" ")
if len(msg) < 2 or msg[1] == "":
print("filter [recipients]")
2023-07-31 17:08:57 +00:00
else:
2023-08-16 17:31:15 +00:00
if msg[1] == "sms":
print("Filtering to SMS")
msg = msg[1:]
sms = True
else:
sms = False
2023-08-15 13:05:08 +00:00
print(f"Filtering to {[fixup_handle(h) for h in msg[1:]]}")
2023-07-31 23:19:16 +00:00
current_participants = [fixup_handle(h) for h in msg[1:]]
2023-08-15 13:05:08 +00:00
im._cache_keys(
current_participants, "com.apple.madrid"
) # Just to make things faster, and to make it error on invalid addresses
elif msg == "handle" or msg.startswith("handle "):
msg = msg.split(" ")
if len(msg) < 2 or msg[1] == "":
print("handle [handle]")
print("Available handles:")
2023-07-31 20:30:06 +00:00
for h in user.handles:
2023-07-31 23:19:16 +00:00
if h == user.current_handle:
2023-08-15 13:05:08 +00:00
print(f"\t{h} (current)")
2023-07-31 23:19:16 +00:00
else:
2023-08-15 13:05:08 +00:00
print(f"\t{h}")
2023-07-31 20:30:06 +00:00
else:
h = msg[1]
2023-07-31 23:19:16 +00:00
h = fixup_handle(h)
2023-07-31 20:30:06 +00:00
if h in user.handles:
2023-08-15 13:05:08 +00:00
print(f"Using {h} as handle")
2023-07-31 20:30:06 +00:00
user.current_handle = h
else:
2023-08-15 13:05:08 +00:00
print(f"Handle {h} not found")
2023-07-31 20:30:06 +00:00
2023-07-31 17:57:50 +00:00
elif current_participants != []:
2023-08-15 13:05:08 +00:00
if msg.startswith("\\"):
2023-07-31 17:25:19 +00:00
msg = msg[1:]
2023-08-15 20:28:02 +00:00
2023-08-16 17:31:15 +00:00
if sms:
import uuid
m = imessage.SMSReflectedMessage(
msg, user.current_handle, current_participants, uuid.uuid4()
)
else:
m = imessage.iMessage.create(im, msg, current_participants)
m.effect = current_effect
2023-08-15 20:28:02 +00:00
2023-08-16 17:31:15 +00:00
im.send(m)
2023-07-31 18:47:27 +00:00
current_effect = None
2023-07-31 17:25:19 +00:00
else:
2023-08-15 13:05:08 +00:00
print("No chat selected, use help for help")
2023-07-31 23:19:16 +00:00
time.sleep(0.1)