Merge origin/master
[src/app-framework-binder.git] / src / helper-api.c
1 /*
2  * Copyright (C) 2015 "IoT.bzh"
3  * Author "Fulup Ar Foll"
4  *
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.
9  *
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.
14  *
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/>.
17  * 
18  */
19
20 #include "../include/local-def.h"
21 #include <dirent.h>
22
23
24 // handle to hold queryAll values
25 typedef struct {
26      char    *msg;
27      int     idx;
28      size_t  len;
29 } queryHandleT;
30
31 // Sample Generic Ping Debug API
32 PUBLIC json_object* getPingTest(AFB_request *request) {
33     static pingcount = 0;
34     json_object *response;
35     char query  [256];
36     char session[256];
37     int len;
38     
39     // request all query key/value
40     len = getQueryAll (request, query, sizeof(query));
41     if (len == 0) strncpy (query, "NoSearchQueryList", sizeof(query));
42     
43     // check if we have some post data
44     if (request->post == NULL)  request->post->data="NoData"; 
45           
46     // return response to caller
47     response = jsonNewMessage(AFB_SUCCESS, "Ping Binder Daemon count=%d uuid=%s query={%s} session={0x%x} PostData: [%s] "
48                , pingcount++, request->uuid, query, session, request->post->data);
49     return (response);
50 }
51
52
53 // Helper to retrieve argument from  connection
54 PUBLIC const char* getQueryValue(const AFB_request * request, const char *name) {
55     const char *value;
56
57     value = MHD_lookup_connection_value(request->connection, MHD_GET_ARGUMENT_KIND, name);
58     return (value);
59 }
60
61 STATIC int getQueryCB (void*handle, enum MHD_ValueKind kind, const char *key, const char *value) {
62     queryHandleT *query = (queryHandleT*)handle;
63         
64     query->idx += snprintf (&query->msg[query->idx],query->len," %s: \'%s\',", key, value);
65 }
66
67 // Helper to retrieve argument from  connection
68 PUBLIC int getQueryAll(AFB_request * request, char *buffer, size_t len) {
69     queryHandleT query;
70     buffer[0] = '\0'; // start with an empty string
71     query.msg= buffer;
72     query.len= len;
73     query.idx= 0;
74
75     MHD_get_connection_values (request->connection, MHD_GET_ARGUMENT_KIND, getQueryCB, &query);
76     return (len);
77 }
78
79 // Helper to retrieve POST handle
80 PUBLIC AFB_PostHandle* getPostHandle (AFB_request *request) {
81     if (request->post == NULL) return (NULL);
82     return ((AFB_PostHandle*) request->post->data);
83 }
84
85 // Helper to retrieve POST file context
86 PUBLIC AFB_PostCtx* getPostContext (AFB_request *request) {
87     AFB_PostHandle* postHandle;
88     if (request->post == NULL) return (NULL);
89     
90     postHandle = (AFB_PostHandle*) request->post->data;
91     if (postHandle == NULL) return NULL;
92        
93     return ((AFB_PostCtx*) postHandle->ctx);
94 }
95
96 PUBLIC json_object* getPostFile (AFB_request *request, AFB_PostItem *item, char* destination) {
97
98     AFB_PostHandle *postHandle = getPostHandle(request);
99     AFB_PostCtx *postFileCtx;
100     char filepath[512];
101     int len;
102             
103     // This is called after PostForm and then after DonePostForm
104     if (item == NULL) {
105         json_object* jresp;
106         postFileCtx = (AFB_PostCtx*) postHandle->ctx;
107         
108         // No Post Application Context [something really bad happen]
109         if (postFileCtx == NULL) {
110             request->errcode = MHD_HTTP_EXPECTATION_FAILED;
111             return(jsonNewMessage(AFB_FAIL,"Error: PostForm no PostContext to free\n"));          
112         }
113         
114         // We have a context but last Xform iteration fail or application set a message
115         if (postFileCtx->jresp != NULL) {
116             jresp = postFileCtx->jresp;  // retrieve previous error from postCtx
117             if (postFileCtx->errcode != 0) request->errcode=postFileCtx->errcode;
118         }
119         else jresp = jsonNewMessage(AFB_FAIL,"getPostFile Post Request done");
120         
121         // Error or not let's free all resources
122         close(postFileCtx->fd);
123         free (postFileCtx->path);
124         free (postFileCtx);
125         return (jresp);  
126     }
127     
128     // Make sure it's a valid PostForm request
129     if (!request->post && request->post->type != AFB_POST_FORM) {
130         postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"This is not a valid PostForm request\n");
131         goto ExitOnError;
132     } 
133     
134     // Check this is a file element
135     if (item->filename == NULL) {
136         postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"No Filename attached to key=%s\n", item->key);
137         goto ExitOnError;
138     }
139     
140     // Check we got something in buffer
141     if (item->len <= 0) {       
142         postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Buffer size NULL key=%s]\n", item->key);
143         goto ExitOnError;
144     }
145
146     // Extract Application Context from posthandle [NULL == 1st iteration]    
147     postFileCtx = (AFB_PostCtx*) postHandle->ctx;
148
149     // This is the 1st Item iteration let's open output file and allocate necessary resources
150     if (postFileCtx == NULL)  {
151         DIR* destDir;
152         
153         // Create an application specific context
154         postFileCtx = calloc (1, sizeof(AFB_PostCtx)); // May place anything here until post->completeCB handle resources liberation
155         postFileCtx->path = strdup (filepath);
156         
157         // attach application to postHandle
158         postHandle->ctx = (void*) postFileCtx;   // May place anything here until post->completeCB handle resources liberation  
159         
160         // Build destination directory full path
161         if (destination[0] != '/') {
162            strncpy (filepath, request->config->sessiondir, sizeof(filepath)); 
163            strncat (filepath, "/", sizeof(filepath));
164            strncat (filepath, destination, sizeof(filepath)); 
165         } else strncpy (filepath, destination, sizeof(filepath));
166         
167
168         // make sure destination directory exist
169         destDir = opendir (filepath);
170         if (destDir == NULL) {
171           if ( 0 <= mkdir(filepath,O_RDWR | S_IRWXU | S_IRGRP)) {
172             postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Fail to Create destination directory=[%s] error=%s\n", filepath, strerror(errno));
173             goto ExitOnError;
174           }
175         } else closedir (destDir);
176         
177         strncat (filepath, "/", sizeof(filepath));
178         strncat (filepath, item->filename, sizeof(filepath));  
179
180         if((postFileCtx->fd = open(filepath, O_RDWR |O_CREAT, S_IRWXU|S_IRGRP)) <= 0) {
181             postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Fail to Create destination File=[%s] error=%s\n", filepath, strerror(errno));
182             goto ExitOnError;
183         } 
184     } else {     
185         // reuse existing application context
186         postFileCtx = (AFB_PostCtx*) postHandle->ctx;  
187     } 
188
189     // Check we successfully wrote full buffer
190     len = write (postFileCtx->fd, item->data, item->len);
191     if (item->len != len) {
192         postFileCtx->jresp= jsonNewMessage(AFB_FAIL,"Fail to write file [%s] at [%s] error=\n", item->filename, strerror(errno));
193         goto ExitOnError;
194     }
195   
196     // every intermediary iteration should return Success & NULL
197     request->errcode = MHD_HTTP_OK;
198     return NULL;
199     
200 ExitOnError:    
201     request->errcode = MHD_HTTP_EXPECTATION_FAILED;
202     return NULL;
203 }