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