common/iobuf.c: Improve iobuf_peek.
[gnupg.git] / common / t-http.c
1 /* t-http.c
2  * Copyright (C) 1999, 2001, 2002, 2003, 2004, 2006, 2009, 2010,
3  *               2011 Free Software Foundation, Inc.
4  * Copyright (C) 2014 Werner Koch
5  *
6  * This file is part of GnuPG.
7  *
8  * This file is free software; you can redistribute it and/or modify
9  * it under the terms of either
10  *
11  *   - the GNU Lesser General Public License as published by the Free
12  *     Software Foundation; either version 3 of the License, or (at
13  *     your option) any later version.
14  *
15  * or
16  *
17  *   - the GNU General Public License as published by the Free
18  *     Software Foundation; either version 2 of the License, or (at
19  *     your option) any later version.
20  *
21  * or both in parallel, as here.
22  *
23  * This file is distributed in the hope that it will be useful,
24  * but WITHOUT ANY WARRANTY; without even the implied warranty of
25  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
26  * GNU General Public License for more details.
27  *
28  * You should have received a copy of the GNU General Public License
29  * along with this program; if not, see <http://www.gnu.org/licenses/>.
30  */
31
32 #include <config.h>
33 #include <stdlib.h>
34 #include <stdio.h>
35 #include <string.h>
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #include <unistd.h>
39
40 #include "util.h"
41 #include "logging.h"
42 #include "http.h"
43
44
45 #if HTTP_USE_NTBTLS
46 # include <ntbtls.h>
47 #elif HTTP_USE_GNUTLS
48 # include <gnutls/gnutls.h>  /* For init, logging, and deinit.  */
49 #endif /*HTTP_USE_GNUTLS*/
50
51 #define PGM "t-http"
52
53 static int verbose;
54 static int debug;
55 static int no_verify;
56
57 /* static void */
58 /* read_dh_params (const char *fname) */
59 /* { */
60 /*   gpg_error_t err; */
61 /*   int rc; */
62 /*   FILE *fp; */
63 /*   struct stat st; */
64 /*   char *buf; */
65 /*   size_t buflen; */
66 /*   gnutls_datum_t datum; */
67
68 /*   fp = fopen (fname, "rb"); */
69 /*   if (!fp) */
70 /*     { */
71 /*       err = gpg_error_from_syserror (); */
72 /*       log_fatal ("can't open '%s': %s\n", fname, gpg_strerror (err)); */
73 /*     } */
74
75 /*   if (fstat (fileno(fp), &st)) */
76 /*     { */
77 /*       err = gpg_error_from_syserror (); */
78 /*       log_fatal ("can't stat '%s': %s\n", fname, gpg_strerror (err)); */
79 /*     } */
80
81 /*   buflen = st.st_size; */
82 /*   buf = xmalloc (buflen+1); */
83 /*   if (fread (buf, buflen, 1, fp) != 1) */
84 /*     { */
85 /*       err = gpg_error_from_syserror (); */
86 /*       log_fatal ("error reading '%s': %s\n", fname, gpg_strerror (err)); */
87 /*     } */
88 /*   fclose (fp); */
89
90 /*   datum.size = buflen; */
91 /*   datum.data = buf; */
92
93 /*   rc = gnutls_dh_params_import_pkcs3 (dh_params, &datum, GNUTLS_X509_FMT_PEM); */
94 /*   if (rc < 0) */
95 /*     log_fatal ("gnutls_dh_param_import failed: %s\n", gnutls_strerror (rc)); */
96
97 /*   xfree (buf); */
98 /* } */
99
100
101
102 #if HTTP_USE_GNUTLS
103 static gpg_error_t
104 verify_callback (http_t hd, http_session_t session, int reserved)
105 {
106   (void)hd;
107   (void)reserved;
108   return no_verify? 0 : http_verify_server_credentials (session);
109 }
110 #endif
111
112 #if HTTP_USE_GNUTLS
113 static void
114 my_gnutls_log (int level, const char *text)
115 {
116   fprintf (stderr, "gnutls:L%d: %s", level, text);
117 }
118 #endif
119
120 /* Prepend FNAME with the srcdir environment variable's value and
121    return an allocated filename. */
122 static char *
123 prepend_srcdir (const char *fname)
124 {
125   static const char *srcdir;
126   char *result;
127
128   if (!srcdir && !(srcdir = getenv ("srcdir")))
129     srcdir = ".";
130
131   result = xmalloc (strlen (srcdir) + 1 + strlen (fname) + 1);
132   strcpy (result, srcdir);
133   strcat (result, "/");
134   strcat (result, fname);
135   return result;
136 }
137
138
139 int
140 main (int argc, char **argv)
141 {
142   int last_argc = -1;
143   gpg_error_t err;
144   int rc;
145   parsed_uri_t uri;
146   uri_tuple_t r;
147   http_t hd;
148   int c;
149   unsigned int my_http_flags = 0;
150   int no_out = 0;
151   int tls_dbg = 0;
152   const char *cafile = NULL;
153   http_session_t session = NULL;
154
155   gpgrt_init ();
156   log_set_prefix (PGM, 1 | 4);
157   if (argc)
158     { argc--; argv++; }
159   while (argc && last_argc != argc )
160     {
161       last_argc = argc;
162       if (!strcmp (*argv, "--"))
163         {
164           argc--; argv++;
165           break;
166         }
167       else if (!strcmp (*argv, "--help"))
168         {
169           fputs ("usage: " PGM " URL\n"
170                  "Options:\n"
171                  "  --verbose         print timings etc.\n"
172                  "  --debug           flyswatter\n"
173                  "  --gnutls-debug N  use GNUTLS debug level N\n"
174                  "  --cacert FNAME    expect CA certificate in file FNAME\n"
175                  "  --no-verify       do not verify the certificate\n"
176                  "  --force-tls       use HTTP_FLAG_FORCE_TLS\n"
177                  "  --no-out          do not print the content\n",
178                  stdout);
179           exit (0);
180         }
181       else if (!strcmp (*argv, "--verbose"))
182         {
183           verbose++;
184           argc--; argv++;
185         }
186       else if (!strcmp (*argv, "--debug"))
187         {
188           verbose += 2;
189           debug++;
190           argc--; argv++;
191         }
192       else if (!strcmp (*argv, "--gnutls-debug"))
193         {
194           argc--; argv++;
195           if (argc)
196             {
197               tls_dbg = atoi (*argv);
198               argc--; argv++;
199             }
200         }
201       else if (!strcmp (*argv, "--cacert"))
202         {
203           argc--; argv++;
204           if (argc)
205             {
206               cafile = *argv;
207               argc--; argv++;
208             }
209         }
210       else if (!strcmp (*argv, "--no-verify"))
211         {
212           no_verify = 1;
213           argc--; argv++;
214         }
215       else if (!strcmp (*argv, "--force-tls"))
216         {
217           my_http_flags |= HTTP_FLAG_FORCE_TLS;
218           argc--; argv++;
219         }
220       else if (!strcmp (*argv, "--no-out"))
221         {
222           no_out = 1;
223           argc--; argv++;
224         }
225       else if (!strncmp (*argv, "--", 2))
226         {
227           fprintf (stderr, PGM ": unknown option '%s'\n", *argv);
228           exit (1);
229         }
230     }
231   if (argc != 1)
232     {
233       fprintf (stderr, PGM ": no or too many URLS given\n");
234       exit (1);
235     }
236
237   if (!cafile)
238     cafile = prepend_srcdir ("tls-ca.pem");
239
240 #if HTTP_USE_NTBTLS
241
242   (void)err;
243
244   ntbtls_set_debug (tls_dbg, NULL, NULL);
245
246 #elif HTTP_USE_GNUTLS
247
248   rc = gnutls_global_init ();
249   if (rc)
250     log_error ("gnutls_global_init failed: %s\n", gnutls_strerror (rc));
251
252   http_register_tls_callback (verify_callback);
253   http_register_tls_ca (cafile);
254
255   err = http_session_new (&session, NULL);
256   if (err)
257     log_error ("http_session_new failed: %s\n", gpg_strerror (err));
258
259   /* rc = gnutls_dh_params_init(&dh_params); */
260   /* if (rc) */
261   /*   log_error ("gnutls_dh_params_init failed: %s\n", gnutls_strerror (rc)); */
262   /* read_dh_params ("dh_param.pem"); */
263
264   /* rc = gnutls_certificate_set_x509_trust_file */
265   /*   (certcred, "ca.pem", GNUTLS_X509_FMT_PEM); */
266   /* if (rc) */
267   /*   log_error ("gnutls_certificate_set_x509_trust_file failed: %s\n", */
268   /*              gnutls_strerror (rc)); */
269
270   /* gnutls_certificate_set_dh_params (certcred, dh_params); */
271
272   gnutls_global_set_log_function (my_gnutls_log);
273   if (tls_dbg)
274     gnutls_global_set_log_level (tls_dbg);
275
276 #endif /*HTTP_USE_GNUTLS*/
277
278   rc = http_parse_uri (&uri, *argv, 1);
279   if (rc)
280     {
281       log_error ("'%s': %s\n", *argv, gpg_strerror (rc));
282       return 1;
283     }
284
285   printf ("Scheme: %s\n", uri->scheme);
286   if (uri->opaque)
287     printf ("Value : %s\n", uri->path);
288   else
289     {
290       printf ("Auth  : %s\n", uri->auth? uri->auth:"[none]");
291       printf ("Host  : %s\n", uri->host);
292       printf ("Port  : %u\n", uri->port);
293       printf ("Path  : %s\n", uri->path);
294       for (r = uri->params; r; r = r->next)
295         {
296           printf ("Params: %s", r->name);
297           if (!r->no_value)
298             {
299               printf ("=%s", r->value);
300               if (strlen (r->value) != r->valuelen)
301                 printf (" [real length=%d]", (int) r->valuelen);
302             }
303           putchar ('\n');
304         }
305       for (r = uri->query; r; r = r->next)
306         {
307           printf ("Query : %s", r->name);
308           if (!r->no_value)
309             {
310               printf ("=%s", r->value);
311               if (strlen (r->value) != r->valuelen)
312                 printf (" [real length=%d]", (int) r->valuelen);
313             }
314           putchar ('\n');
315         }
316       printf ("TLS   : %s\n",
317               uri->use_tls? "yes":
318               (my_http_flags&HTTP_FLAG_FORCE_TLS)? "forced" : "no");
319
320     }
321   fflush (stdout);
322   http_release_parsed_uri (uri);
323   uri = NULL;
324
325   rc = http_open_document (&hd, *argv, NULL, my_http_flags,
326                            NULL, session, NULL, NULL);
327   if (rc)
328     {
329       log_error ("can't get '%s': %s\n", *argv, gpg_strerror (rc));
330       return 1;
331     }
332   log_info ("open_http_document succeeded; status=%u\n",
333             http_get_status_code (hd));
334
335   {
336     const char **names;
337     int i;
338
339     names = http_get_header_names (hd);
340     if (!names)
341       log_fatal ("http_get_header_names failed: %s\n",
342                  gpg_strerror (gpg_error_from_syserror ()));
343     for (i = 0; names[i]; i++)
344       printf ("HDR: %s: %s\n", names[i], http_get_header (hd, names[i]));
345     xfree (names);
346   }
347   fflush (stdout);
348
349   switch (http_get_status_code (hd))
350     {
351     case 200:
352     case 400:
353     case 401:
354     case 403:
355     case 404:
356       {
357         unsigned long count = 0;
358         while ((c = es_getc (http_get_read_ptr (hd))) != EOF)
359           {
360             count++;
361             if (!no_out)
362               putchar (c);
363           }
364         log_info ("Received bytes: %lu\n", count);
365       }
366       break;
367     case 301:
368     case 302:
369     case 307:
370       log_info ("Redirected to: %s\n", http_get_header (hd, "Location"));
371       break;
372     }
373   http_close (hd, 0);
374
375   http_session_release (session);
376 #ifdef HTTP_USE_GNUTLS
377   gnutls_global_deinit ();
378 #endif /*HTTP_USE_GNUTLS*/
379
380   return 0;
381 }