2 * Copyright (C) 2015 "IoT.bzh"
3 * Author "Fulup Ar Foll"
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
29 #include <sys/types.h>
32 #include "local-def.h"
33 #include "afb-req-itf.h"
35 // handle to hold queryAll values
42 // Error code are requested through function to manage json usage count
49 static AFB_errorT AFBerr [AFB_UNAUTH+1];
50 static json_object *jTypeStatic;
54 static const char *ERROR_LABEL[] = {"false", "true", "fatal", "fail", "warning", "empty", "success", "done", "unauth"};
57 // Helper to retrieve argument from connection
58 const char* getQueryValue(const AFB_request * request, const char *name) {
59 return afb_req_argument(*request->areq, name);
62 static int getQueryCB (queryHandleT *query, struct afb_arg arg) {
63 if (query->idx >= query->len)
65 query->idx += (unsigned)snprintf (&query->msg[query->idx], query->len-query->idx, " %s: %s\'%s\',", arg.name, arg.is_file?"FILE=":"", arg.value);
66 return 1; /* continue to iterate */
69 // Helper to retrieve argument from connection
70 size_t getQueryAll(AFB_request * request, char *buffer, size_t len) {
72 buffer[0] = '\0'; // start with an empty string
77 afb_req_iterate(*request->areq, (void*)getQueryCB, &query);
79 return query.idx >= len ? len - 1 : query.idx;
83 char* getPostPath (AFB_request *request) {
84 AFB_PostHandle *postHandle = getPostHandle(request);
85 AFB_PostCtx *postFileCtx;
87 if (postHandle == NULL) return NULL;
89 postFileCtx = (AFB_PostCtx*) postHandle->ctx;
90 if (postFileCtx == NULL) return NULL;
92 return (postFileCtx->path);
95 json_object* getPostFile (AFB_request *request, AFB_PostItem *item, char* destination) {
97 AFB_PostHandle *postHandle = getPostHandle(request);
98 AFB_PostCtx *postFileCtx;
102 // This is called after PostForm and then after DonePostForm
105 postFileCtx = (AFB_PostCtx*) postHandle->ctx;
107 // No Post Application Context [something really bad happen]
108 if (postFileCtx == NULL) {
109 request->errcode = MHD_HTTP_EXPECTATION_FAILED;
110 return(jsonNewMessage(AFB_FAIL,"Error: PostForm no PostContext to free\n"));
113 // We have a context but last Xform iteration fail or application set a message
114 if (request->jresp != NULL) {
115 jresp = request->jresp; // retrieve previous error from postCtx
116 } else jresp = jsonNewMessage(AFB_SUCCESS,"getPostFile Post Request done");
118 // Error or not let's free all resources
119 close(postFileCtx->fd);
120 free (postFileCtx->path);
124 #if defined(PLEASE_FIX_ME_THE_ERROR_IS_postFileCtx_NOT_INITIALIZED)
125 // Make sure it's a valid PostForm request
126 if (!request->post && request->post->type != AFB_POST_FORM) {
127 postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"This is not a valid PostForm request\n");
131 // Check this is a file element
132 if (item->filename == NULL) {
133 postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"No Filename attached to key=%s\n", item->key);
137 // Check we got something in buffer
138 if (item->len <= 0) {
139 postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Buffer size NULL key=%s]\n", item->key);
143 // Extract Application Context from posthandle [NULL == 1st iteration]
144 postFileCtx = (AFB_PostCtx*) postHandle->ctx;
146 // This is the 1st Item iteration let's open output file and allocate necessary resources
147 if (postFileCtx == NULL) {
150 // Create an application specific context
151 postFileCtx = calloc (1, sizeof(AFB_PostCtx)); // May place anything here until post->completeCB handle resources liberation
153 // attach application to postHandle
154 postHandle->ctx = (void*) postFileCtx; // May place anything here until post->completeCB handle resources liberation
156 // Build destination directory full path
157 if (destination[0] != '/') {
158 strncpy (filepath, request->config->sessiondir, sizeof(filepath));
159 strncat (filepath, "/", sizeof(filepath));
160 strncat (filepath, destination, sizeof(filepath));
161 } else strncpy (filepath, destination, sizeof(filepath));
164 // make sure destination directory exist
165 destDir = opendir (filepath);
166 if (destDir == NULL) {
167 if (mkdir(filepath,O_RDWR | S_IRWXU | S_IRGRP) < 0) {
168 postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Fail to Create destination directory=[%s] error=%s\n", filepath, strerror(errno));
171 } else closedir (destDir);
173 strncat (filepath, "/", sizeof(filepath));
174 strncat (filepath, item->filename, sizeof(filepath));
176 postFileCtx->path = strdup (filepath);
177 if (verbose) fprintf(stderr, "getPostFile path=%s\n", filepath);
179 if((postFileCtx->fd = open(filepath, O_RDWR |O_CREAT, S_IRWXU|S_IRGRP)) <= 0) {
180 postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Fail to Create destination File=[%s] error=%s\n", filepath, strerror(errno));
184 // reuse existing application context
185 postFileCtx = (AFB_PostCtx*) postHandle->ctx;
188 // Check we successfully wrote full buffer
189 len = write (postFileCtx->fd, item->data, item->len);
190 if ((ssize_t)item->len != len) {
191 postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Fail to write file [%s] at [%s] error=\n", item->filename, strerror(errno));
195 // every intermediary iteration should return Success & NULL
196 request->errcode = MHD_HTTP_OK;
200 request->errcode = MHD_HTTP_EXPECTATION_FAILED;
206 static void jsoninit()
213 // initialise JSON constant messages and increase reference count to make them permanent
214 verbosesav = verbose;
215 verbose = 0; // run initialisation in silent mode
216 jTypeStatic = json_object_new_string ("AFB_message");
217 for (idx = 0; idx <= AFB_UNAUTH; idx++) {
218 AFBerr[idx].level = idx;
219 AFBerr[idx].label = ERROR_LABEL [idx];
220 AFBerr[idx].json = jsonNewMessage (idx, NULL);
222 verbose = verbosesav;
226 // build an ERROR message and return it as a valid json object
227 json_object *json_add_status (json_object *obj, const char *status, const char *info)
230 obj = json_object_new_object();
231 json_object_object_add(obj, "status", json_object_new_string(status));
233 json_object_object_add(obj, "info", json_object_new_string(info));
237 // build an ERROR message and return it as a valid json object
238 json_object *json_add_status_v (json_object *obj, const char *status, const char *info, va_list args)
241 if (info == NULL || vasprintf(&message, info, args) < 0)
243 obj = json_add_status(obj, status, message);
249 // build an ERROR message and return it as a valid json object
250 json_object *json_add_status_f (json_object *obj, const char *status, const char *info, ...)
253 va_start(args, info);
254 obj = json_add_status_v(obj, status, info, args);
261 // build an ERROR message and return it as a valid json object
262 struct json_object *jsonNewMessage (AFB_error level, char* format, ...) {
263 static int count = 0;
264 json_object * AFBResponse;
271 if (format != NULL) {
272 va_start(args, format);
273 vsnprintf (message, sizeof (message), format, args);
277 AFBResponse = json_object_new_object();
278 json_object_object_add (AFBResponse, "jtype", json_object_get (jTypeStatic));
279 json_object_object_add (AFBResponse, "status" , json_object_new_string (ERROR_LABEL[level]));
280 if (format != NULL) {
281 json_object_object_add (AFBResponse, "info" , json_object_new_string (message));
284 fprintf (stderr, "AFB:%-6s [%3d]: ", AFBerr [level].label, count++);
285 if (format != NULL) {
286 fprintf (stderr, "%s", message);
288 fprintf (stderr, "No Message");
290 fprintf (stderr, "\n");
293 return (AFBResponse);
303 status: "", /* exist, fail, empty, null, processed */