RMG: maint perlhist: change from push to pull
[perl.git] / 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