resource/csdk/stack/samples/linux/SimpleClientServer/occlientcoll.cpp

   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
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <signal.h>
  25 #include <unistd.h>
  26 #include <ocstack.h>
  27 #include <iostream>
  28 #include <sstream>
  29 #include "ocpayload.h"
  30 #include "logger.h"
  31 const char *getResult(OCStackResult result);
  32 std::string getIPAddrTBServer(OCClientResponse * clientResponse);
  33 std::string getPortTBServer(OCClientResponse * clientResponse);
  34 std::string getQueryStrForGetPut();
  35
  36 #define TAG PCF("occlient")
  37 #define DEFAULT_CONTEXT_VALUE 0x99
  38 #ifndef MAX_LENGTH_IPv4_ADDR
  39 #define MAX_LENGTH_IPv4_ADDR 16
  40 #endif
  41
  42 typedef enum
  43 {
  44     TEST_INVALID = 0,
  45     TEST_GET_DEFAULT,
  46     TEST_GET_BATCH,
  47     TEST_GET_LINK_LIST,
  48     TEST_PUT_DEFAULT,
  49     TEST_PUT_BATCH,
  50     TEST_PUT_LINK_LIST,
  51     TEST_UNKNOWN_RESOURCE_GET_DEFAULT,
  52     TEST_UNKNOWN_RESOURCE_GET_BATCH,
  53     TEST_UNKNOWN_RESOURCE_GET_LINK_LIST,
  54     MAX_TESTS
  55 } CLIENT_TEST;
  56
  57 /**
  58  * List of connectivity types that can be initiated from the client
  59  * Required for user input validation
  60  */
  61 typedef enum {
  62     CT_ADAPTER_DEFAULT = 0,
  63     CT_IP,
  64     MAX_CT
  65 } CLIENT_CONNECTIVITY_TYPE;
  66
  67 unsigned static int TEST = TEST_INVALID;
  68 unsigned static int CONNECTIVITY = 0;
  69
  70 typedef struct
  71 {
  72     char text[30];
  73     CLIENT_TEST test;
  74 } testToTextMap;
  75
  76 testToTextMap queryInterface[] = {
  77         {"invalid", TEST_INVALID},
  78         {"?if=oic.if.baseline", TEST_GET_DEFAULT},
  79         {"?if=oic.if.b", TEST_GET_BATCH},
  80         {"?if=oic.if.ll", TEST_GET_LINK_LIST},
  81         {"?if=oic.if.baseline", TEST_UNKNOWN_RESOURCE_GET_DEFAULT},
  82         {"?if=oic.if.b", TEST_UNKNOWN_RESOURCE_GET_BATCH},
  83         {"?if=oic.if.ll", TEST_UNKNOWN_RESOURCE_GET_LINK_LIST},
  84         {"?if=oic.if.baseline", TEST_PUT_DEFAULT},
  85         {"?if=oic.if.b", TEST_PUT_BATCH},
  86         {"?if=oic.if.ll", TEST_PUT_LINK_LIST},
  87 };
  88
  89
  90 //The following variable determines the interface protocol (IP, etc)
  91 //to be used for sending unicast messages. Default set to IP.
  92 static OCConnectivityType OC_CONNTYPE = CT_ADAPTER_IP;
  93 static const char * MULTICAST_RESOURCE_DISCOVERY_QUERY = "/oic/res";
  94
  95 // The handle for the observe registration
  96 OCDoHandle gObserveDoHandle;
  97 // After this crosses a threshold client deregisters for further observations
  98 int gNumObserveNotifies = 1;
  99
 100 int gQuitFlag = 0;
 101 /* SIGINT handler: set gQuitFlag to 1 for graceful termination */
 102 void handleSigInt(int signum)
 103 {
 104     if (signum == SIGINT)
 105     {
 106         gQuitFlag = 1;
 107     }
 108 }
 109
 110 // Forward Declaration
 111 OCStackApplicationResult getReqCB(void* ctx, OCDoHandle handle, OCClientResponse * clientResponse);
 112 int InitGetRequestToUnavailableResource(OCClientResponse * clientResponse);
 113 int InitObserveRequest(OCClientResponse * clientResponse);
 114 int InitPutRequest(OCClientResponse * clientResponse);
 115 int InitGetRequest(OCClientResponse * clientResponse);
 116 int InitDiscovery();
 117
 118 OCPayload* putPayload()
 119 {
 120     OCRepPayload* payload = OCRepPayloadCreate();
 121
 122     if(!payload)
 123     {
 124         std::cout << "Failed to create put payload object"<<std::endl;
 125         std::exit(1);
 126     }
 127
 128     OCRepPayloadSetPropInt(payload, "power", 15);
 129     OCRepPayloadSetPropBool(payload, "state", true);
 130
 131     return (OCPayload*) payload;
 132 }
 133
 134 void PrintUsage()
 135 {
 136     OC_LOG(INFO, TAG, "Usage : occlientcoll -t <Test Case> -c <CA connectivity Type>");
 137     OC_LOG(INFO, TAG, "-c 0 : Default auto-selection");
 138     OC_LOG(INFO, TAG, "-c 1 : IP Connectivity Type");
 139     OC_LOG(INFO, TAG, "Test Case 1 : Discover Resources && Initiate GET Request on an "\
 140             "available resource using default interface.");
 141     OC_LOG(INFO, TAG, "Test Case 2 : Discover Resources && Initiate GET Request on an "\
 142                  "available resource using batch interface.");
 143     OC_LOG(INFO, TAG, "Test Case 3 : Discover Resources && Initiate GET Request on an "\
 144                  "available resource using link list interface.");
 145     OC_LOG(INFO, TAG, "Test Case 4 : Discover Resources && Initiate GET & PUT Request on an "\
 146                  "available resource using default interface.");
 147     OC_LOG(INFO, TAG, "Test Case 5 : Discover Resources && Initiate GET & PUT Request on an "\
 148                  "available resource using batch interface.");
 149     OC_LOG(INFO, TAG, "Test Case 6 : Discover Resources && Initiate GET & PUT Request on an "\
 150                  "available resource using link list interface.");
 151     OC_LOG(INFO, TAG, "Test Case 7 : Discover Resources && Initiate GET Request on an "\
 152                  "unavailable resource using default interface.");
 153     OC_LOG(INFO, TAG, "Test Case 8 : Discover Resources && Initiate GET Request on an "\
 154                  "unavailable resource using batch interface.");
 155     OC_LOG(INFO, TAG, "Test Case 9 : Discover Resources && Initiate GET Request on an "\
 156                  "unavailable resource using link list interface.");
 157 }
 158
 159 OCStackApplicationResult putReqCB(void* ctx, OCDoHandle handle, OCClientResponse * clientResponse)
 160 {
 161     if(clientResponse == NULL)
 162     {
 163         OC_LOG(INFO, TAG, "The clientResponse is NULL");
 164         return   OC_STACK_DELETE_TRANSACTION;
 165     }
 166     if(ctx == (void*)DEFAULT_CONTEXT_VALUE)
 167     {
 168         OC_LOG_V(INFO, TAG, "Callback Context for PUT query recvd successfully");
 169         OC_LOG_PAYLOAD(INFO, TAG, clientResponse->payload);
 170     }
 171
 172     return OC_STACK_KEEP_TRANSACTION;
 173 }
 174
 175 OCStackApplicationResult getReqCB(void* ctx, OCDoHandle handle, OCClientResponse * clientResponse)
 176 {
 177     OC_LOG_V(INFO, TAG, "StackResult: %s",
 178             getResult(clientResponse->result));
 179     if(ctx == (void*)DEFAULT_CONTEXT_VALUE)
 180     {
 181         OC_LOG_V(INFO, TAG, "SEQUENCE NUMBER: %d", clientResponse->sequenceNumber);
 182         if(clientResponse->sequenceNumber == 0)
 183         {
 184             OC_LOG_V(INFO, TAG, "Callback Context for GET query recvd successfully");
 185             OC_LOG_PAYLOAD(INFO, TAG, clientResponse->payload);
 186         }
 187         else
 188         {
 189             OC_LOG_V(INFO, TAG, "Callback Context for Get recvd successfully %d",
 190                     gNumObserveNotifies);
 191             OC_LOG_PAYLOAD(INFO, TAG, clientResponse->payload);;
 192             gNumObserveNotifies++;
 193             if (gNumObserveNotifies == 3)
 194             {
 195                 if (OCCancel (gObserveDoHandle, OC_LOW_QOS, NULL, 0) != OC_STACK_OK)
 196                 {
 197                     OC_LOG(ERROR, TAG, "Observe cancel error");
 198                 }
 199             }
 200         }
 201     }
 202     if(TEST == TEST_PUT_DEFAULT || TEST == TEST_PUT_BATCH || TEST == TEST_PUT_LINK_LIST)
 203     {
 204         InitPutRequest(clientResponse);
 205     }
 206     return OC_STACK_KEEP_TRANSACTION;
 207 }
 208
 209
 210 // This is a function called back when a device is discovered
 211 OCStackApplicationResult discoveryReqCB(void* ctx, OCDoHandle handle,
 212         OCClientResponse * clientResponse)
 213 {
 214     OC_LOG(INFO, TAG,
 215             "Entering discoveryReqCB (Application Layer CB)");
 216     OC_LOG_V(INFO, TAG, "StackResult: %s",
 217             getResult(clientResponse->result));
 218
 219     if (ctx == (void*) DEFAULT_CONTEXT_VALUE)
 220     {
 221         OC_LOG_V(INFO, TAG, "Callback Context recvd successfully");
 222     }
 223
 224     OC_LOG_V(INFO, TAG,
 225             "Device =============> Discovered @ %s:%d",
 226             clientResponse->devAddr.addr,
 227             clientResponse->devAddr.port);
 228     OC_LOG_PAYLOAD(INFO, TAG, clientResponse->payload);
 229
 230     OC_CONNTYPE = clientResponse->connType;
 231
 232     if(TEST == TEST_UNKNOWN_RESOURCE_GET_DEFAULT || TEST == TEST_UNKNOWN_RESOURCE_GET_BATCH ||\
 233             TEST == TEST_UNKNOWN_RESOURCE_GET_LINK_LIST)
 234     {
 235         InitGetRequestToUnavailableResource(clientResponse);
 236     }
 237     else
 238     {
 239         InitGetRequest(clientResponse);
 240     }
 241     return OC_STACK_KEEP_TRANSACTION;
 242 }
 243
 244
 245 int InitGetRequestToUnavailableResource(OCClientResponse * clientResponse)
 246 {
 247     OCStackResult ret;
 248     OCCallbackData cbData;
 249     std::ostringstream getQuery;
 250     getQuery << "coap://" << clientResponse->devAddr.addr << ":" <<
 251             clientResponse->devAddr.port << "/SomeUnknownResource";
 252     cbData.cb = getReqCB;
 253     cbData.context = (void*)DEFAULT_CONTEXT_VALUE;
 254     cbData.cd = NULL;
 255
 256     ret = OCDoResource(NULL, OC_REST_GET, getQuery.str().c_str(), 0, 0, OC_CONNTYPE, OC_LOW_QOS,
 257             &cbData, NULL, 0);
 258     if (ret != OC_STACK_OK)
 259     {
 260         OC_LOG(ERROR, TAG, "OCStack resource error");
 261     }
 262     return ret;
 263 }
 264
 265
 266 int InitObserveRequest(OCClientResponse * clientResponse)
 267 {
 268     OCStackResult ret;
 269     OCCallbackData cbData;
 270     OCDoHandle handle;
 271     std::ostringstream obsReg;
 272     obsReg << "coap://" << clientResponse->devAddr.addr << ":" <<
 273             clientResponse->devAddr.addr <<
 274             getQueryStrForGetPut();
 275     cbData.cb = getReqCB;
 276     cbData.context = (void*)DEFAULT_CONTEXT_VALUE;
 277     cbData.cd = NULL;
 278     OC_LOG_V(INFO, TAG, "OBSERVE payload from client =");
 279     OCPayload* payload = putPayload();
 280     OC_LOG_PAYLOAD(INFO, TAG, payload);
 281     OCPayloadDestroy(payload);
 282
 283     ret = OCDoResource(&handle, OC_REST_OBSERVE, obsReg.str().c_str(), 0, 0, OC_CONNTYPE,
 284             OC_LOW_QOS, &cbData, NULL, 0);
 285     if (ret != OC_STACK_OK)
 286     {
 287         OC_LOG(ERROR, TAG, "OCStack resource error");
 288     }
 289     else
 290     {
 291         gObserveDoHandle = handle;
 292     }
 293     return ret;
 294 }
 295
 296
 297 int InitPutRequest(OCClientResponse * clientResponse)
 298 {
 299     OCStackResult ret;
 300     OCCallbackData cbData;
 301     //* Make a PUT query*/
 302     std::ostringstream getQuery;
 303     getQuery << "coap://" << clientResponse->devAddr.addr << ":" <<
 304             clientResponse->devAddr.port <<
 305             "/a/room" << queryInterface[TEST].text;
 306     cbData.cb = putReqCB;
 307     cbData.context = (void*)DEFAULT_CONTEXT_VALUE;
 308     cbData.cd = NULL;
 309     OC_LOG_V(INFO, TAG, "PUT payload from client = ");
 310     OCPayload* payload = putPayload();
 311     OC_LOG_PAYLOAD(INFO, TAG, payload);
 312     OCPayloadDestroy(payload);
 313
 314     ret = OCDoResource(NULL, OC_REST_PUT, getQuery.str().c_str(), 0, putPayload(),
 315                         OC_CONNTYPE, OC_LOW_QOS, &cbData, NULL, 0);
 316     if (ret != OC_STACK_OK)
 317     {
 318         OC_LOG(ERROR, TAG, "OCStack resource error");
 319     }
 320     return ret;
 321 }
 322
 323
 324 int InitGetRequest(OCClientResponse * clientResponse)
 325 {
 326     OCStackResult ret;
 327     OCCallbackData cbData;
 328
 329     //* Make a GET query*/
 330     std::ostringstream getQuery;
 331     getQuery << "coap://" << clientResponse->devAddr.addr << ":" <<
 332             clientResponse->devAddr.port <<
 333             "/a/room" << queryInterface[TEST].text;
 334
 335     std::cout << "Get Query: " << getQuery.str() << std::endl;
 336
 337     cbData.cb = getReqCB;
 338     cbData.context = (void*)DEFAULT_CONTEXT_VALUE;
 339     cbData.cd = NULL;
 340     ret = OCDoResource(NULL, OC_REST_GET,
 341             getQuery.str().c_str(), 0, 0, OC_CONNTYPE, OC_LOW_QOS,
 342             &cbData, NULL, 0);
 343     if (ret != OC_STACK_OK)
 344     {
 345         OC_LOG(ERROR, TAG, "OCStack resource error");
 346     }
 347     return ret;
 348 }
 349
 350 int InitDiscovery()
 351 {
 352     OCStackResult ret;
 353     OCCallbackData cbData;
 354     /* Start a discovery query*/
 355     char szQueryUri[64] = { 0 };
 356
 357     strcpy(szQueryUri, MULTICAST_RESOURCE_DISCOVERY_QUERY);
 358
 359     cbData.cb = discoveryReqCB;
 360     cbData.context = (void*)DEFAULT_CONTEXT_VALUE;
 361     cbData.cd = NULL;
 362     ret = OCDoResource(NULL, OC_REST_DISCOVER, szQueryUri, 0, 0, OC_CONNTYPE,
 363                         OC_LOW_QOS,
 364             &cbData, NULL, 0);
 365     if (ret != OC_STACK_OK)
 366     {
 367         OC_LOG(ERROR, TAG, "OCStack resource error");
 368     }
 369     return ret;
 370 }
 371
 372 int main(int argc, char* argv[])
 373 {
 374     int opt;
 375
 376     while ((opt = getopt(argc, argv, "t:c:")) != -1)
 377     {
 378         switch (opt)
 379         {
 380             case 't':
 381                 TEST = atoi(optarg);
 382                 break;
 383             case 'c':
 384                 CONNECTIVITY = atoi(optarg);
 385                 break;
 386             default:
 387                 PrintUsage();
 388                 return -1;
 389         }
 390     }
 391     if ((TEST <= TEST_INVALID || TEST >= MAX_TESTS) ||
 392         CONNECTIVITY >= MAX_CT)
 393     {
 394         PrintUsage();
 395         return -1;
 396     }
 397
 398     /* Initialize OCStack*/
 399     if (OCInit(NULL, 0, OC_CLIENT) != OC_STACK_OK)
 400     {
 401         OC_LOG(ERROR, TAG, "OCStack init error");
 402         return 0;
 403     }
 404
 405     if(CONNECTIVITY == CT_ADAPTER_DEFAULT || CONNECTIVITY == CT_IP)
 406     {
 407         OC_CONNTYPE = CT_ADAPTER_IP;
 408     }
 409     else
 410     {
 411         OC_LOG(INFO, TAG, "Default Connectivity type selected...");
 412         OC_CONNTYPE = CT_ADAPTER_IP;
 413     }
 414
 415     InitDiscovery();
 416
 417     // Break from loop with Ctrl+C
 418     OC_LOG(INFO, TAG, "Entering occlient main loop...");
 419     signal(SIGINT, handleSigInt);
 420     while (!gQuitFlag)
 421     {
 422
 423         if (OCProcess() != OC_STACK_OK)
 424         {
 425             OC_LOG(ERROR, TAG, "OCStack process error");
 426             return 0;
 427         }
 428
 429         sleep(2);
 430     } OC_LOG(INFO, TAG, "Exiting occlient main loop...");
 431
 432     if (OCStop() != OC_STACK_OK)
 433     {
 434         OC_LOG(ERROR, TAG, "OCStack stop error");
 435     }
 436
 437     return 0;
 438 }
 439
 440 std::string getQueryStrForGetPut()
 441 {
 442     return "/a/room";
 443 }
 444