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)
62 x = k.rand(); y = k.rand()
63 print ' %s\n %s\n %s;' % (x, y, op(x, y))
67 print ' %s\n %s;' % (x, op(x))
70 def add(k): binop(k, lambda x, y: x + y)
73 def sub(k): binop(k, lambda x, y: x - y)
76 def neg(k): unop(k, lambda x: -x)
79 def mul(k): binop(k, lambda x, y: x*y)
82 def sqr(k): unop(k, lambda x: x*x)
85 def inv(k): unop(k, lambda x: x and x.inv() or k(0))
89 x = k.rand(); y = k.rand()
91 u = yy and x/y or k(0)
93 except ValueError: print ' %s\n %s\n "" "";' % (x, y)
94 else: print ' %s\n %s\n %s\n %s;' % (x, y, zz, -zz)
99 a = C.rand.range(1 << 20) - (1 << 19)
100 print ' %s %d\n %s;' % (x, a, a*x)
102 def mask(): return C.rand.range(2)*0xffffffff
106 x = k.rand(); y = k.rand(); m = mask()
107 print ' %s\n %s\n 0x%08x\n %s;' % (x, y, m, +(m and x or y))
111 x = k.rand(); m = mask()
112 print ' %s\n 0x%08x\n %s;' % (x, m, x*(m and -1 or +1))
116 n = C.rand.range(31) + 1
117 v = [k.rand() for i in xrange(n)]
119 print ' "%s"\n %d\n %s;' % ('\n '.join(map(str, v)), i, +v[i])
123 x = k.rand(); y = k.rand(); m = mask()
124 xx, yy = m and (+y, +x) or (+x, +y)
125 print ' %s\n %s\n 0x%08x\n %s\n %s;' % (x, y, m, xx, yy)
128 def sub_mulc_add_sub_mul(k):
129 u = k.rand(); v = k.rand(); w = k.rand(); x = k.rand(); y = k.rand();
130 a = C.rand.range(1 << 20) - (1 << 19)
131 print ' %s\n %s %d\n %s\n %s\n %s\n %s;' % \
132 (u, v, a, w, x, y, ((u - v)*a + w)*(x - y))
137 for i in xrange(NTEST): TESTS[t](k)