maskmap/addrmap.c

   1 /*
   2  * maskmap - Tcl extension for address mask map data structures
   3  * Copyright 2006 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, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  18  * 02110-1301, USA.
  19  */
  20
  21
  22 #include "chiark_tcl_hbytes.h"
  23
  24 /*---------- operations on AddrMap_Entry ----------*/
  25
  26 static void ame_free(AddrMap_Entry *ame) {
  27   TFREE(ame->start);  ame->start=0;
  28   if (ame->data) { Tcl_DecrRefCount(ame->data); ame->data=0; }
  29 }
  30
  31 static const Byte *ame_parsecheck_addr(Tcl_Interp *ip, const AddrMap_Value *am,
  32                                        const HBytes_Value *hb) {
  33   int hbl= cht_hb_len(hb);
  34   if (hbl < am->byl) {
  35     cht_staticerr(ip,"addr-map address too short","HBYTES ADDRMAP UNDERRUN");
  36     return 0;
  37   }
  38   if (hbl > am->byl) {
  39     cht_staticerr(ip,"addr-map address too long","HBYTES ADDRMAP OVERRUN");
  40     return 0;
  41   }
  42   return cht_hb_data(hb);
  43 }
  44
  45 static int ame_parsecheck_range(Tcl_Interp *ip, const AddrMap_Value *am,
  46                                 const HBytes_Value *starthb,
  47                                 const HBytes_Value *endhb,
  48                                 const Byte *p_r[2]) {
  49   p_r[0]= ame_parsecheck_addr(ip,am,starthb);  if (!p_r[0]) return TCL_ERROR;
  50   p_r[1]= ame_parsecheck_addr(ip,am,endhb);    if (!p_r[0]) return TCL_ERROR;
  51   if (memcmp(p_r[0],p_r[1],am->byl) > 0)
  52     return cht_staticerr(ip, "addr-map range start is after end",
  53                      "HBYTES ADDRMAP BADRANGE");
  54   return TCL_OK;
  55 }
  56
  57 static int ame_ba_addsubtractone(Byte *out, const Byte *in, int byl,
  58                                  unsigned signum, unsigned onoverflow) {
  59   /* On entry:
  60    *   *in is an array of byl bytes
  61    *   signum is 0xff or 0x01
  62    *   onoverflow is what counts as overflowed value,
  63    *     ie (for unsigned arith) 0x00 for add and 0xff for subtract
  64    * On exit:
  65    *   *out is the resulting value (subject to overflow truncation)
  66    *   return value is TCL_OK, or TCL_ERROR if overflow happened
  67    *   (but interpreter result is not set on overflow)
  68    */
  69   int j;
  70
  71   for (j= byl, in += byl, out += byl;
  72        in--, out--, j>0;
  73        j--) {
  74     *out = (*out) + signum;
  75     if (*out != onoverflow)
  76       return TCL_OK;
  77   }
  78   return TCL_ERROR;
  79 }
  80
  81 /*---------- useful operations on AddrMap_Value etc. ----------*/
  82
  83 static void am_init0(AddrMap_Value *am, int byl) {
  84   am->byl= byl;
  85   am->used= 0;
  86   am->space= 0;
  87   am->entries= 0;
  88 }
  89
  90 static void am_reallocentries(AddrMap_Value *am, int len) {
  91   AddrMap_Entry *newentries;
  92
  93   assert(len >= am->space);
  94   if (!len) return;
  95
  96   newentries= TREALLOC(am->entries, sizeof(*newentries)*len);
  97   assert(newentries);
  98
  99   am->space= len;
 100   am->entries= newentries;
 101 }
 102
 103 static void am_free(AddrMap_Value *am) {
 104   AddrMap_Entry *ame;
 105   int i;
 106
 107   if (!am) return;
 108
 109   for (i=0, ame=am->entries; i<am->used; i++, ame++)
 110     ame_free(ame);
 111
 112   TFREE(am->entries);
 113   TFREE(am);
 114 }
 115
 116 /*---------- Tcl type and arg parsing functions ----------*/
 117
 118 int cht_pat_addrmapv(Tcl_Interp *ip, Tcl_Obj *var, AddrMap_Var *agg) {
 119   int rc;
 120   rc= cht_pat_somethingv(ip,var,&agg->sth,&cht_addrmap_type);
 121   if (rc) return rc;
 122   agg->am= agg->sth.obj->internalRep.otherValuePtr;
 123   return TCL_OK;
 124 }
 125
 126 static void addrmap_t_free(Tcl_Obj *o) {
 127   AddrMap_Value *am= o->internalRep.otherValuePtr;
 128   am_free(am);
 129 }
 130
 131 static void addrmap_t_dup(Tcl_Obj *sob, Tcl_Obj *dob) {
 132   AddrMap_Value *sm= sob->internalRep.otherValuePtr;
 133   AddrMap_Value *dm;
 134   AddrMap_Entry *sme, *dme;
 135   int i;
 136
 137   assert(sob->typePtr == &cht_addrmap_type);
 138   cht_objfreeir(dob);
 139   dm= TALLOC(sizeof(*dm));
 140
 141   am_init0(dm,sm->byl);
 142   am_reallocentries(dm,sm->used);
 143   dm->used= sm->used;
 144   for (i=0, sme=sm->entries, dme=dm->entries;
 145        i < dm->used;
 146        i++, sme++, dme++) {
 147     *dme= *sme;
 148     dme->start= TALLOC(sm->byl);  assert(dme->start);
 149     memcpy(dme->start, sme->start, sm->byl);
 150     Tcl_IncrRefCount(dme->data);
 151   }
 152   dob->internalRep.otherValuePtr= dm;
 153   dob->typePtr= &cht_addrmap_type;
 154 }
 155
 156 static void addrmap_t_ustr(Tcl_Obj *so) {
 157   AddrMap_Value *sm= so->internalRep.otherValuePtr;
 158   Tcl_Obj **mainlobjsl, *surrogate;
 159   AddrMap_Entry *sme;
 160   int entnum, listlength;
 161
 162   assert(so->typePtr == &cht_addrmap_type);
 163   mainlobjsl= TALLOC(sizeof(*mainlobjsl) * (sm->used+1));  assert(mainlobjsl);
 164   mainlobjsl[0]= Tcl_NewIntObj(sm->byl * 8);
 165   listlength= 1;
 166
 167   for (entnum=0, sme=sm->entries; entnum<sm->used; entnum++, sme++) {
 168     HBytes_Value hb;
 169     Tcl_Obj *subl[3], *sublo;
 170
 171     if (!sme->data) continue;
 172
 173     cht_hb_array(&hb, sme->start, sm->byl);
 174     subl[0]= cht_ret_hb(0, hb);  assert(subl[0]);
 175
 176     if (entnum+1 < sm->used) {
 177       ame_ba_addsubtractone(cht_hb_arrayspace(&hb, sm->byl),
 178                             (sme+1)->start, sm->byl,
 179                             /*subtract:*/ 0x0ffu, 0x0ffu);
 180     } else {
 181       memset(cht_hb_arrayspace(&hb, sm->byl),
 182              0x0ffu, sm->byl);
 183     }
 184
 185     subl[1]= cht_ret_hb(0, hb);  assert(subl[1]);
 186     subl[2]= sme->data;
 187
 188     sublo= Tcl_NewListObj(3,subl);  assert(sublo);
 189     mainlobjsl[listlength++]= sublo;
 190   }
 191   assert(listlength <= sm->used+1);
 192   surrogate= Tcl_NewListObj(listlength,mainlobjsl);  assert(surrogate);
 193   assert(surrogate);
 194
 195   so->bytes= Tcl_GetStringFromObj(surrogate, &so->length);  assert(so->bytes);
 196   surrogate->bytes= 0; surrogate->length= 0; /* we stole it */
 197 }
 198
 199 static AddrMap_Entry *ame_sfa_alloc(AddrMap_Value *am) {
 200   AddrMap_Entry *ame;
 201
 202   ame= am->entries + am->used;
 203
 204   am->used++;
 205   assert(am->used <= am->space);
 206
 207   ame->start= TALLOC(am->byl);  assert(ame->start);
 208   ame->data= 0;
 209   return ame;
 210 }
 211
 212 static int addrmap_t_sfa(Tcl_Interp *ip, Tcl_Obj *o) {
 213   int rc, inlen, eol, innum, bitlen, cmp;
 214   Tcl_Obj *eo, *starto, *endo;
 215   HBytes_Value starthb, endhb;
 216   const Byte *rangeptrs[2];
 217   AddrMap_Value *am;
 218   AddrMap_Entry *ame;
 219
 220   am= TALLOC(sizeof(*am));  assert(am);
 221   am_init0(am,0);
 222
 223   rc= Tcl_ListObjLength(ip,o,&inlen);  if (rc) goto x_badvalue_rc;
 224
 225   if (inlen<0) {
 226     rc= cht_staticerr(ip, "addr-map overall length < 1", 0);
 227     goto x_badvalue_rc;
 228   }
 229
 230   rc= Tcl_ListObjIndex(ip,o,0,&eo);  if (rc) goto x_badvalue_rc;
 231   rc= Tcl_GetIntFromObj(ip,eo,&bitlen);  if (rc) goto x_badvalue_rc;
 232
 233   if (bitlen<0 || bitlen % 8) {
 234     rc= cht_staticerr(ip, "addr-map overall length < 1", 0);
 235     goto x_badvalue_rc;
 236   }
 237
 238   am->byl= bitlen/8;
 239   am_reallocentries(am, (inlen-1)*2+1);
 240
 241   ame= ame_sfa_alloc(am);
 242   memset(ame->start,0,am->byl);
 243
 244   for (innum=1; innum < inlen; innum++) {
 245     rc= Tcl_ListObjIndex(ip,o,innum,&eo);  if (rc) goto x_badvalue_rc;
 246     rc= Tcl_ListObjLength(ip,eo,&eol);  if (rc) goto x_badvalue_rc;
 247
 248     if (eol != 3) {
 249       rc= cht_staticerr(ip, "addr-map entry length != 3", 0);
 250       goto x_badvalue_rc;
 251     }
 252     rc= Tcl_ListObjIndex(ip,eo,0,&starto);  if (rc) goto x_badvalue_rc;
 253     rc= Tcl_ListObjIndex(ip,eo,1,&endo);    if (rc) goto x_badvalue_rc;
 254
 255     rc= cht_pat_hb(ip,starto,&starthb);  if (rc) goto x_badvalue_rc;
 256     rc= cht_pat_hb(ip,endo,&endhb);  if (rc) goto x_badvalue_rc;
 257
 258     rc= ame_parsecheck_range(ip,am,&starthb,&endhb,rangeptrs);
 259     if (rc) goto x_badvalue_rc;
 260
 261     cmp= memcmp(ame->start, rangeptrs[0], am->byl);
 262     if (cmp < 0) {
 263       rc= cht_staticerr(ip, "addr-map entries out of order", 0);
 264       goto x_badvalue_rc;
 265     }
 266     if (cmp > 0) {
 267       ame= ame_sfa_alloc(am);
 268       memcpy(ame->start, rangeptrs[0], am->byl);
 269     }
 270
 271     assert(!ame->data);
 272     rc= Tcl_ListObjIndex(ip,eo,2,&ame->data);  if (rc) goto x_badvalue_rc;
 273     Tcl_IncrRefCount(ame->data);
 274
 275     ame= ame_sfa_alloc(am);
 276     rc= ame_ba_addsubtractone(ame->start, rangeptrs[1], am->byl,
 277                               /*add:*/ 0x01u, 0x00u);
 278     if (rc) {
 279       /* we've overflowed.  it must have been ffffffff.... */
 280       if (innum != inlen-1) {
 281         rc= cht_staticerr(ip, "addr-map non-last entry end is all-bits-1", 0);
 282         goto x_badvalue_rc;
 283       }
 284       TFREE(ame->start);
 285       am->used--;
 286       break;
 287     }
 288   }
 289
 290   /* we commit now */
 291   cht_objfreeir(o);
 292   o->internalRep.otherValuePtr= am;
 293   o->typePtr= &cht_addrmap_type;
 294   return TCL_OK;
 295
 296  x_badvalue_rc:
 297   if (rc == TCL_ERROR)
 298     Tcl_SetObjErrorCode(ip, Tcl_NewStringObj("HBYTES ADDRMAP VALUE", -1));
 299
 300   am_free(am);
 301   return rc;
 302 }
 303
 304 Tcl_ObjType cht_addrmap_type = {
 305   "addr-map",
 306   addrmap_t_free, addrmap_t_dup, addrmap_t_ustr, addrmap_t_sfa
 307 };