8d91882e726e8298f331389f3c219f922eeda7da
[iotivity.git] / resource / csdk / security / provisioning / sample / sampleserver_preconfpin.cpp
1 /******************************************************************
2 *
3 * Copyright 2015 Samsung Electronics 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 ///////////////////////////////////////////////////////////////////////
21 //NOTE :  This sample server is generated based on ocserverbasicops.cpp
22 ///////////////////////////////////////////////////////////////////////
23 #include "iotivity_config.h"
24 #include <stdio.h>
25 #include <string.h>
26 #include <stdlib.h>
27 #ifdef HAVE_UNISTD_H
28 #include <unistd.h>
29 #endif
30 #ifdef HAVE_PTHREAD_H
31 #include <pthread.h>
32 #endif
33 #include <signal.h>
34 #include "ocstack.h"
35 #include "ocpayload.h"
36 #include "pinoxmcommon.h"
37
38 #ifdef HAVE_WINDOWS_H
39 #include <windows.h>
40 /** @todo stop-gap for naming issue. Windows.h does not like us to use ERROR */
41 #ifdef ERROR
42 #undef ERROR
43 #endif //ERROR
44 #endif //HAVE_WINDOWS_H
45 #include "platform_features.h"
46 #include "logger.h"
47
48
49 #define TAG "SAMPLE_PRECONF_PIN"
50
51 int gQuitFlag = 0;
52
53 /* Structure to represent a LED resource */
54 typedef struct LEDRESOURCE{
55     OCResourceHandle handle;
56     bool state;
57     int power;
58 } LEDResource;
59
60 static LEDResource LED;
61 // This variable determines instance number of the LED resource.
62 // Used by POST method to create a new instance of LED resource.
63 static int gCurrLedInstance = 0;
64 #define SAMPLE_MAX_NUM_POST_INSTANCE  2
65 static LEDResource gLedInstance[SAMPLE_MAX_NUM_POST_INSTANCE];
66
67 char *gResourceUri= (char *)"/a/led";
68
69 //Secure Virtual Resource database for Iotivity Server
70 //It contains Server's Identity and the PSK credentials
71 //of other devices which the server trusts
72 static char CRED_FILE[] = "oic_svr_db_server_preconfpin.dat";
73
74 /* Function that creates a new LED resource by calling the
75  * OCCreateResource() method.
76  */
77 int createLEDResource (char *uri, LEDResource *ledResource, bool resourceState, int resourcePower);
78
79 /* This method converts the payload to JSON format */
80 OCRepPayload* constructResponse (OCEntityHandlerRequest *ehRequest);
81
82 /* Following methods process the PUT, GET, POST
83  * requests
84  */
85 OCEntityHandlerResult ProcessGetRequest (OCEntityHandlerRequest *ehRequest,
86                                          OCRepPayload **payload);
87 OCEntityHandlerResult ProcessPutRequest (OCEntityHandlerRequest *ehRequest,
88                                          OCRepPayload **payload);
89 OCEntityHandlerResult ProcessPostRequest (OCEntityHandlerRequest *ehRequest,
90                                         OCEntityHandlerResponse *response,
91                                         OCRepPayload **payload);
92
93 /* Entity Handler callback functions */
94 OCEntityHandlerResult
95 OCEntityHandlerCb (OCEntityHandlerFlag flag,
96         OCEntityHandlerRequest *entityHandlerRequest,
97         void* callbackParam);
98
99 const char *getResult(OCStackResult result) {
100     switch (result) {
101     case OC_STACK_OK:
102         return "OC_STACK_OK";
103     case OC_STACK_RESOURCE_CREATED:
104         return "OC_STACK_RESOURCE_CREATED";
105     case OC_STACK_RESOURCE_DELETED:
106         return "OC_STACK_RESOURCE_DELETED";
107     case OC_STACK_INVALID_URI:
108         return "OC_STACK_INVALID_URI";
109     case OC_STACK_INVALID_QUERY:
110         return "OC_STACK_INVALID_QUERY";
111     case OC_STACK_INVALID_IP:
112         return "OC_STACK_INVALID_IP";
113     case OC_STACK_INVALID_PORT:
114         return "OC_STACK_INVALID_PORT";
115     case OC_STACK_INVALID_CALLBACK:
116         return "OC_STACK_INVALID_CALLBACK";
117     case OC_STACK_INVALID_METHOD:
118         return "OC_STACK_INVALID_METHOD";
119     case OC_STACK_NO_MEMORY:
120         return "OC_STACK_NO_MEMORY";
121     case OC_STACK_COMM_ERROR:
122         return "OC_STACK_COMM_ERROR";
123     case OC_STACK_INVALID_PARAM:
124         return "OC_STACK_INVALID_PARAM";
125     case OC_STACK_NOTIMPL:
126         return "OC_STACK_NOTIMPL";
127     case OC_STACK_NO_RESOURCE:
128         return "OC_STACK_NO_RESOURCE";
129     case OC_STACK_RESOURCE_ERROR:
130         return "OC_STACK_RESOURCE_ERROR";
131     case OC_STACK_SLOW_RESOURCE:
132         return "OC_STACK_SLOW_RESOURCE";
133     case OC_STACK_NO_OBSERVERS:
134         return "OC_STACK_NO_OBSERVERS";
135     #ifdef WITH_PRESENCE
136     case OC_STACK_PRESENCE_STOPPED:
137         return "OC_STACK_PRESENCE_STOPPED";
138     #endif
139     case OC_STACK_ERROR:
140         return "OC_STACK_ERROR";
141     default:
142         return "UNKNOWN";
143     }
144 }
145
146 OCRepPayload* getPayload(const char* uri, int64_t power, bool state)
147 {
148     OCRepPayload* payload = OCRepPayloadCreate();
149     if(!payload)
150     {
151         OIC_LOG(ERROR, TAG, "Failed to allocate Payload");
152         return NULL;
153     }
154
155     OCRepPayloadSetUri(payload, uri);
156     OCRepPayloadSetPropBool(payload, "state", state);
157     OCRepPayloadSetPropInt(payload, "power", power);
158
159     return payload;
160 }
161
162 //This function takes the request as an input and returns the response
163 OCRepPayload* constructResponse (OCEntityHandlerRequest *ehRequest)
164 {
165     if(ehRequest->payload && ehRequest->payload->type != PAYLOAD_TYPE_REPRESENTATION)
166     {
167         OIC_LOG(ERROR, TAG, "Incoming payload not a representation");
168         return NULL;
169     }
170
171     OCRepPayload* input = (OCRepPayload*)(ehRequest->payload);
172
173     LEDResource *currLEDResource = &LED;
174
175     if (ehRequest->resource == gLedInstance[0].handle)
176     {
177         currLEDResource = &gLedInstance[0];
178         gResourceUri = (char *) "/a/led/0";
179     }
180     else if (ehRequest->resource == gLedInstance[1].handle)
181     {
182         currLEDResource = &gLedInstance[1];
183         gResourceUri = (char *) "/a/led/1";
184     }
185
186     if(OC_REST_PUT == ehRequest->method)
187     {
188         // Get pointer to query
189         int64_t pow;
190         if(OCRepPayloadGetPropInt(input, "power", &pow))
191         {
192             currLEDResource->power =pow;
193         }
194
195         bool state;
196         if(OCRepPayloadGetPropBool(input, "state", &state))
197         {
198             currLEDResource->state = state;
199         }
200     }
201
202     return getPayload(gResourceUri, currLEDResource->power, currLEDResource->state);
203 }
204
205 OCEntityHandlerResult ProcessGetRequest (OCEntityHandlerRequest *ehRequest,
206         OCRepPayload **payload)
207 {
208     OCEntityHandlerResult ehResult;
209
210     OCRepPayload *getResp = constructResponse(ehRequest);
211
212     if(getResp)
213     {
214         *payload = getResp;
215         ehResult = OC_EH_OK;
216     }
217     else
218     {
219         ehResult = OC_EH_ERROR;
220     }
221
222     return ehResult;
223 }
224
225 OCEntityHandlerResult ProcessPutRequest (OCEntityHandlerRequest *ehRequest,
226         OCRepPayload **payload)
227 {
228     OCEntityHandlerResult ehResult;
229
230     OCRepPayload *putResp = constructResponse(ehRequest);
231
232     if(putResp)
233     {
234         *payload = putResp;
235         ehResult = OC_EH_OK;
236     }
237     else
238     {
239         ehResult = OC_EH_ERROR;
240     }
241
242     return ehResult;
243 }
244
245 OCEntityHandlerResult ProcessPostRequest (OCEntityHandlerRequest *ehRequest,
246         OCEntityHandlerResponse *response, OCRepPayload **payload)
247 {
248     OCRepPayload *respPLPost_led = NULL;
249     OCEntityHandlerResult ehResult = OC_EH_OK;
250
251     /*
252      * The entity handler determines how to process a POST request.
253      * Per the REST paradigm, POST can also be used to update representation of existing
254      * resource or create a new resource.
255      * In the sample below, if the POST is for /a/led then a new instance of the LED
256      * resource is created with default representation (if representation is included in
257      * POST payload it can be used as initial values) as long as the instance is
258      * lesser than max new instance count. Once max instance count is reached, POST on
259      * /a/led updated the representation of /a/led (just like PUT)
260      */
261
262     if (ehRequest->resource == LED.handle)
263     {
264         if (gCurrLedInstance < SAMPLE_MAX_NUM_POST_INSTANCE)
265         {
266             // Create new LED instance
267             char newLedUri[15] = "/a/led/";
268             int newLedUriLength = strlen(newLedUri);
269             snprintf (newLedUri + newLedUriLength, sizeof(newLedUri)-newLedUriLength, "%d", gCurrLedInstance);
270
271             respPLPost_led = OCRepPayloadCreate();
272             OCRepPayloadSetUri(respPLPost_led, gResourceUri);
273             OCRepPayloadSetPropString(respPLPost_led, "createduri", newLedUri);
274
275             if (0 == createLEDResource (newLedUri, &gLedInstance[gCurrLedInstance], false, 0))
276             {
277                 OIC_LOG (INFO, TAG, "Created new LED instance");
278                 gLedInstance[gCurrLedInstance].state = 0;
279                 gLedInstance[gCurrLedInstance].power = 0;
280                 gCurrLedInstance++;
281                 strncpy ((char *)response->resourceUri, newLedUri, sizeof(response->resourceUri));
282                 ehResult = OC_EH_RESOURCE_CREATED;
283             }
284         }
285         else
286         {
287             respPLPost_led = constructResponse(ehRequest);
288         }
289     }
290     else
291     {
292         for (int i = 0; i < SAMPLE_MAX_NUM_POST_INSTANCE; i++)
293         {
294             if (ehRequest->resource == gLedInstance[i].handle)
295             {
296                 if (i == 0)
297                 {
298                     respPLPost_led = constructResponse(ehRequest);
299                     break;
300                 }
301                 else if (i == 1)
302                 {
303                     respPLPost_led = constructResponse(ehRequest);
304                 }
305             }
306         }
307     }
308
309     if (respPLPost_led != NULL)
310     {
311         *payload = respPLPost_led;
312         ehResult = OC_EH_OK;
313     }
314     else
315     {
316         OIC_LOG_V (INFO, TAG, "Payload was NULL");
317         ehResult = OC_EH_ERROR;
318     }
319
320     return ehResult;
321 }
322
323 OCEntityHandlerResult
324 OCEntityHandlerCb (OCEntityHandlerFlag flag,
325         OCEntityHandlerRequest *entityHandlerRequest,
326         void* callbackParam)
327 {
328     OIC_LOG_V (INFO, TAG, "Inside entity handler - flags: 0x%x", flag);
329     (void)callbackParam;
330     OCEntityHandlerResult ehResult = OC_EH_ERROR;
331
332     OCEntityHandlerResponse response;
333     memset(&response, 0, sizeof(response));
334
335     // Validate pointer
336     if (!entityHandlerRequest)
337     {
338         OIC_LOG (ERROR, TAG, "Invalid request pointer");
339         return OC_EH_ERROR;
340     }
341
342     OCRepPayload* payload = NULL;
343
344     if (flag & OC_REQUEST_FLAG)
345     {
346         OIC_LOG (INFO, TAG, "Flag includes OC_REQUEST_FLAG");
347         if (entityHandlerRequest)
348         {
349             if (OC_REST_GET == entityHandlerRequest->method)
350             {
351                 OIC_LOG (INFO, TAG, "Received OC_REST_GET from client");
352                 ehResult = ProcessGetRequest (entityHandlerRequest, &payload);
353             }
354             else if (OC_REST_PUT == entityHandlerRequest->method)
355             {
356                 OIC_LOG (INFO, TAG, "Received OC_REST_PUT from client");
357                 ehResult = ProcessPutRequest (entityHandlerRequest, &payload);
358             }
359             else if (OC_REST_POST == entityHandlerRequest->method)
360             {
361                 OIC_LOG (INFO, TAG, "Received OC_REST_POST from client");
362                 ehResult = ProcessPostRequest (entityHandlerRequest, &response, &payload);
363             }
364             else
365             {
366                 OIC_LOG_V (INFO, TAG, "Received unsupported method %d from client",
367                         entityHandlerRequest->method);
368                 ehResult = OC_EH_ERROR;
369             }
370
371             if (ehResult == OC_EH_OK && ehResult != OC_EH_FORBIDDEN)
372             {
373                 // Format the response.  Note this requires some info about the request
374                 response.requestHandle = entityHandlerRequest->requestHandle;
375                 response.resourceHandle = entityHandlerRequest->resource;
376                 response.ehResult = ehResult;
377                 response.payload = (OCPayload*)(payload);
378                 response.numSendVendorSpecificHeaderOptions = 0;
379                 memset(response.sendVendorSpecificHeaderOptions, 0,
380                        sizeof(response.sendVendorSpecificHeaderOptions));
381                 memset(response.resourceUri, 0, sizeof(response.resourceUri));
382                 // Indicate that response is NOT in a persistent buffer
383                 response.persistentBufferFlag = 0;
384
385                 // Send the response
386                 if (OCDoResponse(&response) != OC_STACK_OK)
387                 {
388                     OIC_LOG(ERROR, TAG, "Error sending response");
389                     ehResult = OC_EH_ERROR;
390                 }
391             }
392         }
393     }
394
395     OCPayloadDestroy(response.payload);
396     return ehResult;
397 }
398
399 /* SIGINT handler: set gQuitFlag to 1 for graceful termination */
400 void handleSigInt(int signum)
401 {
402     if (signum == SIGINT)
403     {
404         gQuitFlag = 1;
405     }
406 }
407
408 FILE* server_fopen(const char *path, const char *mode)
409 {
410     if (0 == strcmp(path, OC_SECURITY_DB_DAT_FILE_NAME))
411     {
412         return fopen(CRED_FILE, mode);
413     }
414     else
415     {
416         return fopen(path, mode);
417     }
418 }
419
420 int main()
421 {
422     struct timespec timeout;
423
424     OIC_LOG(DEBUG, TAG, "OCServer is starting...");
425
426     // Initialize Persistent Storage for SVR database
427     OCPersistentStorage ps = {server_fopen, fread, fwrite, fclose, unlink};
428
429     OCRegisterPersistentStorageHandler(&ps);
430
431     if (OCInit(NULL, 0, OC_SERVER) != OC_STACK_OK)
432     {
433         OIC_LOG(ERROR, TAG, "OCStack init error");
434         return 0;
435     }
436
437     /*
438      * Declare and create the example resource: LED
439      */
440     createLEDResource(gResourceUri, &LED, false, 0);
441
442     timeout.tv_sec  = 0;
443     timeout.tv_nsec = 100000000L;
444
445     // Break from loop with Ctrl-C
446     OIC_LOG(INFO, TAG, "Entering ocserver main loop...");
447     signal(SIGINT, handleSigInt);
448     while (!gQuitFlag)
449     {
450         if (OCProcess() != OC_STACK_OK)
451         {
452             OIC_LOG(ERROR, TAG, "OCStack process error");
453             return 0;
454         }
455         nanosleep(&timeout, NULL);
456     }
457
458     OIC_LOG(INFO, TAG, "Exiting ocserver main loop...");
459
460     if (OCStop() != OC_STACK_OK)
461     {
462         OIC_LOG(ERROR, TAG, "OCStack process error");
463     }
464
465     return 0;
466 }
467
468 int createLEDResource (char *uri, LEDResource *ledResource, bool resourceState, int resourcePower)
469 {
470     if (!uri)
471     {
472         OIC_LOG(ERROR, TAG, "Resource URI cannot be NULL");
473         return -1;
474     }
475
476     ledResource->state = resourceState;
477     ledResource->power= resourcePower;
478     OCStackResult res = OCCreateResource(&(ledResource->handle),
479             "core.led",
480             OC_RSRVD_INTERFACE_DEFAULT,
481             uri,
482             OCEntityHandlerCb,
483             NULL,
484             OC_DISCOVERABLE|OC_OBSERVABLE | OC_SECURE);
485     OIC_LOG_V(INFO, TAG, "Created LED resource with result: %s", getResult(res));
486
487     return 0;
488 }