mms2mail/mms2mail

343 lines
12 KiB
Text
Raw Normal View History

2021-05-01 06:43:13 +00:00
#!/usr/bin/python3
2021-05-05 03:49:38 +00:00
"""An mms to mail converter for mmsd."""
# upstream bug dirty fix
# https://github.com/marrow/mailer/issues/87#issuecomment-689586587
2021-05-01 07:32:26 +00:00
import sys
if sys.version_info[0] == 3 and sys.version_info[1] > 7:
sys.modules["cgi.parse_qsl"] = None
2021-05-05 03:49:38 +00:00
# upstream bug dirty fix
# https://github.com/marrow/mailer/issues/87#issuecomment-713319548
2021-05-01 07:32:26 +00:00
import base64
if sys.version_info[0] == 3 and sys.version_info[1] > 8:
def encodestring(value):
2021-05-05 03:49:38 +00:00
"""
Encode string in base64.
:param value: the string to encode.
:type value: str
:rtype str
:return: the base64 encoded string
"""
2021-05-01 07:32:26 +00:00
return base64.b64encode(value)
base64.encodestring = encodestring
2021-05-05 03:49:38 +00:00
# end bugfix
2021-05-01 07:32:26 +00:00
2021-05-01 06:43:13 +00:00
import argparse
import configparser
import getpass
import socket
2021-05-06 20:12:10 +00:00
import mimetypes
import time
2021-05-01 06:43:13 +00:00
from pathlib import Path
2021-05-01 06:43:13 +00:00
from messaging.mms.message import MMSMessage
import mailbox
from marrow.mailer import Message
2021-05-01 06:43:13 +00:00
2021-05-05 03:49:38 +00:00
from gi.repository import GLib
import dbus
import dbus.mainloop.glib
2021-05-01 06:43:13 +00:00
log = __import__('logging').getLogger(__name__)
2021-05-05 03:49:38 +00:00
2021-05-01 06:43:13 +00:00
class MMS2Mail:
"""
2021-05-05 03:49:38 +00:00
The class handling the conversion between MMS and mail format.
2021-05-01 06:43:13 +00:00
MMS support is provided by python-messaging
Mail support is provided by marrow.mailer
"""
2021-05-05 03:49:38 +00:00
def __init__(self, delete=False, force_read=False,
disable_dbus=False, force_unlock=False):
"""
Return class instance.
:param delete: delete MMS after conversion
:type delete: bool
:param force_read: force converting an already read MMS (batch mode)
:type force_read: bool
:param disable_dbus: Disable sending dbus commands to mmsd (batch mode)
:type disable_dbus: bool
:param force_unlock: Force mbox unlocking after a few minutes
:type force_unlock: bool
"""
self.delete = delete
self.force_read = force_read
self.disable_dbus = disable_dbus
self.force_unlock = force_unlock
2021-05-01 06:43:13 +00:00
self.config = configparser.ConfigParser()
self.config.read(f"{Path.home()}/.mms/modemmanager/mms2mail.ini")
self.attach_mms = self.config.getboolean('mail', 'attach_mms',
fallback=False)
2021-05-05 03:49:38 +00:00
self.domain = self.config.get('mail', 'domain',
fallback=socket.getfqdn())
self.user = self.config.get('mail', 'user', fallback=getpass.getuser())
mbox_file = self.config.get('mail', 'mailbox',
fallback=f"/var/mail/{self.user}")
self.mailbox = mailbox.mbox(mbox_file)
if self.disable_dbus:
return
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
self.bus = dbus.SessionBus()
2021-05-01 06:43:13 +00:00
def get_bus(self):
"""
2021-05-05 03:49:38 +00:00
Return the DBus SessionBus.
:rtype dbus.SessionBus()
:return: an active SessionBus
"""
if self.disable_dbus:
return None
return self.bus
2021-05-05 03:49:38 +00:00
def mark_mms_read(self, dbus_path):
"""
Ask mmsd to mark the mms as read.
:param dbus_path: the mms dbus path
2021-05-05 03:49:38 +00:00
:type dbus_path: str
"""
if self.disable_dbus:
return None
2021-05-05 03:49:38 +00:00
message = dbus.Interface(self.bus.get_object('org.ofono.mms',
dbus_path),
'org.ofono.mms.Message')
log.debug(f"Marking MMS as read {dbus_path}")
message.MarkRead()
def delete_mms(self, dbus_path):
"""
Ask mmsd to delete the mms.
2021-05-05 03:49:38 +00:00
:param dbus_path: the mms dbus path
2021-05-05 03:49:38 +00:00
:type dbus_path: str
"""
if self.disable_dbus:
return None
message = dbus.Interface(self.bus.get_object('org.ofono.mms',
dbus_path),
'org.ofono.mms.Message')
log.debug(f"Deleting MMS {dbus_path}")
message.Delete()
def check_mms(self, path):
"""
Check wether the provided file would be converted.
:param path: the mms filesystem path
:type path: str
:rtype bool
"""
# Check for mmsd data file
if not Path(f"{path}").is_file():
log.error("MMS file not found : aborting")
return False
# Check for mmsd status file
2021-05-01 06:43:13 +00:00
status = configparser.ConfigParser()
if not Path(f"{path}.status").is_file():
log.error("MMS status file not found : aborting")
return False
2021-05-01 06:43:13 +00:00
status.read_file(open(f"{path}.status"))
# Allow only incoming MMS for the time beeing
if not (status['info']['state'] == 'downloaded' or
2021-05-05 03:49:38 +00:00
status['info']['state'] == 'received'):
log.error("Outgoing MMS : aborting")
return False
if not (self.force_read or not status.getboolean('info', 'read')):
log.error("Already converted MMS : aborting")
return False
return True
def convert(self, path, dbus_path=None):
"""
Convert a provided mms file to a mail stored in a mbox.
:param path: the mms filesystem path
:type path: str
:param dbus_path: the mms dbus path
:type dbus_path: str
"""
# Check if the provided file present
if not self.check_mms(path):
log.error("MMS file not convertible.")
2021-05-01 06:43:13 +00:00
return
# Generate its dbus path, for future operation (mark as read, delete)
if not dbus_path:
dbus_path = f"/org/ofono/mms/modemmanager/{path.split('/')[-1]}"
2021-05-01 06:43:13 +00:00
2021-05-01 21:11:15 +00:00
mms = MMSMessage.from_file(path)
2021-05-05 03:49:38 +00:00
message = Message()
2021-05-01 06:43:13 +00:00
# Generate Mail Headers
2021-05-05 03:49:38 +00:00
mms_from, mms_from_type = mms.headers.get('From',
'unknown/undef').split('/')
message.author = f"{mms_from}@{self.domain}"
mms_from, mms_from_type = mms.headers.get('To',
'unknown/undef').split('/')
message.to = f"{self.user}@{self.domain}"
2021-05-01 21:11:15 +00:00
if 'Subject' in mms.headers and mms.headers['Subject']:
message.subject = mms.headers['Subject']
2021-05-05 03:49:38 +00:00
else:
message.subject = f"MMS from {mms_from}"
if 'Date' in mms.headers and mms.headers['Date']:
message.date = mms.headers['Date']
2021-05-01 21:11:15 +00:00
# Recopy MMS HEADERS
for header in mms.headers:
2021-05-05 03:49:38 +00:00
message.headers.append((f"X-MMS-{header}",
f"{mms.headers[header]}"))
2021-05-06 20:12:10 +00:00
message.plain = " "
data_id = 1
2021-05-01 06:43:13 +00:00
for data_part in mms.data_parts:
2021-05-05 03:49:38 +00:00
datacontent = data_part.headers['Content-Type']
2021-05-01 06:43:13 +00:00
if datacontent is not None:
if 'text/plain' in datacontent[0]:
2021-05-06 20:12:10 +00:00
encoding = datacontent[1].get('Charset', 'utf-8')
plain = data_part.data.decode(encoding)
message.plain += plain + '\n'
continue
extension = str(mimetypes.guess_extension(datacontent[0]))
2021-05-06 20:12:10 +00:00
filename = datacontent[1].get('Name', str(data_id))
message.attach(filename + extension, data_part.data)
data_id = data_id + 1
2021-05-05 03:49:38 +00:00
# Add MMS binary file, for debugging purpose or reparsing in the future
if self.attach_mms:
2021-05-06 20:12:10 +00:00
message.attach(path, None, None, None, False, path.split('/')[-1])
# Write the mail in case of mbox lock retry for 5 minutes
# Ultimately write in an mbox in the home folder
end_time = time.time() + (5 * 60)
while True:
try:
# self.mailer.send(message)
self.mailbox.lock()
self.mailbox.add(mailbox.mboxMessage(str(message)))
self.mailbox.flush()
self.mailbox.unlock()
break
except (mailbox.ExternalClashError, FileExistsError) as e:
log.warn(f"Exception Mbox lock : {e}")
if time.time() > end_time:
if self.force_unlock:
log.error("Force removing lock")
self.mailbox.unlock()
else:
fs_mbox_path = f"{Path.home()}/.mms/failsafembox"
fs_mbox = mailbox.mbox(fs_mbox_path)
log.warning(f"Writing in internal mbox {fs_mbox_path}")
try:
fs_mbox.unlock()
fs_mbox.lock()
fs_mbox.add(mailbox.mboxMessage(str(message)))
fs_mbox.flush()
fs_mbox.unlock()
break
except (mailbox.ExternalClashError,
FileExistsError) as e:
log.error(f"Failsafe Mbox error : {e}")
log.error(f"MMS cannot be written to any mbox : \
{path.split('/')[-1]}")
finally:
break
else:
time.sleep(5)
# Ask mmsd to mark message as read and delete it
self.mark_mms_read(dbus_path)
if self.delete:
self.delete_mms(dbus_path)
2021-05-05 03:49:38 +00:00
2021-05-07 10:12:08 +00:00
class DbusWatcher():
"""Use DBus Signal notification to watch for new MMS."""
2021-05-07 10:12:08 +00:00
def __init__(self, mms2mail):
"""
2021-05-07 10:12:08 +00:00
Return a DBusWatcher instance.
2021-05-07 10:12:08 +00:00
:param mms2mail: An mms2mail instance to convert new mms
:type mms2mail: mms2mail()
"""
2021-05-07 10:12:08 +00:00
self.mms2mail = mms2mail
2021-05-05 03:49:38 +00:00
def run(self):
"""Run the watcher mainloop."""
2021-05-07 10:12:08 +00:00
bus = self.mms2mail.get_bus()
bus.add_signal_receiver(self.message_added,
2021-05-05 03:49:38 +00:00
bus_name="org.ofono.mms",
signal_name="MessageAdded",
member_keyword="member",
path_keyword="path",
interface_keyword="interface")
mainloop = GLib.MainLoop()
log.info("Starting DBus watcher mainloop")
2021-05-07 10:12:08 +00:00
try:
mainloop.run()
except KeyboardInterrupt:
log.info("Stopping DBus watcher mainloop")
2021-05-07 10:12:08 +00:00
mainloop.quit()
2021-05-05 03:49:38 +00:00
def message_added(self, name, value, member, path, interface):
"""Trigger conversion on MessageAdded signal."""
if value['Status'] == 'downloaded' or value['Status'] == 'received':
log.debug(f"New incoming MMS found ({name.split('/')[-1]})")
2021-05-07 10:12:08 +00:00
self.mms2mail.convert(value['Attachments'][0][2], name)
else:
log.debug(f"New outgoing MMS found ({name.split('/')[-1]})")
2021-05-01 21:11:15 +00:00
2021-05-05 03:49:38 +00:00
2021-05-07 10:12:08 +00:00
def main():
"""Run the different functions handling mms and mail."""
2021-05-01 06:43:13 +00:00
parser = argparse.ArgumentParser()
mode = parser.add_mutually_exclusive_group()
2021-05-05 03:49:38 +00:00
mode.add_argument("-d", "--daemon",
2021-05-07 10:12:08 +00:00
help="Use dbus signal from mmsd to trigger conversion",
action='store_true', dest='watcher')
2021-05-05 03:49:38 +00:00
mode.add_argument("-f", "--file", nargs='+',
help="Parse specified mms files and quit", dest='files')
parser.add_argument('--delete', action='store_true', dest='delete',
help="Ask mmsd to delete the converted MMS")
parser.add_argument('--disable-dbus', action='store_true',
dest='disable_dbus',
help="disable dbus request to mmsd")
parser.add_argument('--force-read', action='store_true',
dest='force_read', help="Force conversion even if MMS \
is marked as read")
parser.add_argument('--force-unlock', action='store_true',
dest='force_unlock', help="BEWARE COULD LEAD TO \
WHOLE MBOX CORRUPTION \
Force unlocking the mbox \
after a few minutes /!\\")
2021-05-01 06:43:13 +00:00
args = parser.parse_args()
m = MMS2Mail(args.delete, args.force_read,
args.disable_dbus, args.force_unlock)
2021-05-05 03:49:38 +00:00
if args.files:
for mms_file in args.files:
m.convert(mms_file)
2021-05-07 10:12:08 +00:00
elif args.watcher:
log.info("Starting mms2mail in daemon mode")
2021-05-07 10:12:08 +00:00
w = DbusWatcher(m)
2021-05-01 06:43:13 +00:00
w.run()
else:
parser.print_help()
2021-05-07 10:12:08 +00:00
if __name__ == '__main__':
main()