Fixed memory handling of HandleSingleResponse
[platform/upstream/iotivity.git] / resource / csdk / stack / src / ocserverrequest.c
1 //******************************************************************
2 //
3 // Copyright 2014 Intel Mobile Communications GmbH All Rights Reserved.
4 //
5 //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
6 //
7 // Licensed under the Apache License, Version 2.0 (the "License");
8 // you may not use this file except in compliance with the License.
9 // You may obtain a copy of the License at
10 //
11 //      http://www.apache.org/licenses/LICENSE-2.0
12 //
13 // Unless required by applicable law or agreed to in writing, software
14 // distributed under the License is distributed on an "AS IS" BASIS,
15 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 // See the License for the specific language governing permissions and
17 // limitations under the License.
18 //
19 //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
20 #include <string.h>
21
22 #include "ocstack.h"
23 #include "ocserverrequest.h"
24 #include "ocresourcehandler.h"
25 #include "ocmalloc.h"
26
27 #include "cacommon.h"
28 #include "cainterface.h"
29
30 #include "utlist.h"
31 #include "pdu.h"
32
33 // Module Name
34 #define VERIFY_NON_NULL(arg) { if (!arg) {OC_LOG(FATAL, TAG, #arg " is NULL"); goto exit;} }
35
36 #define TAG  PCF("ocserverrequest")
37
38 static struct OCServerRequest * serverRequestList = NULL;
39 static struct OCServerResponse * serverResponseList = NULL;
40
41 OCServerRequest * GetServerRequestUsingToken (const CAToken_t token)
42 {
43     if(!token)
44     {
45         OC_LOG(ERROR, TAG, PCF("Invalid Parameter Token"));
46         return NULL;
47     }
48
49     OCServerRequest * out = NULL;
50     LL_FOREACH (serverRequestList, out)
51     {
52         OC_LOG(INFO, TAG,PCF("comparing tokens"));
53         OC_LOG_BUFFER(INFO, TAG, (const uint8_t *)token, CA_MAX_TOKEN_LEN);
54         OC_LOG_BUFFER(INFO, TAG, (const uint8_t *)out->requestToken, CA_MAX_TOKEN_LEN);
55         if(memcmp(out->requestToken, token, CA_MAX_TOKEN_LEN) == 0)
56         {
57             return out;
58         }
59     }
60     OC_LOG(ERROR, TAG, PCF("Server Request not found!!"));
61     return NULL;
62 }
63
64 OCServerRequest * GetServerRequestUsingHandle (const OCServerRequest * handle)
65 {
66     OCServerRequest * out = NULL;
67     LL_FOREACH (serverRequestList, out)
68     {
69         if(out == handle)
70         {
71             return out;
72         }
73     }
74     OC_LOG(ERROR, TAG, PCF("Server Request not found!!"));
75     return NULL;
76 }
77
78 OCServerResponse * GetServerResponseUsingHandle (const OCServerRequest * handle)
79 {
80     OCServerResponse * out = NULL;
81     LL_FOREACH (serverResponseList, out)
82     {
83         if(out->requestHandle == handle)
84         {
85             return out;
86         }
87     }
88     OC_LOG(ERROR, TAG, PCF("Server Response not found!!"));
89     return NULL;
90 }
91
92 OCStackResult AddServerRequest (OCServerRequest ** request, uint16_t coapID,
93         uint8_t delayedResNeeded, uint8_t secured, uint8_t notificationFlag, OCMethod method,
94         uint8_t numRcvdVendorSpecificHeaderOptions, uint32_t observationOption,
95         OCQualityOfService qos, char * query,
96         OCHeaderOption * rcvdVendorSpecificHeaderOptions,
97         char * reqJSONPayload, CAToken_t * requestToken,
98         char * resourceUrl, size_t reqTotalSize,
99         CAAddress_t *addressInfo, CAConnectivityType_t connectivityType)
100 {
101     OCServerRequest * serverRequest = NULL;
102
103     //Note: OCServerRequest includes 1 byte for the JSON Payload.  payloadSize is calculated
104     //as the required length of the string, so this will result in enough room for the
105     //null terminator as well.
106     serverRequest = (OCServerRequest *) OCCalloc(1, sizeof(OCServerRequest) +
107         (reqTotalSize ? reqTotalSize : 1) - 1);
108     VERIFY_NON_NULL(serverRequest);
109
110     serverRequest->coapID = coapID;
111     serverRequest->delayedResNeeded = delayedResNeeded;
112     serverRequest->secured = secured;
113     serverRequest->notificationFlag = notificationFlag;
114
115     serverRequest->method = method;
116     serverRequest->numRcvdVendorSpecificHeaderOptions = numRcvdVendorSpecificHeaderOptions;
117     serverRequest->observationOption = observationOption;
118     serverRequest->observeResult = OC_STACK_ERROR;
119     serverRequest->qos = qos;
120     serverRequest->ehResponseHandler = HandleSingleResponse;
121     serverRequest->numResponses = 1;
122
123     if(query)
124     {
125         strncpy((char*)serverRequest->query,
126                 (const char*)query, sizeof(serverRequest->query) - 1);
127     }
128
129     if(rcvdVendorSpecificHeaderOptions)
130     {
131         memcpy(serverRequest->rcvdVendorSpecificHeaderOptions, rcvdVendorSpecificHeaderOptions,
132             MAX_HEADER_OPTIONS * sizeof(OCHeaderOption));
133     }
134     if(reqJSONPayload && reqTotalSize)
135     {
136         // destination is at least 1 greater than the source, so a NULL always exists in the
137         // last character
138         strncpy((char*)serverRequest->reqJSONPayload,
139                 (const char*)reqJSONPayload, reqTotalSize - 1);
140     }
141     serverRequest->requestComplete = 0;
142     if(requestToken)
143     {
144         serverRequest->requestToken = (CAToken_t)OCMalloc(CA_MAX_TOKEN_LEN+1);
145         VERIFY_NON_NULL (serverRequest->requestToken);
146         memcpy(serverRequest->requestToken, *requestToken, CA_MAX_TOKEN_LEN);
147         serverRequest->requestToken[CA_MAX_TOKEN_LEN]='\0';
148     }
149
150     if(resourceUrl)
151     {
152         strncpy((char*)serverRequest->resourceUrl,
153                 (const char*)resourceUrl, sizeof(serverRequest->resourceUrl) - 1);
154     }
155
156     if (addressInfo)
157     {
158         serverRequest->addressInfo = *addressInfo;
159     }
160     serverRequest->connectivityType = connectivityType;
161
162     *request = serverRequest;
163     OC_LOG(INFO, TAG, PCF("Server Request Added!!"));
164     LL_APPEND (serverRequestList, serverRequest);
165     return OC_STACK_OK;
166
167 exit:
168     if (serverRequest)
169     {
170         OCFree(serverRequest);
171         serverRequest = NULL;
172     }
173     *request = NULL;
174     return OC_STACK_NO_MEMORY;
175 }
176
177 OCStackResult AddServerResponse (OCServerResponse ** response, OCRequestHandle requestHandle)
178 {
179     OCServerResponse * serverResponse = NULL;
180
181     serverResponse = (OCServerResponse *) OCCalloc(1, sizeof(OCServerResponse));
182     VERIFY_NON_NULL(serverResponse);
183
184     serverResponse->payload = (char *) OCCalloc(1, MAX_RESPONSE_LENGTH);
185     VERIFY_NON_NULL(serverResponse->payload);
186
187     serverResponse->remainingPayloadSize = MAX_RESPONSE_LENGTH;
188     serverResponse->requestHandle = requestHandle;
189
190     *response = serverResponse;
191     OC_LOG(INFO, TAG, PCF("Server Response Added!!"));
192     LL_APPEND (serverResponseList, serverResponse);
193     return OC_STACK_OK;
194
195 exit:
196     if (serverResponse)
197     {
198         OCFree(serverResponse);
199         serverResponse = NULL;
200     }
201     *response = NULL;
202     return OC_STACK_NO_MEMORY;
203 }
204
205 // Form the OCEntityHandlerRequest struct
206 OCStackResult FormOCEntityHandlerRequest(
207         OCEntityHandlerRequest * entityHandlerRequest,
208         OCRequestHandle request,
209         OCMethod method,
210         OCResourceHandle resource,
211         char * queryBuf,
212         char * bufReqPayload,
213         uint8_t numVendorOptions,
214         OCHeaderOption * vendorOptions,
215         OCObserveAction observeAction,
216         OCObservationId observeID)
217 {
218     if (entityHandlerRequest)
219     {
220         memset(entityHandlerRequest, 0, sizeof(OCEntityHandlerRequest));
221         entityHandlerRequest->requestHandle = request;
222         entityHandlerRequest->method = method;
223         entityHandlerRequest->resource = (OCResourceHandle) resource;
224         entityHandlerRequest->query = queryBuf;
225         entityHandlerRequest->reqJSONPayload = bufReqPayload;
226         entityHandlerRequest->numRcvdVendorSpecificHeaderOptions = numVendorOptions;
227         entityHandlerRequest->rcvdVendorSpecificHeaderOptions = vendorOptions;
228
229         entityHandlerRequest->obsInfo.action = observeAction;
230         entityHandlerRequest->obsInfo.obsId = observeID;
231         return OC_STACK_OK;
232     }
233
234     return OC_STACK_INVALID_PARAM;
235 }
236
237 void FindAndDeleteServerResponse(OCServerResponse * serverResponse)
238 {
239     OCServerResponse* tmp;
240     if(serverResponse)
241     {
242         LL_FOREACH(serverResponseList, tmp)
243         {
244             if (serverResponse == tmp)
245             {
246                 DeleteServerResponse(tmp);
247                 return;
248             }
249         }
250     }
251 }
252
253 void DeleteServerResponse(OCServerResponse * serverResponse)
254 {
255     if(serverResponse) {
256         LL_DELETE(serverResponseList, serverResponse);
257         OCFree(serverResponse->payload);
258         OCFree(serverResponse);
259         OC_LOG(INFO, TAG, PCF("Server Response Removed!!"));
260     }
261 }
262
263 void FindAndDeleteServerRequest(OCServerRequest * serverRequest)
264 {
265     OCServerRequest* tmp;
266     if(serverRequest)
267     {
268         LL_FOREACH(serverRequestList, tmp)
269         {
270             if (serverRequest == tmp)
271             {
272                 DeleteServerRequest(tmp);
273                 return;
274             }
275         }
276     }
277 }
278
279 void DeleteServerRequest(OCServerRequest * serverRequest)
280 {
281     if(serverRequest) {
282         LL_DELETE(serverRequestList, serverRequest);
283         OCFree(serverRequest);
284         serverRequest = NULL;
285         OC_LOG(INFO, TAG, PCF("Server Request Removed!!"));
286     }
287 }
288
289 OCStackResult HandleSingleResponse(OCEntityHandlerResponse * ehResponse)
290 {
291     OCStackResult result = OC_STACK_ERROR;
292     CARemoteEndpoint_t responseEndpoint = {};
293     CAResponseInfo_t responseInfo = {};
294     CAHeaderOption_t* optionsPointer = NULL;
295
296     OC_LOG_V(INFO, TAG, "Inside HandleSingleResponse: %s", ehResponse->payload);
297
298     OCServerRequest *serverRequest = (OCServerRequest *)ehResponse->requestHandle;
299
300     // Copy the address
301     responseEndpoint.resourceUri      = (CAURI_t) serverRequest->resourceUrl;
302     responseEndpoint.addressInfo      = serverRequest->addressInfo;
303     responseEndpoint.connectivityType = serverRequest->connectivityType;
304     responseEndpoint.isSecured        = serverRequest->secured;
305     // Copy the info
306     switch (ehResponse->ehResult)
307     {
308         case OC_EH_OK:
309             responseInfo.result = CA_SUCCESS;
310             break;
311         case OC_EH_ERROR:
312             responseInfo.result = CA_BAD_REQ;
313             break;
314         case OC_EH_RESOURCE_CREATED:
315             responseInfo.result = CA_CREATED;
316             break;
317         case OC_EH_RESOURCE_DELETED:
318             responseInfo.result = CA_DELETED;
319             break;
320         case OC_EH_SLOW:
321             responseInfo.result = CA_SUCCESS;
322             break;
323         case OC_EH_FORBIDDEN:
324             responseInfo.result = CA_BAD_REQ;
325             break;
326         default:
327             responseInfo.result = CA_BAD_REQ;
328             break;
329     }
330     responseInfo.info.type = qualityOfServiceToMessageType(serverRequest->qos);
331     char token[CA_MAX_TOKEN_LEN + 1] = {};
332     responseInfo.info.token = token;
333     memcpy(responseInfo.info.token, serverRequest->requestToken, CA_MAX_TOKEN_LEN);
334
335     if(serverRequest->observeResult == OC_STACK_OK)
336     {
337         responseInfo.info.numOptions = ehResponse->numSendVendorSpecificHeaderOptions + 1;
338     }
339     else
340     {
341         responseInfo.info.numOptions = ehResponse->numSendVendorSpecificHeaderOptions;
342     }
343
344     if(responseInfo.info.numOptions > 0)
345     {
346         responseInfo.info.options = (CAHeaderOption_t *)
347                                       OCCalloc(responseInfo.info.numOptions,
348                                               sizeof(CAHeaderOption_t));
349
350         if(!responseInfo.info.options)
351         {
352             OC_LOG(FATAL, TAG, PCF("options is NULL"));
353             return OC_STACK_NO_MEMORY;
354         }
355
356         optionsPointer = responseInfo.info.options;
357
358         // TODO: This exposes CoAP specific details.  At some point, this should be
359         // re-factored and handled in the CA layer.
360         if(serverRequest->observeResult == OC_STACK_OK)
361         {
362             responseInfo.info.options[0].protocolID = CA_COAP_ID;
363             responseInfo.info.options[0].optionID = COAP_OPTION_OBSERVE;
364             responseInfo.info.options[0].optionLength = sizeof(uint32_t);
365             memcpy(responseInfo.info.options[0].optionData,
366                     &(serverRequest->observationOption), sizeof(uint32_t));
367
368             // Point to the next header option before copying vender specific header options
369             optionsPointer += 1;
370         }
371
372         if (ehResponse->numSendVendorSpecificHeaderOptions)
373         {
374             memcpy(optionsPointer, ehResponse->sendVendorSpecificHeaderOptions,
375                             sizeof(OCHeaderOption) *
376                             ehResponse->numSendVendorSpecificHeaderOptions);
377         }
378     }
379     else
380     {
381         responseInfo.info.options = NULL;
382     }
383
384     char payload[MAX_RESPONSE_LENGTH] = {};
385
386     // Put the JSON prefix and suffix around the payload
387     strcpy(payload, (const char *)OC_JSON_PREFIX);
388     strcat(payload, (const char *)ehResponse->payload);
389     strcat(payload, (const char *)OC_JSON_SUFFIX);
390     responseInfo.info.payload = (CAPayload_t)payload;
391
392     #ifdef WITH_PRESENCE
393     //TODO: Add other connectivity types to CAConnTypes[] when enabled
394     CAConnectivityType_t CAConnTypes[] = {CA_ETHERNET, CA_WIFI};
395     const char * connTypes[] = {"ethernet", "wifi"};
396     int size = sizeof(CAConnTypes)/ sizeof(CAConnectivityType_t);
397     CAConnectivityType_t connType = responseEndpoint.connectivityType;
398     CAResult_t caResult = CA_STATUS_FAILED;
399     result = OC_STACK_OK;
400
401     //Sending response on all n/w interfaces
402     for(int i = 0; i < size; i++ )
403     {
404         responseEndpoint.connectivityType = (CAConnectivityType_t)(connType & CAConnTypes[i]);
405         if(responseEndpoint.connectivityType)
406         {
407             //The result is set to OC_STACK_OK only if CASendResponse succeeds in sending the
408             //response on all the n/w interfaces else it is set to OC_STACK_ERROR
409             caResult = CASendResponse(&responseEndpoint, &responseInfo);
410             if(caResult != CA_STATUS_OK)
411             {
412                 OC_LOG_V(ERROR, TAG, "CASendResponse failed on %s", connTypes[i]);
413                 result = CAResultToOCResult(caResult);
414             }
415             else
416             {
417                 OC_LOG_V(INFO, TAG, "CASendResponse succeeded on %s", connTypes[i]);
418             }
419         }
420     }
421     #else
422     CAResult_t caResult = CASendResponse(&responseEndpoint, &responseInfo);
423     if(caResult != CA_STATUS_OK)
424     {
425         OC_LOG(ERROR, TAG, PCF("CASendResponse failed"));
426         result = CAResultToOCResult(caResult);
427     }
428     else
429     {
430         result = OC_STACK_OK;
431     }
432     #endif
433
434     OCFree(responseInfo.info.options);
435     //Delete the request
436     FindAndDeleteServerRequest(serverRequest);
437     return result;
438 }
439
440 OCStackResult HandleAggregateResponse(OCEntityHandlerResponse * ehResponse)
441 {
442     OCStackResult stackRet = OC_STACK_ERROR;
443     OCServerRequest * serverRequest = NULL;
444     OCServerResponse * serverResponse = NULL;
445     uint16_t bufferNeeded = 0;
446
447     if(!ehResponse || !ehResponse->payload)
448     {
449         OC_LOG(ERROR, TAG, PCF("HandleAggregateResponse invalid parameters"));
450         return OC_STACK_INVALID_PARAM;
451     }
452
453     OC_LOG_V(INFO, TAG, "Inside HandleAggregateResponse: %s", ehResponse->payload);
454
455     serverRequest = GetServerRequestUsingHandle((OCServerRequest *)ehResponse->requestHandle);
456     serverResponse = GetServerResponseUsingHandle((OCServerRequest *)ehResponse->requestHandle);
457
458     if(serverRequest)
459     {
460         if(!serverResponse)
461         {
462             OC_LOG(INFO, TAG, PCF("This is the first response fragment"));
463             stackRet = AddServerResponse(&serverResponse, ehResponse->requestHandle);
464             if (OC_STACK_OK != stackRet)
465             {
466                 OC_LOG(ERROR, TAG, PCF("Error adding server response"));
467                 return stackRet;
468             }
469             VERIFY_NON_NULL(serverResponse);
470             VERIFY_NON_NULL(serverResponse->payload);
471         }
472
473         // If there is more than 1 response, then we need to allow for a null-termination
474         // in the server response payload buffer AND the JSON response separator
475         bufferNeeded = ehResponse->payloadSize + 1;
476         if (serverRequest->numResponses > 1)
477         {
478             bufferNeeded += strlen(OC_JSON_SEPARATOR_STR);
479         }
480         if(serverResponse->remainingPayloadSize >= bufferNeeded)
481         {
482             OC_LOG(ERROR, TAG, PCF("There is room in response buffer"));
483             // append
484             strncat((char *)serverResponse->payload,
485                     (char *)ehResponse->payload,
486                     serverResponse->remainingPayloadSize);
487             OC_LOG_V(INFO, TAG, "Current aggregated response  ...%s", serverResponse->payload);
488             serverResponse->remainingPayloadSize -= strlen((char *)ehResponse->payload);
489             (serverRequest->numResponses)--;
490             if(serverRequest->numResponses == 0)
491             {
492                 OC_LOG(INFO, TAG, PCF("This is the last response fragment"));
493                 ehResponse->payload = serverResponse->payload;
494                 ehResponse->payloadSize = strlen((char *) serverResponse->payload) + 1;
495                 stackRet = HandleSingleResponse(ehResponse);
496                 //Delete the request and response
497                 FindAndDeleteServerRequest(serverRequest);
498                 FindAndDeleteServerResponse(serverResponse);
499             }
500             else
501             {
502                 OC_LOG(INFO, TAG, PCF("More response fragments to come"));
503                 strncat((char *)serverResponse->payload,
504                         OC_JSON_SEPARATOR_STR,
505                         serverResponse->remainingPayloadSize);
506                 OC_LOG_V(INFO, TAG, "Current aggregated response  ...%s", serverResponse->payload);
507                 serverResponse->remainingPayloadSize -= strlen(OC_JSON_SEPARATOR_STR);
508                 stackRet = OC_STACK_OK;
509             }
510         }
511         else
512         {
513             OC_LOG(ERROR, TAG, PCF("No room in response buffer"));
514             //Delete the request and response
515             FindAndDeleteServerRequest(serverRequest);
516             FindAndDeleteServerResponse(serverResponse);
517             stackRet = OC_STACK_NO_MEMORY;
518         }
519     }
520 exit:
521     return stackRet;
522 }
523