chiark / gitweb /
changelog: start 1.3.3
[chiark-tcl.git] / hbytes / hook.c
1 /*
2  * hbytes - hex-stringrep efficient byteblocks for Tcl
3  * Copyright 2006-2012 Ian Jackson
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License as
7  * published by the Free Software Foundation; either version 2 of the
8  * License, or (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this library; if not, see <http://www.gnu.org/licenses/>.
17  */
18
19
20 #include <errno.h>
21
22 #include "chiark_tcl_hbytes.h"
23
24 int cht_do_hbytes_rep_info(ClientData cd, Tcl_Interp *ip,
25                        Tcl_Obj *obj, Tcl_Obj **result) {
26   const char *tn;
27   int nums[3], i, lnl;
28   Tcl_Obj *objl[4];
29
30   if (obj->typePtr == &cht_hbytes_type) {
31     HBytes_Value *v= OBJ_HBYTES(obj);
32     memset(nums,0,sizeof(nums));
33     nums[1]= cht_hb_len(v);
34   
35     if (HBYTES_ISEMPTY(v)) tn= "empty";
36     else if (HBYTES_ISSENTINEL(v)) tn= "sentinel!";
37     else if (HBYTES_ISSIMPLE(v)) tn= "simple";
38     else {
39       HBytes_ComplexValue *cx= v->begin_complex;
40       tn= "complex";
41       nums[0]= cx->prespace;
42       nums[2]= cx->avail - cx->len;
43     }
44     lnl= 3;
45   } else {
46     tn= "other";
47     lnl= 0;
48   }
49     
50   objl[0]= Tcl_NewStringObj((char*)tn,-1);
51   for (i=0; i<lnl; i++) objl[i+1]= Tcl_NewIntObj(nums[i]);
52   *result= Tcl_NewListObj(lnl+1,objl);
53     
54   return TCL_OK;
55 }
56
57 static void hbytes_t_dup(Tcl_Obj *src, Tcl_Obj *dup) {
58   cht_hb_array(OBJ_HBYTES(dup),
59                cht_hb_data(OBJ_HBYTES(src)),
60                cht_hb_len(OBJ_HBYTES(src)));
61   dup->typePtr= &cht_hbytes_type;
62 }
63
64 static void hbytes_t_free(Tcl_Obj *o) {
65   cht_hb_free(OBJ_HBYTES(o));
66 }
67
68 void cht_obj_updatestr_array_prefix(Tcl_Obj *o, const Byte *byte,
69                                 int l, const char *prefix) {
70   char *str;
71   int pl;
72
73   pl= strlen(prefix);
74   assert(l < INT_MAX/2 - 1 - pl);
75   o->length= l*2+pl;
76   str= o->bytes= TALLOC(o->length+1);
77   
78   memcpy(str,prefix,pl);
79   str += pl;
80
81   while (l>0) {
82     sprintf(str,"%02x",*byte);
83     str+=2; byte++; l--;
84   }
85   *str= 0;
86 }
87
88 void cht_obj_updatestr_array(Tcl_Obj *o, const Byte *byte, int l) {
89   cht_obj_updatestr_array_prefix(o,byte,l,"");
90 }
91
92 static void hbytes_t_ustr(Tcl_Obj *o) {
93   cht_obj_updatestr_array(o,
94                       cht_hb_data(OBJ_HBYTES(o)),
95                       cht_hb_len(OBJ_HBYTES(o)));
96 }
97
98 static int hbytes_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
99   char *str, *ep;
100   Byte *bytes;
101   int l;
102   char cbuf[3];
103
104   if (o->typePtr == &cht_ulong_type) {
105     uint32_t ul;
106
107     ul= htonl(*(const uint32_t*)&o->internalRep.longValue);
108     cht_hb_array(OBJ_HBYTES(o), (const Byte*)&ul, 4);
109
110   } else {
111   
112     str= Tcl_GetStringFromObj(o,&l);  assert(str);
113     cht_objfreeir(o);
114
115     if (l & 1) return cht_staticerr(ip, "hbytes: conversion from hex:"
116                                 " odd length in hex", "HBYTES SYNTAX");
117
118     bytes= cht_hb_arrayspace(OBJ_HBYTES(o), l/2);
119
120     cbuf[2]= 0;
121     while (l>0) {
122       cbuf[0]= *str++;
123       cbuf[1]= *str++;
124       *bytes++= strtoul(cbuf,&ep,16);
125       if (ep != cbuf+2) {
126         cht_hb_free(OBJ_HBYTES(o));
127         return cht_staticerr(ip, "hbytes: conversion from hex:"
128                          " bad hex digit", "HBYTES SYNTAX");
129       }
130       l -= 2;
131     }
132
133   }
134
135   o->typePtr = &cht_hbytes_type;
136   return TCL_OK;
137 }
138
139 Tcl_ObjType cht_hbytes_type = {
140   "hbytes",
141   hbytes_t_free, hbytes_t_dup, hbytes_t_ustr, hbytes_t_sfa
142 };
143
144 int cht_do_hbytes_raw2h(ClientData cd, Tcl_Interp *ip,
145                     Tcl_Obj *binary, HBytes_Value *result) {
146   const unsigned char *str;
147   int l;
148
149   str= Tcl_GetByteArrayFromObj(binary,&l);
150   cht_hb_array(result, str, l);
151   return TCL_OK;
152 }
153
154 int cht_do_hbytes_h2raw(ClientData cd, Tcl_Interp *ip,
155                     HBytes_Value hex, Tcl_Obj **result) {
156   *result= Tcl_NewByteArrayObj(cht_hb_data(&hex), cht_hb_len(&hex));
157   return TCL_OK;
158 }
159
160 int cht_do_hbytes_length(ClientData cd, Tcl_Interp *ip,
161                      HBytes_Value v, int *result) {
162   *result= cht_hb_len(&v);
163   return TCL_OK;
164 }
165
166 int cht_do_hbytes_random(ClientData cd, Tcl_Interp *ip,
167                      int length, HBytes_Value *result) {
168   Byte *space;
169   int rc;
170   
171   space= cht_hb_arrayspace(result, length);
172   rc= cht_get_urandom(ip, space, length);
173   if (rc) { cht_hb_free(result); return rc; }
174   return TCL_OK;
175 }  
176   
177 int cht_do_hbytes_overwrite(ClientData cd, Tcl_Interp *ip,
178                         HBytes_Var v, int start, HBytes_Value sub) {
179   int sub_l;
180
181   sub_l= cht_hb_len(&sub);
182   if (start < 0)
183     return cht_staticerr(ip, "hbytes overwrite start -ve",
184                      "HBYTES LENGTH RANGE");
185   if (start + sub_l > cht_hb_len(v.hb))
186     return cht_staticerr(ip, "hbytes overwrite out of range",
187                      "HBYTES LENGTH UNDERRUN");
188   memcpy(cht_hb_data(v.hb) + start, cht_hb_data(&sub), sub_l);
189   return TCL_OK;
190 }
191
192 int cht_do_hbytes_trimleft(ClientData cd, Tcl_Interp *ip, HBytes_Var v) {
193   const Byte *o, *p, *e;
194   o= p= cht_hb_data(v.hb);
195   e= p + cht_hb_len(v.hb);
196
197   while (p<e && !*p) p++;
198   if (p != o)
199     cht_hb_unprepend(v.hb, p-o);
200
201   return TCL_OK;
202 }
203
204 int cht_do_hbytes_repeat(ClientData cd, Tcl_Interp *ip,
205                      HBytes_Value sub, int count, HBytes_Value *result) {
206   int sub_l;
207   Byte *data;
208   const Byte *sub_d;
209
210   sub_l= cht_hb_len(&sub);
211   if (count < 0) return cht_staticerr(ip, "hbytes repeat count -ve",
212                                   "HBYTES LENGTH RANGE");
213   if (count > INT_MAX/sub_l) return cht_staticerr(ip, "hbytes repeat too long", 0);
214
215   data= cht_hb_arrayspace(result, sub_l*count);
216   sub_d= cht_hb_data(&sub);
217   while (count) {
218     memcpy(data, sub_d, sub_l);
219     count--; data += sub_l;
220   }
221   return TCL_OK;
222 }  
223
224 int cht_do_hbytes_xor(ClientData cd, Tcl_Interp *ip,
225                   HBytes_Var v, HBytes_Value d) {
226   int l;
227   Byte *dest;
228   const Byte *source;
229
230   l= cht_hb_len(v.hb);
231   if (cht_hb_len(&d) != l) return
232     cht_staticerr(ip, "hbytes xor lengths do not match", "HBYTES LENGTH MISMATCH");
233
234   dest= cht_hb_data(v.hb);
235   source= cht_hb_data(&d);
236   memxor(dest,source,l);
237   return TCL_OK;
238 }
239   
240 int cht_do_hbytes_zeroes(ClientData cd, Tcl_Interp *ip,
241                      int length, HBytes_Value *result) {
242   Byte *space;
243   space= cht_hb_arrayspace(result, length);
244   memset(space,0,length);
245   return TCL_OK;
246 }
247
248 int cht_do_hbytes_compare(ClientData cd, Tcl_Interp *ip,
249                       HBytes_Value a, HBytes_Value b, int *result) {
250   int al, bl, minl, r;
251
252   al= cht_hb_len(&a);
253   bl= cht_hb_len(&b);
254   minl= al<bl ? al : bl;
255
256   r= memcmp(cht_hb_data(&a), cht_hb_data(&b), minl);
257   
258   if (r<0) *result= -2;
259   else if (r>0) *result= +2;
260   else {
261     if (al<bl) *result= -1;
262     else if (al>bl) *result= +1;
263     else *result= 0;
264   }
265   return TCL_OK;
266 }
267
268 int cht_do_hbytes_range(ClientData cd, Tcl_Interp *ip,
269                     HBytes_Value v, int start, int size,
270                     HBytes_Value *result) {
271   const Byte *data;
272   int l;
273
274   l= cht_hb_len(&v);
275   if (start<0 || size<0)
276     return cht_staticerr(ip,"hbytes range subscript(s) -ve","HBYTES LENGTH RANGE");
277   if (l<start+size)
278     return cht_staticerr(ip, "hbytes range subscripts too big",
279                      "HBYTES LENGTH UNDERRUN");
280
281   data= cht_hb_data(&v);
282   cht_hb_array(result, data+start, size);
283   return TCL_OK;
284 }
285
286 /* hbytes representing uint16_t's */
287
288 int cht_do_hbytes_h2ushort(ClientData cd, Tcl_Interp *ip,
289                        HBytes_Value hex, long *result) {
290   const Byte *data;
291   int l;
292
293   l= cht_hb_len(&hex);
294   if (l>2)
295     return cht_staticerr(ip, "hbytes h2ushort input more than 4 hex digits",
296                      "HBYTES VALUE OVERFLOW");
297
298   data= cht_hb_data(&hex);
299   *result= data[l-1] | (l>1 ? data[0]<<8 : 0);
300   return TCL_OK;
301 }
302
303 int cht_do_hbytes_ushort2h(ClientData cd, Tcl_Interp *ip,
304                        long input, HBytes_Value *result) {
305   uint16_t us;
306
307   if (input > 0x0ffff)
308     return cht_staticerr(ip, "hbytes ushort2h input >2^16",
309                      "HBYTES VALUE OVERFLOW");
310
311   us= htons(input);
312   cht_hb_array(result,(const Byte*)&us,2);
313   return TCL_OK;
314 }
315
316 /* toplevel functions */
317
318 CHT_INIT(hbytes,
319          CHTI_TYPE(cht_hbytes_type) CHTI_TYPE(cht_ulong_type),
320          CHTI_COMMANDS(cht_hbytestoplevel_entries))