Add other string search method for cgi table.
[bertos.git] / bertos / net / http.c
1 /**
2  * \file
3  * <!--
4  * This file is part of BeRTOS.
5  *
6  * Bertos is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19  *
20  * As a special exception, you may use this file as part of a free software
21  * library without restriction.  Specifically, if other files instantiate
22  * templates or use macros or inline functions from this file, or you compile
23  * this file and link it with other files to produce an executable, this
24  * file does not by itself cause the resulting executable to be covered by
25  * the GNU General Public License.  This exception does not however
26  * invalidate any other reasons why the executable file might be covered by
27  * the GNU General Public License.
28  *
29  * Copyright 2011 Develer S.r.l. (http://www.develer.com/)
30  *
31  * -->
32  *
33  * \author Daniele Basile <asterix@develer.com>
34  *
35  * \brief Simple Http server.
36  *
37  * This simple web server read the site's pages from SD card, and manage
38  * the cases where SD is not present or page not found, using embedded pages.
39  * Quering from browser the /status page, the server return a json dictionary where are store
40  * some board status info, like board temperature, up-time, etc.
41  *
42  * notest: avr
43  */
44
45 #include "http.h"
46
47 #include "hw/hw_sd.h"
48 #include "hw/hw_http.h"
49
50 #include "cfg/cfg_http.h"
51
52 // Define logging setting (for cfg/log.h module).
53 #define LOG_LEVEL         HTTP_LOG_LEVEL
54 #define LOG_VERBOSITY     HTTP_LOG_FORMAT
55 #include <cfg/log.h>
56
57 #include <stdio.h>
58 #include <string.h>
59
60
61 static const char http_html_hdr_200[] = "HTTP/1.1 200 OK\r\nContent-type: text/html\r\n\r\n";
62 static const char http_html_hdr_404[] = "HTTP/1.1 404 Not Found\r\nContent-type: text/html\r\n\r\n";
63 static const char http_html_hdr_500[] = "HTTP/1.1 500 Internal Server Error\r\nContent-type: text/html\r\n\r\n";
64
65 static HttpCGI *cgi_table;
66 static http_handler_t http_callback;
67
68 /**
69  * Send on \param client socket
70  * the 200 Ok http header
71  */
72 void http_sendOk(struct netconn *client)
73 {
74         netconn_write(client, http_html_hdr_200, sizeof(http_html_hdr_200) - 1, NETCONN_NOCOPY);
75 }
76
77
78 /**
79  * Send on \param client socket
80  * the 404 File not found http header
81  */
82 void http_sendFileNotFound(struct netconn *client)
83 {
84         netconn_write(client, http_html_hdr_404, sizeof(http_html_hdr_404) - 1, NETCONN_NOCOPY);
85 }
86
87 /**
88  * Send on \param client socket
89  * the 500 internal server error http header
90  */
91 void http_sendInternalErr(struct netconn *client)
92 {
93         netconn_write(client, http_html_hdr_500, sizeof(http_html_hdr_500) - 1, NETCONN_NOCOPY);
94 }
95
96 void http_getPageName(const char *revc_buf, size_t recv_len, char *page_name, size_t len)
97 {
98         int i = 0;
99         bool str_ok = false;
100
101         if (revc_buf && (recv_len > sizeof("GET /")))
102         {
103                 if (*revc_buf++ == 'G' &&
104                         *revc_buf++ == 'E' && *revc_buf++ == 'T')
105                         {
106                                 str_ok = true;
107                                 /* skip the space and "/" */
108                                 revc_buf += 2;
109                         }
110         }
111
112         if (str_ok)
113         {
114                 while ((size_t)i < recv_len)
115                 {
116                         char ch = *(revc_buf++);
117                         if (ch == ' ' || ch == '\t' || ch == '\n')
118                                 break;
119                         if((size_t)i == len - 1)
120                                 break;
121                         page_name[i++] = ch;
122                 }
123         }
124
125         page_name[i] = '\0';
126 }
127
128 INLINE const char *get_ext(const char *name)
129 {
130         const char *ext = strstr(name, ".");
131         if(ext && (ext + 1))
132                 return (ext + 1);
133
134         return NULL;
135 }
136
137 static http_handler_t cgi_search(const char *name,  HttpCGI *table)
138 {
139         if (!table)
140                 return NULL;
141
142         int i = 0;
143         const char *ext = get_ext(name);
144         LOG_INFO("EXT %s\n", ext);
145         while(table[i].name)
146         {
147                 if (ext && table[i].type == CGI_MATCH_EXT)
148                 {
149                         LOG_INFO("Match all ext %s\n", ext);
150                         if (!strcmp(table[i].name, ext))
151                                 break;
152                 }
153                 else if (table[i].type == CGI_MATCH_NAME)
154                 {
155                         LOG_INFO("Match all name %s\n", name);
156                         if (strstr(name, table[i].name) != NULL)
157                                 break;
158                 }
159                 else /* (table[i].type == CGI_MATCH_WORD) */
160                 {
161                         LOG_INFO("Match all word %s\n", name);
162                         if (!strcmp(table[i].name, name))
163                                 break;
164                 }
165
166                 i++;
167         }
168
169         return table[i].handler;
170 }
171
172 static char req_string[80];
173
174 /**
175  * Http polling function.
176  *
177  * Call this functions to process each client connections.
178  *
179  */
180 void http_poll(struct netconn *server)
181 {
182         struct netconn *client;
183         struct netbuf *rx_buf_conn;
184         char *rx_buf;
185         uint16_t len;
186
187         client = netconn_accept(server);
188         if (!client)
189                 return;
190
191         rx_buf_conn = netconn_recv(client);
192         if (rx_buf_conn)
193         {
194                 netbuf_data(rx_buf_conn, (void **)&rx_buf, &len);
195                 if (rx_buf)
196                 {
197                         memset(req_string, 0, sizeof(req_string));
198                         http_getPageName(rx_buf, len, req_string, sizeof(req_string));
199
200                         LOG_INFO("Search %s\n", req_string);
201                         if (req_string[0] == '\0')
202                                 strcpy(req_string, HTTP_DEFAULT_PAGE);
203
204                         http_handler_t cgi = cgi_search(req_string, cgi_table);
205                         if (cgi)
206                         {
207                                 if (cgi(client, req_string, rx_buf, len) < 0)
208                                 {
209                                         LOG_ERR("Internal server error\n");
210                                         http_sendInternalErr(client);
211                                         netconn_write(client, http_server_error, http_server_error_len - 1, NETCONN_NOCOPY);
212                                 }
213                         }
214                         else
215                         {
216                                 http_callback(client, req_string, rx_buf, len);
217                         }
218                 }
219                 netconn_close(client);
220                 netbuf_delete(rx_buf_conn);
221         }
222         netconn_delete(client);
223 }
224
225 /**
226  * Init the http server.
227  *
228  * The simply http server call for each client request the default_callback function. The
229  * user should define this callback to manage the client request, i.e. reading site's page
230  * from SD card. The user can define the cgi_table, where associate one callback to the user string.
231  * In this way the user could filter some client request and redirect they to custom callback, i.e.
232  * the client could request status of the device only loading the particular page name.
233  *
234  * \param default_callback fuction that server call for all request, that does'nt match cgi table.
235  * \param table of callcack to call when client request a particular page.
236  */
237 void http_init(http_handler_t default_callback, struct HttpCGI *table)
238 {
239         ASSERT(default_callback);
240
241         cgi_table = table;
242         http_callback = default_callback;
243 }
244