1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
#!/usr/bin/env python3
import argparse
import sys
import smtplib
import traceback
from emailcanary import canarydb
from emailcanary import canary
SUCCESS=0
FAILURE=1
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('-d', '--database', default='/etc/emailcanary.db', help='Specify the database to use.')
parser.add_argument('-s', '--smtp', default='localhost:465', help='SMTP Server to send chirps to.')
parser.add_argument('-f', '--from', dest='fromaddress', help='Specify the email address to send the ping from.')
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--chirp', metavar='listAddress', help='Send an email to the given canary list address.')
group.add_argument('--check', metavar='listAddress', help='Check the recepient addresses for the list address.')
group.add_argument('--add', nargs=4, metavar=('listAddress', 'address', 'imapserver', 'password'), help='Add a new recepient and list.')
group.add_argument('--remove', nargs='+', metavar=('listAddress', 'address'), help='Remove recepients from the list.')
group.add_argument('--list', action='store_true', help='List all configured addresses.')
args = parser.parse_args()
return args
def get_smtp(address):
index = address.find(':')
if index == -1:
server = address
port = 465
else:
(server, port) = address.split(':')
return smtplib.SMTP_SSL(server, port)
def list(db):
accounts = db.get_accounts()
if len(accounts) == 0:
print('No accounts configured.')
return SUCCESS
print("%-25s %-25s %-25s" % ('List Address', 'Recepient', 'IMAP Server'))
print("-" * 80)
for account in accounts:
print("%-25s %-25s %-25s" % (account[0], account[1], account[2]))
return SUCCESS
def add(db, listAddress, recepient, imapserver, password):
db.add_account(listAddress, recepient, imapserver, password)
return SUCCESS
def remove(db, listAddress, recepients):
if len(recepients) == 0:
recepients = db.get_accounts(listAddress)
for address in recepients:
db.remove_account(listAddress, address)
return SUCCESS
def check(db, birdie, listAddress):
missing = birdie.check(listAddress)
missing = [x for x in missing if x[3].total_seconds() < 86400]
missing = [x for x in missing if x[3].total_seconds() > 3600]
if len(missing) == 0:
return SUCCESS
print("list recepient uuid time")
for chirp in missing:
# Only Print chirps which were missing for more than 15 minutes.
print("%s %s %s %d" % (chirp[0], chirp[1], chirp[2], chirp[3].total_seconds()))
return FAILURE
def main():
args = parse_args()
if not args:
return
smtp = None
db = None
try:
db = canarydb.CanaryDB(args.database)
if args.list:
return list(db)
elif args.add:
return add(db, args.add[0], args.add[1], args.add[2], args.add[3])
elif args.remove:
return remove(db, args.remove[0], args.remove[1:])
else:
smtp = get_smtp(args.smtp)
birdie = canary.Canary(db, smtp, args.fromaddress)
if args.chirp:
birdie.chirp(args.chirp)
return SUCCESS
elif args.check:
return check(db, birdie, args.check)
else:
raise Exception('Unknown action')
except Exception as e:
sys.stderr.write("Error: %s\n" % (str(e)))
traceback.print_exc()
return FAILURE
finally:
if smtp:
smtp.quit()
if db:
db.close()
if __name__ == "__main__":
sys.exit(main())
|