13 if isinstance(arg, str): return lambda fn: reg(fn, arg)
14 else: return reg(arg, arg.__name__.replace('_', '-'))
18 class FieldElt (object):
19 def __init__(me, k, n):
21 me.v = (C.MP(n)%k.p).storel(k.len)
22 def __str__(me): return hex(me.v)
24 def n(me): return C.MP.loadl(me.v)
25 def __pos__(me): return FieldElt(me.k, me.n)
26 def __neg__(me): return FieldElt(me.k, -me.n)
27 def __nonzero__(me): return me.n != 0
28 def __add__(me, you): return FieldElt(me.k, me.n + me.k(you).n)
29 def __radd__(me, you): return FieldElt(me.k, me.k(you).n + me.n)
30 def __sub__(me, you): return FieldElt(me.k, me.n - me.k(you).n)
31 def __rsub__(me, you): return FieldElt(me.k, me.k(you).n - me.n)
32 def __mul__(me, you): return FieldElt(me.k, me.n*me.k(you).n)
33 def __rmul__(me, you): return FieldElt(me.k, me.k(you).n*me.n)
34 def __div__(me, you): return FieldElt(me.k, me.n*me.k(you).inv().n)
35 def __rdiv__(me, you): return FieldElt(me.k, me.k(you).n*me.inv().n)
36 def inv(me): return FieldElt(me.k, me.k.p.modinv(me.n))
37 def sqrt(me): return FieldElt(me.k, me.k.p.modsqrt(me.n))
41 me.v = C.rand.block(k. len)
45 def __init__(me, p, len = None):
47 me.len = len is None and me.p.noctets or len
49 def register(cls, name, *args, **kw):
50 FIELDS[name] = cls(*args, **kw)
51 def rand(me): return FieldElt.rand(me)
53 if isinstance(n, FieldElt):
57 return FieldElt(me, n)
59 Field.register('f25519', C.MP(0).setbit(255) - 19)
60 Field.register('fgoldi', C.MP(0).setbit(448) - C.MP(0).setbit(224) - 1)
63 x = k.rand(); y = k.rand()
64 print ' %s\n %s\n %s;' % (x, y, op(x, y))
68 print ' %s\n %s;' % (x, op(x))
71 def add(k): binop(k, lambda x, y: x + y)
74 def sub(k): binop(k, lambda x, y: x - y)
77 def neg(k): unop(k, lambda x: -x)
80 def mul(k): binop(k, lambda x, y: x*y)
83 def sqr(k): unop(k, lambda x: x*x)
86 def inv(k): unop(k, lambda x: x and x.inv() or k(0))
90 x = k.rand(); y = k.rand()
92 u = yy and x/y or k(0)
94 except ValueError: print ' %s\n %s\n "" "";' % (x, y)
95 else: print ' %s\n %s\n %s\n %s;' % (x, y, zz, -zz)
100 a = C.rand.range(1 << 20) - (1 << 19)
101 print ' %s %d\n %s;' % (x, a, a*x)
103 def mask(): return C.rand.range(2)*0xffffffff
107 x = k.rand(); y = k.rand(); m = mask()
108 print ' %s\n %s\n 0x%08x\n %s;' % (x, y, m, +(m and x or y))
112 x = k.rand(); m = mask()
113 print ' %s\n 0x%08x\n %s;' % (x, m, x*(m and -1 or +1))
117 n = C.rand.range(31) + 1
118 v = [k.rand() for i in xrange(n)]
120 print ' "%s"\n %d\n %s;' % ('\n '.join(map(str, v)), i, +v[i])
124 x = k.rand(); y = k.rand(); m = mask()
125 xx, yy = m and (+y, +x) or (+x, +y)
126 print ' %s\n %s\n 0x%08x\n %s\n %s;' % (x, y, m, xx, yy)
129 def sub_mulc_add_sub_mul(k):
130 u = k.rand(); v = k.rand(); w = k.rand(); x = k.rand(); y = k.rand();
131 a = C.rand.range(1 << 20) - (1 << 19)
132 print ' %s\n %s %d\n %s\n %s\n %s\n %s;' % \
133 (u, v, a, w, x, y, ((u - v)*a + w)*(x - y))
138 for i in xrange(NTEST): TESTS[t](k)