0000: 23 21 2f 75 73 72 2f 62 69 6e 2f 65 6e 76 20 70 #!/usr/bin/env p
0010: 79 74 68 6f 6e 33 2e 31 0a 0a 69 6d 70 6f 72 74 ython3.1..import
0020: 20 63 6f 6e 66 69 67 70 61 72 73 65 72 2c 20 6f configparser, o
0030: 70 74 70 61 72 73 65 2c 20 6f 73 2c 20 70 6f 73 ptparse, os, pos
0040: 74 67 72 65 73 71 6c 2e 61 70 69 2c 20 72 65 2c tgresql.api, re,
0050: 20 73 79 73 2c 20 5f 74 68 72 65 61 64 0a 0a 63 sys, _thread..c
0060: 6c 61 73 73 20 4c 6f 67 67 65 72 3a 0a 09 5f 5f lass Logger:..__
0070: 73 6c 6f 74 73 5f 5f 20 3d 20 66 72 6f 7a 65 6e slots__ = frozen
0080: 73 65 74 28 5b 27 5f 73 79 73 6c 6f 67 27 5d 29 set(['_syslog'])
0090: 0a 0a 09 64 65 66 20 5f 5f 69 6e 69 74 5f 5f 28 ...def __init__(
00a0: 73 65 6c 66 29 3a 0a 09 09 63 6f 6e 66 69 67 2e self):...config.
00b0: 73 65 63 74 69 6f 6e 28 27 6c 6f 67 27 29 0a 09 section('log')..
00c0: 09 69 66 20 63 6f 6e 66 69 67 5b 27 73 69 6c 65 .if config['sile
00d0: 6e 74 27 5d 20 3d 3d 20 27 79 65 73 27 3a 0a 09 nt'] == 'yes':..
00e0: 09 09 73 65 6c 66 2e 5f 73 79 73 6c 6f 67 20 3d ..self._syslog =
00f0: 20 4e 6f 6e 65 0a 09 09 65 6c 73 65 3a 0a 09 09 None...else:...
0100: 09 69 6d 70 6f 72 74 20 73 79 73 6c 6f 67 0a 09 .import syslog..
0110: 09 09 73 65 6c 66 2e 5f 73 79 73 6c 6f 67 20 3d ..self._syslog =
0120: 20 73 79 73 6c 6f 67 0a 09 09 09 73 65 6c 66 2e syslog....self.
0130: 5f 73 79 73 6c 6f 67 2e 6f 70 65 6e 6c 6f 67 28 _syslog.openlog(
0140: 27 73 71 75 69 64 54 61 67 27 29 0a 0a 09 64 65 'squidTag')...de
0150: 66 20 69 6e 66 6f 28 73 65 6c 66 2c 20 6d 65 73 f info(self, mes
0160: 73 61 67 65 29 3a 0a 09 09 69 66 20 73 65 6c 66 sage):...if self
0170: 2e 5f 73 79 73 6c 6f 67 3a 0a 09 09 09 73 65 6c ._syslog:....sel
0180: 66 2e 5f 73 79 73 6c 6f 67 2e 73 79 73 6c 6f 67 f._syslog.syslog
0190: 28 73 65 6c 66 2e 5f 73 79 73 6c 6f 67 2e 4c 4f (self._syslog.LO
01a0: 47 5f 49 4e 46 4f 2c 20 6d 65 73 73 61 67 65 29 G_INFO, message)
01b0: 0a 0a 09 64 65 66 20 6e 6f 74 69 63 65 28 73 65 ...def notice(se
01c0: 6c 66 2c 20 6d 65 73 73 61 67 65 29 3a 0a 09 09 lf, message):...
01d0: 69 66 20 73 65 6c 66 2e 5f 73 79 73 6c 6f 67 3a if self._syslog:
01e0: 0a 09 09 09 73 65 6c 66 2e 5f 73 79 73 6c 6f 67 ....self._syslog
01f0: 2e 73 79 73 6c 6f 67 28 73 65 6c 66 2e 5f 73 79 .syslog(self._sy
0200: 73 6c 6f 67 2e 4c 4f 47 5f 4e 4f 54 49 43 45 2c slog.LOG_NOTICE,
0210: 20 6d 65 73 73 61 67 65 29 0a 0a 63 6c 61 73 73 message)..class
0220: 20 74 61 67 44 42 3a 0a 09 5f 5f 73 6c 6f 74 73 tagDB:..__slots
0230: 5f 5f 20 3d 20 66 72 6f 7a 65 6e 73 65 74 28 5b __ = frozenset([
0240: 27 5f 70 72 65 70 61 72 65 64 27 2c 20 27 5f 63 '_prepared', '_c
0250: 68 65 63 6b 5f 73 74 6d 74 27 2c 20 27 5f 64 62 heck_stmt', '_db
0260: 27 5d 29 0a 0a 09 64 65 66 20 5f 5f 69 6e 69 74 '])...def __init
0270: 5f 5f 28 73 65 6c 66 29 3a 0a 09 09 73 65 6c 66 __(self):...self
0280: 2e 5f 70 72 65 70 61 72 65 64 20 3d 20 73 65 74 ._prepared = set
0290: 28 29 0a 09 09 73 65 6c 66 2e 5f 64 62 20 3d 20 ()...self._db =
02a0: 46 61 6c 73 65 0a 09 09 73 65 6c 66 2e 5f 63 68 False...self._ch
02b0: 65 63 6b 5f 73 74 6d 74 20 3d 20 73 65 6c 66 2e eck_stmt = self.
02c0: 5f 63 75 72 73 28 29 2e 70 72 65 70 61 72 65 28 _curs().prepare(
02d0: 22 73 65 6c 65 63 74 20 72 65 64 69 72 65 63 74 "select redirect
02e0: 5f 75 72 6c 2c 20 72 65 67 65 78 70 20 66 72 6f _url, regexp fro
02f0: 6d 20 73 69 74 65 5f 72 75 6c 65 20 77 68 65 72 m site_rule wher
0300: 65 20 73 69 74 65 20 3c 40 20 74 72 69 70 64 6f e site <@ tripdo
0310: 6d 61 69 6e 28 24 31 29 20 61 6e 64 20 6e 65 74 main($1) and net
0320: 6d 61 73 6b 20 3e 3e 20 24 32 3a 3a 74 65 78 74 mask >> $2::text
0330: 3a 3a 69 6e 65 74 20 6f 72 64 65 72 20 62 79 20 ::inet order by
0340: 61 72 72 61 79 5f 6c 65 6e 67 74 68 28 73 69 74 array_length(sit
0350: 65 2c 20 31 29 20 64 65 73 63 22 29 0a 0a 09 64 e, 1) desc")...d
0360: 65 66 20 5f 63 75 72 73 28 73 65 6c 66 29 3a 0a ef _curs(self):.
0370: 09 09 69 66 20 6e 6f 74 20 73 65 6c 66 2e 5f 64 ..if not self._d
0380: 62 3a 0a 09 09 09 63 6f 6e 66 69 67 2e 73 65 63 b:....config.sec
0390: 74 69 6f 6e 28 27 64 61 74 61 62 61 73 65 27 29 tion('database')
03a0: 0a 09 09 09 73 65 6c 66 2e 5f 64 62 20 3d 20 70 ....self._db = p
03b0: 6f 73 74 67 72 65 73 71 6c 2e 6f 70 65 6e 28 0a ostgresql.open(.
03c0: 09 09 09 09 27 70 71 3a 2f 2f 7b 7d 3a 7b 7d 40 ....'pq://{}:{}@
03d0: 7b 7d 2f 7b 7d 27 2e 66 6f 72 6d 61 74 28 0a 09 {}/{}'.format(..
03e0: 09 09 09 09 63 6f 6e 66 69 67 5b 27 75 73 65 72 ....config['user
03f0: 27 5d 2c 0a 09 09 09 09 09 63 6f 6e 66 69 67 5b '],......config[
0400: 27 70 61 73 73 77 6f 72 64 27 5d 2c 0a 09 09 09 'password'],....
0410: 09 09 63 6f 6e 66 69 67 5b 27 68 6f 73 74 27 5d ..config['host']
0420: 2c 0a 09 09 09 09 09 63 6f 6e 66 69 67 5b 27 64 ,......config['d
0430: 61 74 61 62 61 73 65 27 5d 2c 0a 09 09 09 29 20 atabase'],....)
0440: 29 0a 09 09 72 65 74 75 72 6e 28 73 65 6c 66 2e )...return(self.
0450: 5f 64 62 29 0a 0a 09 64 65 66 20 63 68 65 63 6b _db)...def check
0460: 28 73 65 6c 66 2c 20 69 70 5f 61 64 64 72 65 73 (self, ip_addres
0470: 73 2c 20 73 69 74 65 29 3a 0a 09 09 72 65 74 75 s, site):...retu
0480: 72 6e 20 73 65 6c 66 2e 5f 63 68 65 63 6b 5f 73 rn self._check_s
0490: 74 6d 74 28 73 69 74 65 2c 20 69 70 5f 61 64 64 tmt(site, ip_add
04a0: 72 65 73 73 29 0a 0a 63 6c 61 73 73 20 43 68 65 ress)..class Che
04b0: 63 6b 65 72 54 68 72 65 61 64 3a 0a 09 5f 5f 73 ckerThread:..__s
04c0: 6c 6f 74 73 5f 5f 20 3d 20 66 72 6f 7a 65 6e 73 lots__ = frozens
04d0: 65 74 28 5b 27 5f 64 62 27 2c 20 27 5f 6c 6f 63 et(['_db', '_loc
04e0: 6b 27 2c 20 27 5f 6c 6f 63 6b 5f 71 75 65 75 65 k', '_lock_queue
04f0: 27 2c 20 27 5f 6c 6f 67 27 2c 20 27 5f 71 75 65 ', '_log', '_que
0500: 75 65 27 5d 29 0a 0a 09 64 65 66 20 5f 5f 69 6e ue'])...def __in
0510: 69 74 5f 5f 28 73 65 6c 66 2c 20 64 62 2c 20 6c it__(self, db, l
0520: 6f 67 29 3a 0a 09 09 73 65 6c 66 2e 5f 64 62 20 og):...self._db
0530: 3d 20 64 62 0a 09 09 73 65 6c 66 2e 5f 6c 6f 67 = db...self._log
0540: 20 3d 20 6c 6f 67 0a 09 09 23 20 53 70 69 6e 20 = log...# Spin
0550: 6c 6f 63 6b 2e 20 4c 6f 6f 70 20 61 63 71 75 69 lock. Loop acqui
0560: 72 65 73 20 69 74 20 6f 6e 20 73 74 61 72 74 20 res it on start
0570: 74 68 65 6e 20 72 65 6c 65 61 73 65 73 20 69 74 then releases it
0580: 20 77 68 65 6e 20 68 6f 6c 64 69 6e 67 20 71 75 when holding qu
0590: 65 75 65 0a 09 09 23 20 6c 6f 63 6b 2e 20 54 68 eue...# lock. Th
05a0: 69 73 20 77 61 79 20 74 68 65 20 74 68 72 65 61 is way the threa
05b0: 64 20 70 72 6f 63 65 65 64 73 20 77 69 74 68 6f d proceeds witho
05c0: 75 74 20 73 74 6f 70 73 20 77 68 69 6c 65 20 71 ut stops while q
05d0: 75 65 75 65 20 68 61 73 20 64 61 74 61 20 61 6e ueue has data an
05e0: 64 0a 09 09 23 20 67 65 74 73 20 73 74 61 6c 6c d...# gets stall
05f0: 65 64 20 77 68 65 6e 20 6e 6f 20 64 61 74 61 20 ed when no data
0600: 70 72 65 73 65 6e 74 2e 20 54 68 65 20 6c 6f 63 present. The loc
0610: 6b 20 69 73 20 72 65 6c 65 61 73 65 64 20 62 79 k is released by
0620: 20 71 75 65 75 65 20 77 72 69 74 65 72 0a 09 09 queue writer...
0630: 23 20 61 66 74 65 72 20 73 74 6f 72 69 6e 67 20 # after storing
0640: 73 6f 6d 65 74 68 69 6e 67 20 69 6e 74 6f 20 74 something into t
0650: 68 65 20 71 75 65 75 65 0a 09 09 73 65 6c 66 2e he queue...self.
0660: 5f 6c 6f 63 6b 20 3d 20 5f 74 68 72 65 61 64 2e _lock = _thread.
0670: 61 6c 6c 6f 63 61 74 65 5f 6c 6f 63 6b 28 29 0a allocate_lock().
0680: 09 09 73 65 6c 66 2e 5f 6c 6f 63 6b 5f 71 75 65 ..self._lock_que
0690: 75 65 20 3d 20 5f 74 68 72 65 61 64 2e 61 6c 6c ue = _thread.all
06a0: 6f 63 61 74 65 5f 6c 6f 63 6b 28 29 0a 09 09 73 ocate_lock()...s
06b0: 65 6c 66 2e 5f 6c 6f 63 6b 2e 61 63 71 75 69 72 elf._lock.acquir
06c0: 65 28 29 0a 09 09 73 65 6c 66 2e 5f 71 75 65 75 e()...self._queu
06d0: 65 20 3d 20 5b 5d 0a 09 09 5f 74 68 72 65 61 64 e = []..._thread
06e0: 2e 73 74 61 72 74 5f 6e 65 77 5f 74 68 72 65 61 .start_new_threa
06f0: 64 28 73 65 6c 66 2e 5f 73 74 61 72 74 2c 20 28 d(self._start, (
0700: 29 29 0a 0a 09 64 65 66 20 5f 73 74 61 72 74 28 ))...def _start(
0710: 73 65 6c 66 29 3a 0a 09 09 77 68 69 6c 65 20 54 self):...while T
0720: 72 75 65 3a 0a 09 09 09 73 65 6c 66 2e 5f 6c 6f rue:....self._lo
0730: 63 6b 2e 61 63 71 75 69 72 65 28 29 0a 09 09 09 ck.acquire()....
0740: 73 65 6c 66 2e 5f 6c 6f 63 6b 5f 71 75 65 75 65 self._lock_queue
0750: 2e 61 63 71 75 69 72 65 28 29 0a 09 09 09 23 20 .acquire()....#
0760: 79 65 73 20 74 68 69 73 20 73 68 6f 75 6c 64 20 yes this should
0770: 62 65 20 77 72 69 74 74 65 6e 20 74 68 69 73 20 be written this
0780: 77 61 79 2c 20 61 6e 64 20 79 65 73 2c 20 74 68 way, and yes, th
0790: 69 73 20 69 73 20 77 68 79 20 49 20 68 61 74 65 is is why I hate
07a0: 20 74 68 72 65 61 64 69 6e 67 0a 09 09 09 69 66 threading....if
07b0: 20 6c 65 6e 28 73 65 6c 66 2e 5f 71 75 65 75 65 len(self._queue
07c0: 29 20 3e 20 31 20 61 6e 64 20 73 65 6c 66 2e 5f ) > 1 and self._
07d0: 6c 6f 63 6b 2e 6c 6f 63 6b 65 64 28 29 3a 0a 09 lock.locked():..
07e0: 09 09 09 73 65 6c 66 2e 5f 6c 6f 63 6b 2e 72 65 ...self._lock.re
07f0: 6c 65 61 73 65 28 29 0a 09 09 09 72 65 71 20 3d lease()....req =
0800: 20 73 65 6c 66 2e 5f 71 75 65 75 65 2e 70 6f 70 self._queue.pop
0810: 28 30 29 0a 09 09 09 73 65 6c 66 2e 5f 6c 6f 63 (0)....self._loc
0820: 6b 5f 71 75 65 75 65 2e 72 65 6c 65 61 73 65 28 k_queue.release(
0830: 29 0a 09 09 09 73 65 6c 66 2e 5f 6c 6f 67 2e 69 )....self._log.i
0840: 6e 66 6f 28 27 74 72 79 69 6e 67 20 7b 7d 5c 6e nfo('trying {}\n
0850: 27 2e 66 6f 72 6d 61 74 28 72 65 71 5b 31 5d 29 '.format(req[1])
0860: 29 0a 09 09 09 72 65 73 75 6c 74 20 3d 20 73 65 )....result = se
0870: 6c 66 2e 5f 64 62 2e 63 68 65 63 6b 28 72 65 71 lf._db.check(req
0880: 5b 32 5d 2c 20 72 65 71 5b 31 5d 29 0a 09 09 09 [2], req[1])....
0890: 66 6f 72 20 72 6f 77 20 69 6e 20 72 65 73 75 6c for row in resul
08a0: 74 3a 0a 09 09 09 09 69 66 20 72 6f 77 20 21 3d t:.....if row !=
08b0: 20 4e 6f 6e 65 20 61 6e 64 20 72 6f 77 5b 30 5d None and row[0]
08c0: 20 21 3d 20 4e 6f 6e 65 3a 0a 09 09 09 09 09 69 != None:......i
08d0: 66 20 72 6f 77 5b 31 5d 20 21 3d 20 4e 6f 6e 65 f row[1] != None
08e0: 3a 0a 09 09 09 09 09 09 73 65 6c 66 2e 5f 6c 6f :.......self._lo
08f0: 67 2e 69 6e 66 6f 28 27 74 72 79 69 6e 67 20 72 g.info('trying r
0900: 65 67 65 78 70 20 22 7b 7d 22 20 76 65 72 73 75 egexp "{}" versu
0910: 73 20 22 7b 7d 22 5c 6e 27 2e 66 6f 72 6d 61 74 s "{}"\n'.format
0920: 28 72 6f 77 5b 31 5d 2c 20 72 65 71 5b 33 5d 29 (row[1], req[3])
0930: 29 0a 09 09 09 09 09 09 69 66 20 72 65 2e 63 6f ).......if re.co
0940: 6d 70 69 6c 65 28 72 6f 77 5b 31 5d 29 2e 6d 61 mpile(row[1]).ma
0950: 74 63 68 28 72 65 71 5b 33 5d 29 3a 0a 09 09 09 tch(req[3]):....
0960: 09 09 09 09 77 72 69 74 65 6c 69 6e 65 28 27 7b ....writeline('{
0970: 7d 20 33 30 32 3a 7b 7d 5c 6e 27 2e 66 6f 72 6d } 302:{}\n'.form
0980: 61 74 28 72 65 71 5b 30 5d 2c 20 72 6f 77 5b 30 at(req[0], row[0
0990: 5d 29 29 0a 09 09 09 09 09 09 09 62 72 65 61 6b ]))........break
09a0: 0a 09 09 09 09 09 09 65 6c 73 65 3a 0a 09 09 09 .......else:....
09b0: 09 09 09 09 63 6f 6e 74 69 6e 75 65 0a 09 09 09 ....continue....
09c0: 09 09 65 6c 73 65 3a 0a 09 09 09 09 09 09 77 72 ..else:.......wr
09d0: 69 74 65 6c 69 6e 65 28 27 7b 7d 20 33 30 32 3a iteline('{} 302:
09e0: 7b 7d 5c 6e 27 2e 66 6f 72 6d 61 74 28 72 65 71 {}\n'.format(req
09f0: 5b 30 5d 2c 20 72 6f 77 5b 30 5d 29 29 0a 09 09 [0], row[0]))...
0a00: 09 09 09 09 62 72 65 61 6b 0a 09 09 09 77 72 69 ....break....wri
0a10: 74 65 6c 69 6e 65 28 27 7b 7d 20 7b 7d 3a 2f 2f teline('{} {}://
0a20: 7b 7d 2f 7b 7d 5c 6e 27 2e 66 6f 72 6d 61 74 28 {}/{}\n'.format(
0a30: 72 65 71 5b 30 5d 2c 20 72 65 71 5b 34 5d 2c 20 req[0], req[4],
0a40: 72 65 71 5b 31 5d 2c 20 72 65 71 5b 33 5d 29 29 req[1], req[3]))
0a50: 0a 0a 09 64 65 66 20 63 68 65 63 6b 28 73 65 6c ...def check(sel
0a60: 66 2c 20 6c 69 6e 65 29 3a 0a 09 09 72 65 71 75 f, line):...requ
0a70: 65 73 74 20 3d 20 72 65 2e 63 6f 6d 70 69 6c 65 est = re.compile
0a80: 28 27 5e 28 5b 30 2d 39 5d 2b 29 5c 20 28 68 74 ('^([0-9]+)\ (ht
0a90: 74 70 7c 66 74 70 29 3a 5c 2f 5c 2f 28 5b 2d 5c tp|ftp):\/\/([-\
0aa0: 77 2e 3a 5d 2b 29 5c 2f 28 5b 5e 20 5d 2a 29 5c w.:]+)\/([^ ]*)\
0ab0: 20 28 5b 30 2d 39 2e 5d 2b 29 5c 2f 28 2d 7c 5b ([0-9.]+)\/(-|[
0ac0: 5c 77 5c 2e 5d 2b 29 5c 20 28 2d 7c 5c 77 2b 29 \w\.]+)\ (-|\w+)
0ad0: 5c 20 28 2d 7c 47 45 54 7c 48 45 41 44 7c 50 4f \ (-|GET|HEAD|PO
0ae0: 53 54 29 2e 2a 24 27 29 2e 6d 61 74 63 68 28 6c ST).*$').match(l
0af0: 69 6e 65 29 0a 09 09 69 66 20 72 65 71 75 65 73 ine)...if reques
0b00: 74 3a 0a 09 09 09 69 64 20 3d 20 72 65 71 75 65 t:....id = reque
0b10: 73 74 2e 67 72 6f 75 70 28 31 29 0a 09 09 09 70 st.group(1)....p
0b20: 72 6f 74 6f 20 3d 20 72 65 71 75 65 73 74 2e 67 roto = request.g
0b30: 72 6f 75 70 28 32 29 0a 09 09 09 73 69 74 65 20 roup(2)....site
0b40: 3d 20 72 65 71 75 65 73 74 2e 67 72 6f 75 70 28 = request.group(
0b50: 33 29 0a 09 09 09 75 72 6c 5f 70 61 74 68 20 3d 3)....url_path =
0b60: 20 72 65 71 75 65 73 74 2e 67 72 6f 75 70 28 34 request.group(4
0b70: 29 0a 09 09 09 69 70 5f 61 64 64 72 65 73 73 20 )....ip_address
0b80: 3d 20 72 65 71 75 65 73 74 2e 67 72 6f 75 70 28 = request.group(
0b90: 35 29 0a 09 09 09 73 65 6c 66 2e 5f 6c 6f 63 6b 5)....self._lock
0ba0: 5f 71 75 65 75 65 2e 61 63 71 75 69 72 65 28 29 _queue.acquire()
0bb0: 0a 09 09 09 73 65 6c 66 2e 5f 71 75 65 75 65 2e ....self._queue.
0bc0: 61 70 70 65 6e 64 28 28 69 64 2c 20 73 69 74 65 append((id, site
0bd0: 2c 20 69 70 5f 61 64 64 72 65 73 73 2c 20 75 72 , ip_address, ur
0be0: 6c 5f 70 61 74 68 2c 20 70 72 6f 74 6f 29 29 0a l_path, proto)).
0bf0: 09 09 09 69 66 20 73 65 6c 66 2e 5f 6c 6f 63 6b ...if self._lock
0c00: 2e 6c 6f 63 6b 65 64 28 29 3a 0a 09 09 09 09 73 .locked():.....s
0c10: 65 6c 66 2e 5f 6c 6f 63 6b 2e 72 65 6c 65 61 73 elf._lock.releas
0c20: 65 28 29 0a 09 09 09 73 65 6c 66 2e 5f 6c 6f 63 e()....self._loc
0c30: 6b 5f 71 75 65 75 65 2e 72 65 6c 65 61 73 65 28 k_queue.release(
0c40: 29 0a 09 09 09 73 65 6c 66 2e 5f 6c 6f 67 2e 69 )....self._log.i
0c50: 6e 66 6f 28 27 72 65 71 75 65 73 74 20 7b 7d 20 nfo('request {}
0c60: 71 75 65 75 65 64 20 28 7b 7d 29 5c 6e 27 2e 66 queued ({})\n'.f
0c70: 6f 72 6d 61 74 28 69 64 2c 20 6c 69 6e 65 29 29 ormat(id, line))
0c80: 0a 09 09 65 6c 73 65 3a 0a 09 09 09 73 65 6c 66 ...else:....self
0c90: 2e 5f 6c 6f 67 2e 69 6e 66 6f 28 27 62 61 64 20 ._log.info('bad
0ca0: 72 65 71 75 65 73 74 5c 6e 27 29 0a 09 09 09 77 request\n')....w
0cb0: 72 69 74 65 6c 69 6e 65 28 6c 69 6e 65 29 0a 0a riteline(line)..
0cc0: 64 65 66 20 77 72 69 74 65 6c 69 6e 65 28 73 74 def writeline(st
0cd0: 72 69 6e 67 29 3a 0a 09 6c 6f 67 2e 69 6e 66 6f ring):..log.info
0ce0: 28 27 73 65 6e 64 69 6e 67 3a 20 27 20 2b 20 73 ('sending: ' + s
0cf0: 74 72 69 6e 67 29 0a 09 73 79 73 2e 73 74 64 6f tring)..sys.stdo
0d00: 75 74 2e 77 72 69 74 65 28 73 74 72 69 6e 67 29 ut.write(string)
0d10: 0a 09 73 79 73 2e 73 74 64 6f 75 74 2e 66 6c 75 ..sys.stdout.flu
0d20: 73 68 28 29 0a 0a 63 6c 61 73 73 20 43 6f 6e 66 sh()..class Conf
0d30: 69 67 3a 0a 09 5f 5f 73 6c 6f 74 73 5f 5f 20 3d ig:..__slots__ =
0d40: 20 66 72 6f 7a 65 6e 73 65 74 28 5b 27 5f 63 6f frozenset(['_co
0d50: 6e 66 69 67 27 2c 20 27 5f 73 65 63 74 69 6f 6e nfig', '_section
0d60: 27 5d 29 0a 0a 09 64 65 66 20 5f 5f 69 6e 69 74 '])...def __init
0d70: 5f 5f 28 73 65 6c 66 29 3a 0a 09 09 70 61 72 73 __(self):...pars
0d80: 65 72 20 3d 20 6f 70 74 70 61 72 73 65 2e 4f 70 er = optparse.Op
0d90: 74 69 6f 6e 50 61 72 73 65 72 28 29 0a 09 09 70 tionParser()...p
0da0: 61 72 73 65 72 2e 61 64 64 5f 6f 70 74 69 6f 6e arser.add_option
0db0: 28 27 2d 63 27 2c 20 27 2d 2d 63 6f 6e 66 69 67 ('-c', '--config
0dc0: 27 2c 20 64 65 73 74 20 3d 20 27 63 6f 6e 66 69 ', dest = 'confi
0dd0: 67 27 2c 0a 09 09 09 68 65 6c 70 20 3d 20 27 63 g',....help = 'c
0de0: 6f 6e 66 69 67 20 66 69 6c 65 20 6c 6f 63 61 74 onfig file locat
0df0: 69 6f 6e 27 2c 20 6d 65 74 61 76 61 72 20 3d 20 ion', metavar =
0e00: 27 46 49 4c 45 27 2c 0a 09 09 09 64 65 66 61 75 'FILE',....defau
0e10: 6c 74 20 3d 20 27 2f 75 73 72 2f 6c 6f 63 61 6c lt = '/usr/local
0e20: 2f 65 74 63 2f 73 71 75 69 64 2d 74 61 67 67 65 /etc/squid-tagge
0e30: 72 2e 63 6f 6e 66 27 29 0a 0a 09 09 28 6f 70 74 r.conf')....(opt
0e40: 69 6f 6e 73 2c 20 61 72 67 73 29 20 3d 20 70 61 ions, args) = pa
0e50: 72 73 65 72 2e 70 61 72 73 65 5f 61 72 67 73 28 rser.parse_args(
0e60: 29 0a 0a 09 09 69 66 20 6e 6f 74 20 6f 73 2e 61 )....if not os.a
0e70: 63 63 65 73 73 28 6f 70 74 69 6f 6e 73 2e 63 6f ccess(options.co
0e80: 6e 66 69 67 2c 20 6f 73 2e 52 5f 4f 4b 29 3a 0a nfig, os.R_OK):.
0e90: 09 09 09 70 72 69 6e 74 28 22 43 61 6e 27 74 20 ...print("Can't
0ea0: 72 65 61 64 20 7b 7d 3a 20 65 78 69 74 74 69 6e read {}: exittin
0eb0: 67 22 2e 66 6f 72 6d 61 74 28 6f 70 74 69 6f 6e g".format(option
0ec0: 73 2e 63 6f 6e 66 69 67 29 29 0a 09 09 09 73 79 s.config))....sy
0ed0: 73 2e 65 78 69 74 28 32 29 0a 0a 09 09 73 65 6c s.exit(2)....sel
0ee0: 66 2e 5f 63 6f 6e 66 69 67 20 3d 20 63 6f 6e 66 f._config = conf
0ef0: 69 67 70 61 72 73 65 72 2e 43 6f 6e 66 69 67 50 igparser.ConfigP
0f00: 61 72 73 65 72 28 29 0a 09 09 73 65 6c 66 2e 5f arser()...self._
0f10: 63 6f 6e 66 69 67 2e 72 65 61 64 66 70 28 6f 70 config.readfp(op
0f20: 65 6e 28 6f 70 74 69 6f 6e 73 2e 63 6f 6e 66 69 en(options.confi
0f30: 67 29 29 0a 0a 09 64 65 66 20 73 65 63 74 69 6f g))...def sectio
0f40: 6e 28 73 65 6c 66 2c 20 73 65 63 74 69 6f 6e 29 n(self, section)
0f50: 3a 0a 09 09 73 65 6c 66 2e 5f 73 65 63 74 69 6f :...self._sectio
0f60: 6e 20 3d 20 73 65 63 74 69 6f 6e 0a 0a 09 64 65 n = section...de
0f70: 66 20 5f 5f 67 65 74 69 74 65 6d 5f 5f 28 73 65 f __getitem__(se
0f80: 6c 66 2c 20 6e 61 6d 65 29 3a 0a 09 09 72 65 74 lf, name):...ret
0f90: 75 72 6e 20 73 65 6c 66 2e 5f 63 6f 6e 66 69 67 urn self._config
0fa0: 2e 67 65 74 28 73 65 6c 66 2e 5f 73 65 63 74 69 .get(self._secti
0fb0: 6f 6e 2c 20 6e 61 6d 65 29 0a 0a 63 6f 6e 66 69 on, name)..confi
0fc0: 67 20 3d 20 43 6f 6e 66 69 67 28 29 0a 0a 6c 6f g = Config()..lo
0fd0: 67 20 3d 20 4c 6f 67 67 65 72 28 29 0a 64 62 20 g = Logger().db
0fe0: 3d 20 74 61 67 44 42 28 29 0a 63 68 65 63 6b 65 = tagDB().checke
0ff0: 72 20 3d 20 43 68 65 63 6b 65 72 54 68 72 65 61 r = CheckerThrea
1000: 64 28 64 62 2c 6c 6f 67 29 0a 0a 77 68 69 6c 65 d(db,log)..while
1010: 20 54 72 75 65 3a 0a 09 6c 69 6e 65 20 3d 20 73 True:..line = s
1020: 79 73 2e 73 74 64 69 6e 2e 72 65 61 64 6c 69 6e ys.stdin.readlin
1030: 65 28 29 0a 09 69 66 20 6c 65 6e 28 6c 69 6e 65 e()..if len(line
1040: 29 20 3d 3d 20 30 3a 0a 09 09 62 72 65 61 6b 0a ) == 0:...break.
1050: 09 63 68 65 63 6b 65 72 2e 63 68 65 63 6b 28 6c .checker.check(l
1060: 69 6e 65 29 0a ine).