x2p/str.c

   1 /*    str.c
   2  *
   3  *    Copyright (C) 1991, 1992, 1993, 1994, 1995, 1996, 1997, 1999,
   4  *    2001, 2002, 2005 by Larry Wall and others
   5  *
   6  *    You may distribute under the terms of either the GNU General Public
   7  *    License or the Artistic License, as specified in the README file.
   8  */
   9
  10 #include "EXTERN.h"
  11 #include "a2p.h"
  12 #include "util.h"
  13
  14 void
  15 str_numset(STR *str, double num)
  16 {
  17     str->str_nval = num;
  18     str->str_pok = 0;           /* invalidate pointer */
  19     str->str_nok = 1;           /* validate number */
  20 }
  21
  22 char *
  23 str_2ptr(STR *str)
  24 {
  25     char *s;
  26
  27     if (!str)
  28         return (char *)"";      /* probably safe - won't be written to */
  29     GROWSTR(&(str->str_ptr), &(str->str_len), 24);
  30     s = str->str_ptr;
  31     if (str->str_nok) {
  32         sprintf(s,"%.20g",str->str_nval);
  33         while (*s) s++;
  34     }
  35     *s = '\0';
  36     str->str_cur = s - str->str_ptr;
  37     str->str_pok = 1;
  38 #ifdef DEBUGGING
  39     if (debug & 32)
  40         fprintf(stderr,"0x%lx ptr(%s)\n",(unsigned long)str,str->str_ptr);
  41 #endif
  42     return str->str_ptr;
  43 }
  44
  45 void
  46 str_sset(STR *dstr, STR *sstr)
  47 {
  48     if (!sstr)
  49         str_nset(dstr,No,0);
  50     else if (sstr->str_nok)
  51         str_numset(dstr,sstr->str_nval);
  52     else if (sstr->str_pok)
  53         str_nset(dstr,sstr->str_ptr,sstr->str_cur);
  54     else
  55         str_nset(dstr,"",0);
  56 }
  57
  58 void
  59 str_nset(STR *str, const char *ptr, int len)
  60 {
  61     GROWSTR(&(str->str_ptr), &(str->str_len), len + 1);
  62     memcpy(str->str_ptr,ptr,len);
  63     str->str_cur = len;
  64     *(str->str_ptr+str->str_cur) = '\0';
  65     str->str_nok = 0;           /* invalidate number */
  66     str->str_pok = 1;           /* validate pointer */
  67 }
  68
  69 void
  70 str_set(STR *str, const char *ptr)
  71 {
  72     int len;
  73
  74     if (!ptr)
  75         ptr = "";
  76     len = strlen(ptr);
  77     GROWSTR(&(str->str_ptr), &(str->str_len), len + 1);
  78     memcpy(str->str_ptr,ptr,len+1);
  79     str->str_cur = len;
  80     str->str_nok = 0;           /* invalidate number */
  81     str->str_pok = 1;           /* validate pointer */
  82 }
  83
  84 void
  85 str_ncat(STR *str, const char *ptr, int len)
  86 {
  87     if (!(str->str_pok))
  88         str_2ptr(str);
  89     GROWSTR(&(str->str_ptr), &(str->str_len), str->str_cur + len + 1);
  90     memcpy(str->str_ptr+str->str_cur, ptr, len);
  91     str->str_cur += len;
  92     *(str->str_ptr+str->str_cur) = '\0';
  93     str->str_nok = 0;           /* invalidate number */
  94     str->str_pok = 1;           /* validate pointer */
  95 }
  96
  97 void
  98 str_scat(STR *dstr, STR *sstr)
  99 {
 100     if (!(sstr->str_pok))
 101         str_2ptr(sstr);
 102     if (sstr)
 103         str_ncat(dstr,sstr->str_ptr,sstr->str_cur);
 104 }
 105
 106 void
 107 str_cat(STR *str, const char *ptr)
 108 {
 109     int len;
 110
 111     if (!ptr)
 112         return;
 113     if (!(str->str_pok))
 114         str_2ptr(str);
 115     len = strlen(ptr);
 116     GROWSTR(&(str->str_ptr), &(str->str_len), str->str_cur + len + 1);
 117     memcpy(str->str_ptr+str->str_cur, ptr, len+1);
 118     str->str_cur += len;
 119     str->str_nok = 0;           /* invalidate number */
 120     str->str_pok = 1;           /* validate pointer */
 121 }
 122
 123 STR *
 124 str_new(int len)
 125 {
 126     STR *str;
 127
 128     if (freestrroot) {
 129         str = freestrroot;
 130         freestrroot = str->str_link.str_next;
 131     }
 132     else {
 133         str = (STR *) safemalloc(sizeof(STR));
 134         memset((char*)str,0,sizeof(STR));
 135     }
 136     if (len)
 137         GROWSTR(&(str->str_ptr), &(str->str_len), len + 1);
 138     return str;
 139 }
 140
 141 /* make str point to what nstr did */
 142
 143 void
 144 str_free(STR *str)
 145 {
 146     if (!str)
 147         return;
 148     if (str->str_len)
 149         str->str_ptr[0] = '\0';
 150     str->str_cur = 0;
 151     str->str_nok = 0;
 152     str->str_pok = 0;
 153     str->str_link.str_next = freestrroot;
 154     freestrroot = str;
 155 }
 156
 157 int
 158 str_len(STR *str)
 159 {
 160     if (!str)
 161         return 0;
 162     if (!(str->str_pok))
 163         str_2ptr(str);
 164     if (str->str_len)
 165         return str->str_cur;
 166     else
 167         return 0;
 168 }
 169
 170 char *
 171 str_gets(STR *str, FILE *fp)
 172 {
 173 #if defined(USE_STDIO_PTR) && defined(STDIO_PTR_LVALUE) && defined(STDIO_CNT_LVALUE)
 174     /* Here is some breathtakingly efficient cheating */
 175
 176     char *bp;           /* we're going to steal some values */
 177     int cnt;            /*  from the stdio struct and put EVERYTHING */
 178     STDCHAR *ptr;       /*   in the innermost loop into registers */
 179     char newline = '\n';        /* (assuming at least 6 registers) */
 180     int i;
 181     int bpx;
 182
 183 #if defined(VMS)
 184     /* An ungetc()d char is handled separately from the regular
 185      * buffer, so we getc() it back out and stuff it in the buffer.
 186      */
 187     i = getc(fp);
 188     if (i == EOF) return NULL;
 189     *(--((*fp)->_ptr)) = (unsigned char) i;
 190     (*fp)->_cnt++;
 191 #endif
 192
 193     cnt = FILE_cnt(fp);                 /* get count into register */
 194     str->str_nok = 0;                   /* invalidate number */
 195     str->str_pok = 1;                   /* validate pointer */
 196     if (str->str_len <= cnt)            /* make sure we have the room */
 197         GROWSTR(&(str->str_ptr), &(str->str_len), cnt+1);
 198     bp = str->str_ptr;                  /* move these two too to registers */
 199     ptr = (STDCHAR*)FILE_ptr(fp);
 200     for (;;) {
 201         while (--cnt >= 0) {
 202             if ((*bp++ = *ptr++) == newline) {
 203                 if (bp <= str->str_ptr || bp[-2] != '\\')
 204                     goto thats_all_folks;
 205                 else {
 206                     line++;
 207                     bp -= 2;
 208                 }
 209             }
 210         }
 211
 212         FILE_cnt(fp) = cnt;             /* deregisterize cnt and ptr */
 213         FILE_ptr(fp) = ptr;
 214         i = getc(fp);           /* get more characters */
 215         cnt = FILE_cnt(fp);
 216         ptr = (STDCHAR*)FILE_ptr(fp);           /* reregisterize cnt and ptr */
 217
 218         bpx = bp - str->str_ptr;        /* prepare for possible relocation */
 219         GROWSTR(&(str->str_ptr), &(str->str_len), str->str_cur + cnt + 1);
 220         bp = str->str_ptr + bpx;        /* reconstitute our pointer */
 221
 222         if (i == newline) {             /* all done for now? */
 223             *bp++ = i;
 224             goto thats_all_folks;
 225         }
 226         else if (i == EOF)              /* all done for ever? */
 227             goto thats_all_folks;
 228         *bp++ = i;                      /* now go back to screaming loop */
 229     }
 230
 231 thats_all_folks:
 232     FILE_cnt(fp) = cnt;                 /* put these back or we're in trouble */
 233     FILE_ptr(fp) = ptr;
 234     *bp = '\0';
 235     str->str_cur = bp - str->str_ptr;   /* set length */
 236
 237 #else /* USE_STDIO_PTR && STDIO_PTR_LVALUE && STDIO_CNT_LVALUE */
 238     /* The big, slow, and stupid way */
 239
 240     static char buf[4192];
 241
 242     if (fgets(buf, sizeof buf, fp) != NULL)
 243         str_set(str, buf);
 244     else
 245         str_set(str, No);
 246
 247 #endif /* USE_STDIO_PTR && STDIO_PTR_LVALUE && STDIO_CNT_LVALUE */
 248
 249     return str->str_cur ? str->str_ptr : NULL;
 250 }
 251
 252 STR *
 253 str_make(const char *s)
 254 {
 255     STR *str = str_new(0);
 256
 257     str_set(str,s);
 258     return str;
 259 }
 260