Report SRU diagnostic 12 (too many chars in query) when rendering the
[yaz-moved-to-github.git] / src / zgdu.c
1 /*
2  * Copyright (C) 1995-2005, Index Data ApS
3  * See the file LICENSE for details.
4  *
5  * $Id: zgdu.c,v 1.14 2006-03-01 23:24:25 adam Exp $
6  */
7
8 /**
9  * \file zgdu.c
10  * \brief Implements HTTP and Z39.50 encoding and decoding.
11  */
12
13 #include <ctype.h>
14 #include <yaz/odr.h>
15 #include <yaz/yaz-version.h>
16 #include <yaz/yaz-iconv.h>
17 #include <yaz/zgdu.h>
18
19 #ifdef WIN32
20 #define strncasecmp _strnicmp
21 #define strcasecmp _stricmp
22 #endif
23  
24 static int decode_headers_content(ODR o, int off, Z_HTTP_Header **headers,
25                                   char **content_buf, int *content_len)
26 {
27     int i = off;
28     int chunked = 0;
29
30     *headers = 0;
31     while (i < o->size-1 && o->buf[i] == '\r')
32     {
33         int po;
34         i++;
35         if (o->buf[i] != '\n')
36         {
37             o->error = OHTTP;
38             return 0;
39         }
40         i++;
41         if (o->buf[i] == '\r')
42             break;
43         for (po = i; ; i++)
44         {
45             if (i == o->size)
46             {
47                 o->error = OHTTP;
48                 return 0;
49             }
50             else if (o->buf[i] == ':')
51                 break;
52         }
53         *headers = (Z_HTTP_Header *) odr_malloc(o, sizeof(**headers));
54         (*headers)->name = (char*) odr_malloc(o, i - po + 1);
55         memcpy ((*headers)->name, o->buf + po, i - po);
56         (*headers)->name[i - po] = '\0';
57         i++;
58         while (i < o->size-1 && o->buf[i] == ' ')
59             i++;
60         for (po = i; i < o->size-1 && o->buf[i] != '\r' ; i++)
61             ;
62         
63         (*headers)->value = (char*) odr_malloc(o, i - po + 1);
64         memcpy ((*headers)->value, o->buf + po, i - po);
65         (*headers)->value[i - po] = '\0';
66         
67         if (!strcasecmp((*headers)->name, "Transfer-Encoding")
68             &&
69             !strcasecmp((*headers)->value, "chunked"))
70             chunked = 1;
71         headers = &(*headers)->next;
72     }
73     *headers = 0;
74     i++;
75     if (o->buf[i] != '\n')
76     {
77         o->error = OHTTP;
78         return 0;
79     }
80     i++;
81
82     if (chunked)
83     {
84         int off = 0;
85         
86         /* we know buffer will be smaller than o->size - i*/
87         *content_buf = (char*) odr_malloc(o, o->size - i);  
88         
89         while (1)
90         {
91             /* chunk length .. */
92             int chunk_len = 0;
93             for (; i  < o->size-2; i++)
94                 if (isdigit(o->buf[i]))
95                     chunk_len = chunk_len * 16 + 
96                         (o->buf[i] - '0');
97                 else if (isupper(o->buf[i]))
98                     chunk_len = chunk_len * 16 + 
99                         (o->buf[i] - ('A'-10));
100                 else if (islower(o->buf[i]))
101                     chunk_len = chunk_len * 16 + 
102                         (o->buf[i] - ('a'-10));
103                 else
104                     break;
105             /* chunk extension ... */
106             while (o->buf[i] != '\r' && o->buf[i+1] != '\n')
107             {
108                 if (i >= o->size-2)
109                 {
110                     o->error = OHTTP;
111                     return 0;
112                 }
113                 i++;
114             }
115             i += 2;  /* skip CRLF */
116             if (chunk_len == 0)
117                 break;
118             if (chunk_len < 0 || off + chunk_len > o->size)
119             {
120                 o->error = OHTTP;
121                 return 0;
122             }
123             /* copy chunk .. */
124             memcpy (*content_buf + off, o->buf + i, chunk_len);
125             i += chunk_len + 2; /* skip chunk+CRLF */
126             off += chunk_len;
127         }
128         if (!off)
129             *content_buf = 0;
130         *content_len = off;
131     }
132     else
133     {
134         if (i > o->size)
135         {
136             o->error = OHTTP;
137             return 0;
138         }
139         else if (i == o->size)
140         {
141             *content_buf = 0;
142             *content_len = 0;
143         }
144         else 
145         {
146             *content_len = o->size - i;
147             *content_buf = (char*) odr_malloc(o, *content_len + 1);
148             memcpy(*content_buf, o->buf + i, *content_len);
149             (*content_buf)[*content_len] = '\0';
150         }
151     }
152     return 1;
153 }
154
155 void z_HTTP_header_add_content_type(ODR o, Z_HTTP_Header **hp,
156                                     const char *content_type,
157                                     const char *charset)
158 {
159     const char *l = "Content-Type";
160     if (charset)
161     {
162         char *ctype = odr_malloc(o, strlen(content_type)+strlen(charset) + 15);
163         sprintf(ctype, "%s; charset=%s", content_type, charset);
164         z_HTTP_header_add(o, hp, l, ctype);
165     }
166     else
167         z_HTTP_header_add(o, hp, l, content_type);
168
169 }
170
171 void z_HTTP_header_add(ODR o, Z_HTTP_Header **hp, const char *n,
172                        const char *v)
173 {
174     while (*hp)
175         hp = &(*hp)->next;
176     *hp = (Z_HTTP_Header *) odr_malloc(o, sizeof(**hp));
177     (*hp)->name = odr_strdup(o, n);
178     (*hp)->value = odr_strdup(o, v);
179     (*hp)->next = 0;
180 }
181
182 const char *z_HTTP_header_lookup(Z_HTTP_Header *hp, const char *n)
183 {
184     for (; hp; hp = hp->next)
185         if (!yaz_matchstr(hp->name, n))
186             return hp->value;
187     return 0;
188 }
189
190
191 Z_GDU *z_get_HTTP_Request(ODR o)
192 {
193     Z_GDU *p = (Z_GDU *) odr_malloc(o, sizeof(*p));
194     Z_HTTP_Request *hreq;
195
196     p->which = Z_GDU_HTTP_Request;
197     p->u.HTTP_Request = (Z_HTTP_Request *) odr_malloc(o, sizeof(*hreq));
198     hreq = p->u.HTTP_Request;
199     hreq->headers = 0;
200     hreq->content_len = 0;
201     hreq->content_buf = 0;
202     hreq->version = "1.1";
203     hreq->method = "POST";
204     hreq->path = "/";
205     z_HTTP_header_add(o, &hreq->headers, "User-Agent",
206                       "YAZ/" YAZ_VERSION);
207     return p;
208 }
209
210 Z_GDU *z_get_HTTP_Response(ODR o, int code)
211 {
212     Z_GDU *p = (Z_GDU *) odr_malloc(o, sizeof(*p));
213     Z_HTTP_Response *hres;
214
215     p->which = Z_GDU_HTTP_Response;
216     p->u.HTTP_Response = (Z_HTTP_Response *) odr_malloc(o, sizeof(*hres));
217     hres = p->u.HTTP_Response;
218     hres->headers = 0;
219     hres->content_len = 0;
220     hres->content_buf = 0;
221     hres->code = code;
222     hres->version = "1.1";
223     z_HTTP_header_add(o, &hres->headers, "Server",
224                       "YAZ/" YAZ_VERSION);
225     if (code != 200)
226     {
227         hres->content_buf = (char*) odr_malloc(o, 400);
228         sprintf (hres->content_buf, 
229                  "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">\n"
230                  "<HTML>\n"
231                  " <HEAD>\n"
232                  "  <TITLE>YAZ " YAZ_VERSION "</TITLE>\n"
233                  " </HEAD>\n"
234                  " <BODY>\n"
235                  "  <P><A HREF=\"http://www.indexdata.dk/yaz/\">YAZ</A> " 
236                  YAZ_VERSION "</P>\n"
237                  "  <P>Error: %d</P>\n"
238                  "  <P>Description: %.50s</P>\n"
239                  " </BODY>\n"
240                  "</HTML>\n",
241                  code, z_HTTP_errmsg(code));
242         hres->content_len = strlen(hres->content_buf);
243         z_HTTP_header_add(o, &hres->headers, "Content-Type", "text/html");
244     }
245     return p;
246 }
247
248 const char *z_HTTP_errmsg(int code)
249 {
250     if (code == 200)
251         return "OK";
252     else if (code == 400)
253         return "Bad Request";
254     else if (code == 404)
255         return "Not Found";
256     else if (code == 405)
257         return "Method Not Allowed";
258     else if (code == 500)
259         return "Internal Error";
260     else
261         return "Unknown Error";
262 }
263
264 int z_GDU (ODR o, Z_GDU **p, int opt, const char *name)
265 {
266     if (o->direction == ODR_DECODE) {
267         *p = (Z_GDU *) odr_malloc(o, sizeof(**p));
268         if (o->size > 10 && !memcmp(o->buf, "HTTP/", 5))
269         {
270             int i, po;
271             Z_HTTP_Response *hr;
272             (*p)->which = Z_GDU_HTTP_Response;
273
274             hr = (*p)->u.HTTP_Response = (Z_HTTP_Response *)
275                 odr_malloc(o, sizeof(*hr));
276             hr->content_buf = 0;
277             hr->content_len = 0;
278
279             po = i = 5;
280             while (i < o->size-2 && o->buf[i] != ' ' && o->buf[i] != '\r')
281                 i++;
282             hr->version = (char *) odr_malloc(o, i - po + 1);
283             if (i - po)
284                 memcpy(hr->version, o->buf + po, i - po);
285             hr->version[i-po] = 0;
286             if (o->buf[i] != ' ')
287             {
288                 o->error = OHTTP;
289                 return 0;
290             }
291             i++;
292             hr->code = 0;
293             while (i < o->size-2 && o->buf[i] >= '0' && o->buf[i] <= '9')
294             {
295                 hr->code = hr->code*10 + (o->buf[i] - '0');
296                 i++;
297             }
298             while (i < o->size-1 && o->buf[i] != '\r')
299                 i++;
300             return decode_headers_content(o, i, &hr->headers,
301                                           &hr->content_buf, &hr->content_len);            
302         }
303         else if (o->size > 5 &&
304             o->buf[0] >= 0x20 && o->buf[0] < 0x7f
305             && o->buf[1] >= 0x20 && o->buf[1] < 0x7f
306             && o->buf[2] >= 0x20 && o->buf[2] < 0x7f
307             && o->buf[3] >= 0x20 && o->buf[3] < 0x7f)
308         {
309             int i, po;
310             Z_HTTP_Request *hr;
311
312             (*p)->which = Z_GDU_HTTP_Request;
313             hr = (*p)->u.HTTP_Request = 
314                 (Z_HTTP_Request *) odr_malloc(o, sizeof(*hr));
315
316             /* method .. */
317             for (i = 0; o->buf[i] != ' '; i++)
318                 if (i >= o->size-5 || i > 30)
319                 {
320                     o->error = OHTTP;
321                     return 0;
322                 }
323             hr->method = (char *) odr_malloc(o, i+1);
324             memcpy (hr->method, o->buf, i);
325             hr->method[i] = '\0';
326             /* path */
327             po = i+1;
328             for (i = po; o->buf[i] != ' '; i++)
329                 if (i >= o->size-5)
330                 {
331                     o->error = OHTTP;
332                     return 0;
333                 }
334             hr->path = (char *) odr_malloc(o, i - po + 1);
335             memcpy (hr->path, o->buf+po, i - po);
336             hr->path[i - po] = '\0';
337             /* HTTP version */
338             i++;
339             if (i > o->size-5 || memcmp(o->buf+i, "HTTP/", 5))
340             {
341                 o->error = OHTTP;
342                 return 0;
343             }
344             i+= 5;
345             po = i;
346             while (o->buf[i] != '\r')
347             {
348                 if (i >= o->size-1)
349                 {
350                     o->error = OHTTP;
351                     return 0;
352                 }
353                 i++;
354             }
355             hr->version = (char *) odr_malloc(o, i - po + 1);
356             memcpy(hr->version, o->buf + po, i - po);
357             hr->version[i - po] = '\0';
358             /* headers */
359             return decode_headers_content(o, i, &hr->headers,
360                                           &hr->content_buf, &hr->content_len);
361
362         }
363         else
364         {
365             (*p)->which = Z_GDU_Z3950;
366             return z_APDU(o, &(*p)->u.z3950, opt, 0);
367         }
368     }
369     else /* ENCODE or PRINT */
370     {
371         int top0 = o->top;
372         char sbuf[80];
373         Z_HTTP_Header *h;
374         switch((*p)->which)
375         {
376         case Z_GDU_HTTP_Response:
377             sprintf(sbuf, "HTTP/%s %d %s\r\n", (*p)->u.HTTP_Response->version,
378                     (*p)->u.HTTP_Response->code,
379                     z_HTTP_errmsg((*p)->u.HTTP_Response->code));
380             odr_write(o, (unsigned char *) sbuf, strlen(sbuf));
381             /* apply Content-Length if not already applied */
382             if (!z_HTTP_header_lookup((*p)->u.HTTP_Response->headers,
383                                       "Content-Length"))
384             {
385                 char lstr[60];
386                 sprintf(lstr, "Content-Length: %d\r\n",
387                         (*p)->u.HTTP_Response->content_len);
388                 odr_write(o, (unsigned char *) lstr, strlen(lstr));
389             }
390             for (h = (*p)->u.HTTP_Response->headers; h; h = h->next)
391             {
392                 odr_write(o, (unsigned char *) h->name, strlen(h->name));
393                 odr_write(o, (unsigned char *) ": ", 2);
394                 odr_write(o, (unsigned char *) h->value, strlen(h->value));
395                 odr_write(o, (unsigned char *) "\r\n", 2);
396             }
397             odr_write(o, (unsigned char *) "\r\n", 2);
398             if ((*p)->u.HTTP_Response->content_buf)
399                 odr_write(o, (unsigned char *) 
400                           (*p)->u.HTTP_Response->content_buf,
401                           (*p)->u.HTTP_Response->content_len);
402             if (o->direction == ODR_PRINT)
403             {
404                 odr_printf(o, "-- HTTP response:\n%.*s\n", o->top - top0,
405                            o->buf + top0);
406                 odr_printf(o, "-- \n");
407             }
408             break;
409         case Z_GDU_HTTP_Request:
410             odr_write(o, (unsigned char *) (*p)->u.HTTP_Request->method,
411                       strlen((*p)->u.HTTP_Request->method));
412             odr_write(o, (unsigned char *) " ", 1);
413             odr_write(o, (unsigned char *) (*p)->u.HTTP_Request->path,
414                       strlen((*p)->u.HTTP_Request->path));
415             odr_write(o, (unsigned char *) " HTTP/", 6);
416             odr_write(o, (unsigned char *) (*p)->u.HTTP_Request->version,
417                       strlen((*p)->u.HTTP_Request->version));
418             odr_write(o, (unsigned char *) "\r\n", 2);
419             if ((*p)->u.HTTP_Request->content_len &&
420                 !z_HTTP_header_lookup((*p)->u.HTTP_Request->headers,
421                                       "Content-Length"))
422             {
423                 char lstr[60];
424                 sprintf(lstr, "Content-Length: %d\r\n",
425                         (*p)->u.HTTP_Request->content_len);
426                 odr_write(o, (unsigned char *) lstr, strlen(lstr));
427             }
428             for (h = (*p)->u.HTTP_Request->headers; h; h = h->next)
429             {
430                 odr_write(o, (unsigned char *) h->name, strlen(h->name));
431                 odr_write(o, (unsigned char *) ": ", 2);
432                 odr_write(o, (unsigned char *) h->value, strlen(h->value));
433                 odr_write(o, (unsigned char *) "\r\n", 2);
434             }
435             odr_write(o, (unsigned char *) "\r\n", 2);
436             if ((*p)->u.HTTP_Request->content_buf)
437                 odr_write(o, (unsigned char *)
438                           (*p)->u.HTTP_Request->content_buf,
439                           (*p)->u.HTTP_Request->content_len);
440             if (o->direction == ODR_PRINT)
441             {
442                 odr_printf(o, "-- HTTP request:\n%.*s\n", o->top - top0,
443                         o->buf + top0);
444                 odr_printf(o, "-- \n");
445             }
446             break;
447         case Z_GDU_Z3950:
448             return z_APDU(o, &(*p)->u.z3950, opt, 0);
449         }
450     }
451     return 1;
452 }
453
454 /*
455  * Local variables:
456  * c-basic-offset: 4
457  * indent-tabs-mode: nil
458  * End:
459  * vim: shiftwidth=4 tabstop=8 expandtab
460  */
461