fix arguments to uclient_http_reset_headers
[project/uclient.git] / uclient-example.c
1 /*
2  * uclient - ustream based protocol client library
3  *
4  * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
5  *
6  * Permission to use, copy, modify, and/or distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18 #include <libubox/blobmsg.h>
19 #include <unistd.h>
20 #include <stdio.h>
21
22 #include "uclient.h"
23
24 static void example_header_done(struct uclient *cl)
25 {
26         struct blob_attr *cur;
27         int rem;
28
29         printf("Headers (%d): \n", cl->status_code);
30         blobmsg_for_each_attr(cur, cl->meta, rem) {
31                 printf("%s=%s\n", blobmsg_name(cur), (char *) blobmsg_data(cur));
32         }
33
34         printf("Contents:\n");
35 }
36
37 static void example_read_data(struct uclient *cl)
38 {
39         char buf[256];
40         int len;
41
42         while (1) {
43                 len = uclient_read(cl, buf, sizeof(buf));
44                 if (!len)
45                         return;
46
47                 write(STDOUT_FILENO, buf, len);
48         }
49 }
50
51 static void example_request_sm(struct uclient *cl)
52 {
53         static int i = 0;
54
55         switch (i++) {
56         case 0:
57                 uclient_connect(cl);
58                 uclient_http_set_request_type(cl, "HEAD");
59                 uclient_request(cl);
60                 break;
61         case 1:
62                 uclient_connect(cl);
63                 uclient_http_set_request_type(cl, "GET");
64                 uclient_request(cl);
65                 break;
66         default:
67                 uloop_end();
68                 break;
69         };
70 }
71
72 static void example_eof(struct uclient *cl)
73 {
74         static int retries;
75
76         if (retries < 10 && uclient_http_redirect(cl)) {
77                 retries++;
78                 return;
79         }
80
81         retries = 0;
82         example_request_sm(cl);
83 }
84
85 static void example_error(struct uclient *cl, int code)
86 {
87         fprintf(stderr, "Error %d!\n", code);
88         example_request_sm(cl);
89 }
90
91 static const struct uclient_cb cb = {
92         .header_done = example_header_done,
93         .data_read = example_read_data,
94         .data_eof = example_eof,
95         .error = example_error,
96 };
97
98 static int usage(const char *progname)
99 {
100         fprintf(stderr,
101                 "Usage: %s [options] <hostname> <port>\n"
102                 "Options:\n"
103                 "       -c <cert>:         Load CA certificates from file <cert>\n"
104                 "       -C:                Skip certificate CN verification against hostname\n"
105                 "\n", progname);
106         return 1;
107 }
108
109
110 int main(int argc, char **argv)
111 {
112         struct ustream_ssl_ctx *ctx;
113         const char *progname = argv[0];
114         struct uclient *cl;
115         bool verify = true;
116         int ch;
117
118         ctx = ustream_ssl_context_new(false);
119
120         while ((ch = getopt(argc, argv, "Cc:")) != -1) {
121                 switch(ch) {
122                 case 'c':
123                         ustream_ssl_context_add_ca_crt_file(ctx, optarg);
124                         break;
125                 case 'C':
126                         verify = false;
127                         break;
128                 default:
129                         return usage(progname);
130                 }
131         }
132
133         argv += optind;
134         argc -= optind;
135
136         if (argc != 1)
137                 return usage(progname);
138
139         uloop_init();
140
141         cl = uclient_new(argv[0], NULL, &cb);
142         if (!cl) {
143                 fprintf(stderr, "Failed to allocate uclient context\n");
144                 return 1;
145         }
146
147         uclient_http_set_ssl_ctx(cl, ctx, verify);
148         example_request_sm(cl);
149         uloop_run();
150         uloop_done();
151
152         uclient_free(cl);
153         ustream_ssl_context_free(ctx);
154
155
156         return 0;
157 }