chiark / gitweb /
Drop alg->byteswap; implement counter mode; split padding so now we do PKCS#5 and...
[chiark-tcl.git] / base / hook.c
1 /*
2  */
3
4 #include <errno.h>
5
6 #include "hbytes.h"
7 #include "tables.h"
8
9 int staticerr(Tcl_Interp *ip, const char *m, const char *ec) {
10   Tcl_SetResult(ip, (char*)m, TCL_STATIC);
11   if (ec) Tcl_SetObjErrorCode(ip, Tcl_NewStringObj(ec,-1));
12   return TCL_ERROR;
13 }
14
15 int posixerr(Tcl_Interp *ip, int errnoval, const char *m) {
16   const char *em;
17   
18   Tcl_ResetResult(ip);
19   errno= errnoval;
20   em= Tcl_PosixError(ip);
21   Tcl_AppendResult(ip, m, ": ", em, (char*)0);
22   return TCL_ERROR;
23 }
24
25 void objfreeir(Tcl_Obj *o) {
26   if (o->typePtr && o->typePtr->freeIntRepProc)
27     o->typePtr->freeIntRepProc(o);
28   o->typePtr= 0;
29 }  
30
31 int do_hbytes_rep_info(ClientData cd, Tcl_Interp *ip,
32                        Tcl_Obj *obj, Tcl_Obj **result) {
33   const char *tn;
34   int nums[3], i, lnl;
35   Tcl_Obj *objl[4];
36
37   if (obj->typePtr == &hbytes_type) {
38     HBytes_Value *v= OBJ_HBYTES(obj);
39     memset(nums,0,sizeof(nums));
40     nums[1]= hbytes_len(v);
41   
42     if (HBYTES_ISEMPTY(v)) tn= "empty";
43     else if (HBYTES_ISSENTINEL(v)) tn= "sentinel!";
44     else if (HBYTES_ISSIMPLE(v)) tn= "simple";
45     else {
46       HBytes_ComplexValue *cx= v->begin_complex;
47       tn= "complex";
48       nums[0]= cx->prespace;
49       nums[2]= cx->avail - cx->len;
50     }
51     lnl= 3;
52   } else {
53     tn= "other";
54     lnl= 0;
55   }
56     
57   objl[0]= Tcl_NewStringObj((char*)tn,-1);
58   for (i=0; i<lnl; i++) objl[i+1]= Tcl_NewIntObj(nums[i]);
59   *result= Tcl_NewListObj(lnl+1,objl);
60     
61   return TCL_OK;
62 }
63
64 static void hbytes_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
65   hbytes_array(OBJ_HBYTES(dup),
66                hbytes_data(OBJ_HBYTES(src)),
67                hbytes_len(OBJ_HBYTES(src)));
68   dup->typePtr= &hbytes_type;
69 }
70
71 static void hbytes_t_free(Tcl_Obj *o) {
72   hbytes_free(OBJ_HBYTES(o));
73 }
74
75 void obj_updatestr_array_prefix(Tcl_Obj *o, const Byte *byte,
76                                 int l, const char *prefix) {
77   char *str;
78   int pl;
79
80   pl= strlen(prefix);
81   o->length= l*2+pl;
82   str= o->bytes= TALLOC(o->length+1);
83   
84   memcpy(str,prefix,pl);
85   str += pl;
86
87   while (l>0) {
88     sprintf(str,"%02x",*byte);
89     str+=2; byte++; l--;
90   }
91   *str= 0;
92 }
93
94 void obj_updatestr_array(Tcl_Obj *o, const Byte *byte, int l) {
95   obj_updatestr_array_prefix(o,byte,l,"");
96 }
97
98 void obj_updatestr_vstringls(Tcl_Obj *o, ...) {
99   va_list al;
100   char *p;
101   const char *part;
102   int l, pl;
103
104   va_start(al,o);
105   for (l=0; (part= va_arg(al, const char*)); )
106     l+= va_arg(al, int);
107   va_end(al);
108   
109   o->length= l;
110   o->bytes= TALLOC(l+1);
111
112   va_start(al,o);
113   for (p= o->bytes; (part= va_arg(al, const char*)); p += pl) {
114     pl= va_arg(al, int);
115     memcpy(p, part, pl);
116   }
117   va_end(al);
118
119   *p= 0;
120 }
121
122 void obj_updatestr_string(Tcl_Obj *o, const char *str) {
123   obj_updatestr_vstringls(o, str, strlen(str), (char*)0);
124 }
125
126 static void hbytes_t_ustr(Tcl_Obj *o) {
127   obj_updatestr_array(o,
128                       hbytes_data(OBJ_HBYTES(o)),
129                       hbytes_len(OBJ_HBYTES(o)));
130 }
131
132 static int hbytes_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
133   char *str, *ep, *os;
134   Byte *startbytes, *bytes;
135   int l;
136   char cbuf[3];
137
138   if (o->typePtr == &ulong_type) {
139     uint32_t ul;
140
141     ul= htonl(*(const uint32_t*)&o->internalRep.longValue);
142     hbytes_array(OBJ_HBYTES(o), (const Byte*)&ul, 4);
143
144   } else {
145   
146     os= str= Tcl_GetStringFromObj(o,&l);  assert(str);
147     objfreeir(o);
148
149     if (l & 1) return staticerr(ip, "hbytes: conversion from hex:"
150                                 " odd length in hex", "HBYTES SYNTAX");
151
152     startbytes= bytes= hbytes_arrayspace(OBJ_HBYTES(o), l/2);
153
154     cbuf[2]= 0;
155     while (l>0) {
156       cbuf[0]= *str++;
157       cbuf[1]= *str++;
158       *bytes++= strtoul(cbuf,&ep,16);
159       if (ep != cbuf+2) {
160         hbytes_free(OBJ_HBYTES(o));
161         return staticerr(ip, "hbytes: conversion from hex:"
162                          " bad hex digit", "HBYTES SYNTAX");
163       }
164       l -= 2;
165     }
166
167   }
168
169   o->typePtr = &hbytes_type;
170   return TCL_OK;
171 }
172
173 Tcl_ObjType hbytes_type = {
174   "hbytes",
175   hbytes_t_free, hbytes_t_dup, hbytes_t_ustr, hbytes_t_sfa
176 };
177
178 int do_hbytes_raw2h(ClientData cd, Tcl_Interp *ip,
179                     Tcl_Obj *binary, HBytes_Value *result) {
180   const unsigned char *str;
181   int l;
182
183   str= Tcl_GetByteArrayFromObj(binary,&l);
184   hbytes_array(result, str, l);
185   return TCL_OK;
186 }
187
188 int do_hbytes_h2raw(ClientData cd, Tcl_Interp *ip,
189                     HBytes_Value hex, Tcl_Obj **result) {
190   *result= Tcl_NewByteArrayObj(hbytes_data(&hex), hbytes_len(&hex));
191   return TCL_OK;
192 }
193
194 int do_hbytes_length(ClientData cd, Tcl_Interp *ip,
195                      HBytes_Value v, int *result) {
196   *result= hbytes_len(&v);
197   return TCL_OK;
198 }
199
200 int do_hbytes_random(ClientData cd, Tcl_Interp *ip,
201                      int length, HBytes_Value *result) {
202   Byte *space;
203   int rc;
204   
205   space= hbytes_arrayspace(result, length);
206   rc= get_urandom(ip, space, length);
207   if (rc) { hbytes_free(result); return rc; }
208   return TCL_OK;
209 }  
210   
211 int do_hbytes_overwrite(ClientData cd, Tcl_Interp *ip,
212                         HBytes_Var v, int start, HBytes_Value sub) {
213   int sub_l;
214
215   sub_l= hbytes_len(&sub);
216   if (start < 0)
217     return staticerr(ip, "hbytes overwrite start -ve",
218                      "HBYTES LENGTH RANGE");
219   if (start + sub_l > hbytes_len(v.hb))
220     return staticerr(ip, "hbytes overwrite out of range",
221                      "HBYTES LENGTH UNDERRUN");
222   memcpy(hbytes_data(v.hb) + start, hbytes_data(&sub), sub_l);
223   return TCL_OK;
224 }
225
226 int do_hbytes_trimleft(ClientData cd, Tcl_Interp *ip, HBytes_Var v) {
227   const Byte *o, *p, *e;
228   o= p= hbytes_data(v.hb);
229   e= p + hbytes_len(v.hb);
230
231   while (p<e && !*p) p++;
232   if (p != o)
233     hbytes_unprepend(v.hb, p-o);
234
235   return TCL_OK;
236 }
237
238 int do_hbytes_repeat(ClientData cd, Tcl_Interp *ip,
239                      HBytes_Value sub, int count, HBytes_Value *result) {
240   int sub_l;
241   Byte *data;
242   const Byte *sub_d;
243
244   sub_l= hbytes_len(&sub);
245   if (count < 0) return staticerr(ip, "hbytes repeat count -ve",
246                                   "HBYTES LENGTH RANGE");
247   if (count > INT_MAX/sub_l) return staticerr(ip, "hbytes repeat too long", 0);
248
249   data= hbytes_arrayspace(result, sub_l*count);
250   sub_d= hbytes_data(&sub);
251   while (count) {
252     memcpy(data, sub_d, sub_l);
253     count--; data += sub_l;
254   }
255   return TCL_OK;
256 }  
257
258 int do_hbytes_zeroes(ClientData cd, Tcl_Interp *ip,
259                      int length, HBytes_Value *result) {
260   Byte *space;
261   space= hbytes_arrayspace(result, length);
262   memset(space,0,length);
263   return TCL_OK;
264 }
265
266 int do_hbytes_compare(ClientData cd, Tcl_Interp *ip,
267                       HBytes_Value a, HBytes_Value b, int *result) {
268   int al, bl, minl, r;
269
270   al= hbytes_len(&a);
271   bl= hbytes_len(&b);
272   minl= al<bl ? al : bl;
273
274   r= memcmp(hbytes_data(&a), hbytes_data(&b), minl);
275   
276   if (r<0) *result= -2;
277   else if (r>0) *result= +2;
278   else {
279     if (al<bl) *result= -1;
280     else if (al>bl) *result= +1;
281     else *result= 0;
282   }
283   return TCL_OK;
284 }
285
286 int do_hbytes_range(ClientData cd, Tcl_Interp *ip,
287                     HBytes_Value v, int start, int size,
288                     HBytes_Value *result) {
289   const Byte *data;
290   int l;
291
292   l= hbytes_len(&v);
293   if (start<0 || size<0)
294     return staticerr(ip,"hbytes range subscript(s) -ve","HBYTES LENGTH RANGE");
295   if (l<start+size)
296     return staticerr(ip, "hbytes range subscripts too big",
297                      "HBYTES LENGTH UNDERRUN");
298
299   data= hbytes_data(&v);
300   hbytes_array(result, data+start, size);
301   return TCL_OK;
302 }
303
304 /* hbytes representing uint16_t's */
305
306 int do_hbytes_h2ushort(ClientData cd, Tcl_Interp *ip,
307                        HBytes_Value hex, long *result) {
308   const Byte *data;
309   int l;
310
311   l= hbytes_len(&hex);
312   if (l>2)
313     return staticerr(ip, "hbytes h2ushort input more than 4 hex digits",
314                      "HBYTES VALUE OVERFLOW");
315
316   data= hbytes_data(&hex);
317   *result= data[l-1] | (l>1 ? data[0]<<8 : 0);
318   return TCL_OK;
319 }
320
321 int do_hbytes_ushort2h(ClientData cd, Tcl_Interp *ip,
322                        long input, HBytes_Value *result) {
323   uint16_t us;
324
325   if (input > 0x0ffff)
326     return staticerr(ip, "hbytes ushort2h input >2^16",
327                      "HBYTES VALUE OVERFLOW");
328
329   us= htons(input);
330   hbytes_array(result,(const Byte*)&us,2);
331   return TCL_OK;
332 }
333
334 /* toplevel functions */
335
336 int do_toplevel_hbytes(ClientData cd, Tcl_Interp *ip,
337                        const HBytes_SubCommand *subcmd,
338                        int objc, Tcl_Obj *const *objv) {
339   return subcmd->func(0,ip,objc,objv);
340 }
341
342 int do_toplevel_dgram_socket(ClientData cd, Tcl_Interp *ip,
343                              const DgramSocket_SubCommand *subcmd,
344                              int objc, Tcl_Obj *const *objv) {
345   return subcmd->func(0,ip,objc,objv);
346 }
347
348 int do_toplevel_ulong(ClientData cd, Tcl_Interp *ip,
349                       const ULong_SubCommand *subcmd,
350                       int objc, Tcl_Obj *const *objv) {
351   return subcmd->func(0,ip,objc,objv);
352 }
353
354 #define URANDOM "/dev/urandom"
355
356 int get_urandom(Tcl_Interp *ip, Byte *buffer, int l) {
357   static FILE *urandom;
358
359   int r, esave;
360
361   if (!urandom) {
362     urandom= fopen(URANDOM,"rb");
363     if (!urandom) return posixerr(ip,errno,"open " URANDOM);
364   }
365   r= fread(buffer,1,l,urandom);
366   if (r==l) return 0;
367
368   esave= errno;
369   fclose(urandom); urandom=0;
370
371   if (ferror(urandom)) {
372     return posixerr(ip,errno,"read " URANDOM);
373   } else {
374     assert(feof(urandom));
375     return staticerr(ip, URANDOM " gave eof!", 0);
376   }
377 }
378
379 int Hbytes_Init(Tcl_Interp *ip) {
380   const TopLevel_Command *cmd;
381
382   Tcl_RegisterObjType(&hbytes_type);
383   Tcl_RegisterObjType(&blockcipherkey_type);
384   Tcl_RegisterObjType(&enum_nearlytype);
385   Tcl_RegisterObjType(&enum1_nearlytype);
386   Tcl_RegisterObjType(&sockaddr_type);
387   Tcl_RegisterObjType(&dgramsockid_type);
388   Tcl_RegisterObjType(&ulong_type);
389
390   for (cmd=toplevel_commands;
391        cmd->name;
392        cmd++)
393     Tcl_CreateObjCommand(ip, cmd->name, cmd->func, 0,0);
394
395   return TCL_OK;
396 }