resource/csdk/security/provisioning/src/pmutility.c

   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 #ifndef _POSIX_C_SOURCE
  21 #define _POSIX_C_SOURCE 200112L
  22 #endif
  23
  24 #if HAVE_UNISTD_H
  25 #include <unistd.h>
  26 #endif
  27 #ifdef HAVE_WINDOWS_H
  28 #include <windows.h>
  29 #endif
  30 #ifdef HAVE_STRING_H
  31 #include <string.h>
  32 #endif
  33 #ifdef HAVE_TIME_H
  34 #include <time.h>
  35 #endif
  36 #ifdef HAVE_SYS_TIME_H
  37 #include <sys/time.h>
  38 #endif
  39
  40 #include "ocstack.h"
  41 #include "oic_malloc.h"
  42 #include "oic_string.h"
  43 #include "logger.h"
  44 #include "cJSON.h"
  45 #include "utlist.h"
  46 #include "ocpayload.h"
  47
  48 #include "securevirtualresourcetypes.h"
  49 #include "srmresourcestrings.h" //@note: SRM's internal header
  50 #include "doxmresource.h"       //@note: SRM's internal header
  51 #include "pstatresource.h"      //@note: SRM's internal header
  52 #include "verresource.h"      //@note: SRM's internal header
  53
  54 #include "pmtypes.h"
  55 #include "pmutility.h"
  56
  57 #include "srmutility.h"
  58
  59 #define TAG ("PM-UTILITY")
  60
  61 typedef struct _DiscoveryInfo{
  62     OCProvisionDev_t    **ppDevicesList;
  63     bool                isOwnedDiscovery;
  64 } DiscoveryInfo;
  65
  66 /*
  67  * Function to discover secre port information through unicast
  68  *
  69  * @param[in] discoveryInfo The pointer of discovery information to matain result of discovery
  70  * @param[in] clientResponse  Response information(It will contain payload)
  71  *
  72  * @return OC_STACK_OK on success otherwise error.
  73  */
  74 static OCStackResult SecurePortDiscovery(DiscoveryInfo* discoveryInfo,
  75                                          const OCClientResponse *clientResponse);
  76
  77 /*
  78  * Function to discover security version information through unicast
  79  *
  80  * @param[in] discoveryInfo The pointer of discovery information to matain result of discovery
  81  * @param[in] clientResponse  Response information(It will contain payload)
  82  *
  83  * @return OC_STACK_OK on success otherwise error.
  84  */
  85 static OCStackResult SecurityVersionDiscovery(DiscoveryInfo* discoveryInfo,
  86                                               const OCClientResponse *clientResponse);
  87
  88 /**
  89  * Callback handler for PMDeviceDiscovery API.
  90  *
  91  * @param[in] ctx             User context
  92  * @param[in] handle          Handler for response
  93  * @param[in] clientResponse  Response information (It will contain payload)
  94  * @return OC_STACK_KEEP_TRANSACTION to keep transaction and
  95  *         OC_STACK_DELETE_TRANSACTION to delete it.
  96  */
  97 static OCStackApplicationResult DeviceDiscoveryHandler(void *ctx, OCDoHandle UNUSED,
  98                                 OCClientResponse *clientResponse);
  99
 100 /**
 101  * Callback handler for getting secure port information using /oic/res discovery.
 102  *
 103  * @param[in] ctx             user context
 104  * @param[in] handle          Handle for response
 105  * @param[in] clientResponse  Response information(It will contain payload)
 106  *
 107  * @return OC_STACK_KEEP_TRANSACTION to keep transaction and
 108  *         OC_STACK_DELETE_TRANSACTION to delete it.
 109  */
 110 static OCStackApplicationResult SecurePortDiscoveryHandler(void *ctx, OCDoHandle UNUSED,
 111                                  OCClientResponse *clientResponse);
 112
 113 /**
 114  * Callback handler for security version discovery.
 115  *
 116  * @param[in] ctx             User context
 117  * @param[in] handle          Handler for response
 118  * @param[in] clientResponse  Response information (It will contain payload)
 119  * @return OC_STACK_KEEP_TRANSACTION to keep transaction and
 120  *         OC_STACK_DELETE_TRANSACTION to delete it.
 121  */
 122 static OCStackApplicationResult SecVersionDiscoveryHandler(void *ctx, OCDoHandle UNUSED,
 123                                 OCClientResponse *clientResponse);
 124
 125 /**
 126  * Function to search node in linked list that matches given IP and port.
 127  *
 128  * @param[in] pList         List of OCProvisionDev_t.
 129  * @param[in] addr          address of target device.
 130  * @param[in] port          port of remote server.
 131  *
 132  * @return pointer of OCProvisionDev_t if exist, otherwise NULL
 133  */
 134 OCProvisionDev_t* GetDevice(OCProvisionDev_t **ppDevicesList, const char* addr, const uint16_t port)
 135 {
 136     if(NULL == addr || NULL == *ppDevicesList)
 137     {
 138         OIC_LOG_V(ERROR, TAG, "Invalid Input parameters in [%s]\n", __FUNCTION__);
 139         return NULL;
 140     }
 141
 142     OCProvisionDev_t *ptr = NULL;
 143     LL_FOREACH(*ppDevicesList, ptr)
 144     {
 145         if( strcmp(ptr->endpoint.addr, addr) == 0 && port == ptr->endpoint.port)
 146         {
 147             return ptr;
 148         }
 149     }
 150
 151     return NULL;
 152 }
 153
 154
 155 /**
 156  * Add device information to list.
 157  *
 158  * @param[in] pList         List of OCProvisionDev_t.
 159  * @param[in] addr          address of target device.
 160  * @param[in] port          port of remote server.
 161  * @param[in] adapter       adapter type of endpoint.
 162  * @param[in] doxm          pointer to doxm instance.
 163  * @param[in] connType  connectivity type of endpoint
 164  *
 165  * @return OC_STACK_OK for success and errorcode otherwise.
 166  */
 167 OCStackResult AddDevice(OCProvisionDev_t **ppDevicesList, const char* addr, const uint16_t port,
 168                         OCTransportAdapter adapter, OCConnectivityType connType, OicSecDoxm_t *doxm)
 169 {
 170     if (NULL == addr)
 171     {
 172         return OC_STACK_INVALID_PARAM;
 173     }
 174
 175     OCProvisionDev_t *ptr = GetDevice(ppDevicesList, addr, port);
 176     if(!ptr)
 177     {
 178         ptr = (OCProvisionDev_t *)OICCalloc(1, sizeof (OCProvisionDev_t));
 179         if (NULL == ptr)
 180         {
 181             OIC_LOG(ERROR, TAG, "Error while allocating memory for linkedlist node !!");
 182             return OC_STACK_NO_MEMORY;
 183         }
 184
 185         OICStrcpy(ptr->endpoint.addr, MAX_ADDR_STR_SIZE, addr);
 186         ptr->endpoint.port = port;
 187         ptr->doxm = doxm;
 188         ptr->securePort = DEFAULT_SECURE_PORT;
 189         ptr->endpoint.adapter = adapter;
 190         ptr->next = NULL;
 191         ptr->connType = connType;
 192         ptr->devStatus = DEV_STATUS_ON; //AddDevice is called when discovery(=alive)
 193         OICStrcpy(ptr->secVer, MAX_VERSION_LEN, DEFAULT_SEC_VERSION); // version initialization
 194
 195         LL_PREPEND(*ppDevicesList, ptr);
 196     }
 197
 198     return OC_STACK_OK;
 199 }
 200
 201 /**
 202  * Function to set secure port information from the given list of devices.
 203  *
 204  * @param[in] pList         List of OCProvisionDev_t.
 205  * @param[in] addr          address of target device.
 206  * @param[in] port          port of remote server.
 207  * @param[in] secureport    secure port information.
 208  *
 209  * @return OC_STACK_OK for success and errorcode otherwise.
 210  */
 211 OCStackResult UpdateSecurePortOfDevice(OCProvisionDev_t **ppDevicesList, const char *addr,
 212                                        uint16_t port, uint16_t securePort)
 213 {
 214     OCProvisionDev_t *ptr = GetDevice(ppDevicesList, addr, port);
 215
 216     if(!ptr)
 217     {
 218         OIC_LOG(ERROR, TAG, "Can not find device information in the discovery device list");
 219         return OC_STACK_ERROR;
 220     }
 221
 222     ptr->securePort = securePort;
 223
 224     return OC_STACK_OK;
 225 }
 226
 227 /**
 228  * Function to set security version information from the given list of devices.
 229  *
 230  * @param[in] pList         List of OCProvisionDev_t.
 231  * @param[in] addr          address of target device.
 232  * @param[in] port          port of remote server.
 233  * @param[in] secVer    security version information.
 234  *
 235  * @return OC_STACK_OK for success and errorcode otherwise.
 236  */
 237 OCStackResult UpdateSecVersionOfDevice(OCProvisionDev_t **ppDevicesList, const char *addr,
 238                                        uint16_t port, const char* secVer)
 239 {
 240     if (NULL == secVer)
 241     {
 242         return OC_STACK_INVALID_PARAM;
 243     }
 244
 245     OCProvisionDev_t *ptr = GetDevice(ppDevicesList, addr, port);
 246
 247     if(!ptr)
 248     {
 249         OIC_LOG(ERROR, TAG, "Can not find device information in the discovery device list");
 250         return OC_STACK_ERROR;
 251     }
 252
 253     OICStrcpy(ptr->secVer, MAX_VERSION_LEN, secVer);
 254
 255     return OC_STACK_OK;
 256 }
 257
 258 /**
 259  * This function deletes list of provision target devices
 260  *
 261  * @param[in] pDevicesList         List of OCProvisionDev_t.
 262  */
 263 void PMDeleteDeviceList(OCProvisionDev_t *pDevicesList)
 264 {
 265     if(pDevicesList)
 266     {
 267         OCProvisionDev_t *del = NULL, *tmp = NULL;
 268         LL_FOREACH_SAFE(pDevicesList, del, tmp)
 269         {
 270             LL_DELETE(pDevicesList, del);
 271
 272             DeleteDoxmBinData(del->doxm);
 273             DeletePstatBinData(del->pstat);
 274             OICFree(del);
 275         }
 276     }
 277 }
 278
 279 OCProvisionDev_t* PMCloneOCProvisionDev(const OCProvisionDev_t* src)
 280 {
 281     OIC_LOG(DEBUG, TAG, "IN PMCloneOCProvisionDev");
 282
 283     if (!src)
 284     {
 285         OIC_LOG(ERROR, TAG, "PMCloneOCProvisionDev : Invalid parameter");
 286         return NULL;
 287     }
 288
 289     // TODO: Consider use VERIFY_NON_NULL instead of if ( null check ) { goto exit; }
 290     OCProvisionDev_t* newDev = (OCProvisionDev_t*)OICCalloc(1, sizeof(OCProvisionDev_t));
 291     VERIFY_NON_NULL(TAG, newDev, ERROR);
 292
 293     memcpy(&newDev->endpoint, &src->endpoint, sizeof(OCDevAddr));
 294
 295     if (src->pstat)
 296     {
 297         newDev->pstat= (OicSecPstat_t*)OICCalloc(1, sizeof(OicSecPstat_t));
 298         VERIFY_NON_NULL(TAG, newDev->pstat, ERROR);
 299
 300         memcpy(newDev->pstat, src->pstat, sizeof(OicSecPstat_t));
 301         // We have to assign NULL for not necessary information to prevent memory corruption.
 302         newDev->pstat->sm = NULL;
 303     }
 304
 305     if (src->doxm)
 306     {
 307         newDev->doxm = (OicSecDoxm_t*)OICCalloc(1, sizeof(OicSecDoxm_t));
 308         VERIFY_NON_NULL(TAG, newDev->doxm, ERROR);
 309
 310         memcpy(newDev->doxm, src->doxm, sizeof(OicSecDoxm_t));
 311         // We have to assign NULL for not necessary information to prevent memory corruption.
 312         newDev->doxm->oxmType = NULL;
 313         newDev->doxm->oxm = NULL;
 314     }
 315
 316     if (0 == strlen(src->secVer))
 317     {
 318         OICStrcpy(newDev->secVer, MAX_VERSION_LEN, DEFAULT_SEC_VERSION);
 319     }
 320     else
 321     {
 322         OICStrcpy(newDev->secVer, MAX_VERSION_LEN, src->secVer);
 323     }
 324
 325     newDev->securePort = src->securePort;
 326     newDev->devStatus = src->devStatus;
 327     newDev->connType = src->connType;
 328     newDev->next = NULL;
 329
 330     OIC_LOG(DEBUG, TAG, "OUT PMCloneOCProvisionDev");
 331
 332     return newDev;
 333
 334 exit:
 335     OIC_LOG(ERROR, TAG, "PMCloneOCProvisionDev : Failed to allocate memory");
 336     if (newDev)
 337     {
 338         OICFree(newDev->pstat);
 339         OICFree(newDev->doxm);
 340         OICFree(newDev);
 341     }
 342     return NULL;
 343 }
 344
 345 /**
 346  * Timeout implementation for secure discovery. When performing secure discovery,
 347  * we should wait a certain period of time for getting response of each devices.
 348  *
 349  * @param[in]  waittime  Timeout in seconds.
 350  * @param[in]  waitForStackResponse if true timeout function will call OCProcess while waiting.
 351  * @return OC_STACK_OK on success otherwise error.
 352  */
 353 OCStackResult PMTimeout(unsigned short waittime, bool waitForStackResponse)
 354 {
 355     struct timespec startTime = {.tv_sec=0, .tv_nsec=0};
 356     struct timespec currTime  = {.tv_sec=0, .tv_nsec=0};
 357
 358     OCStackResult res = OC_STACK_OK;
 359 #ifdef _POSIX_MONOTONIC_CLOCK
 360     int clock_res = clock_gettime(CLOCK_MONOTONIC, &startTime);
 361 #else
 362     int clock_res = clock_gettime(CLOCK_REALTIME, &startTime);
 363 #endif
 364     if (0 != clock_res)
 365     {
 366         return OC_STACK_ERROR;
 367     }
 368     while (OC_STACK_OK == res)
 369     {
 370 #ifdef _POSIX_MONOTONIC_CLOCK
 371         clock_res = clock_gettime(CLOCK_MONOTONIC, &currTime);
 372 #else
 373         clock_res = clock_gettime(CLOCK_REALTIME, &currTime);
 374 #endif
 375         if (0 != clock_res)
 376         {
 377             return OC_STACK_TIMEOUT;
 378         }
 379         long elapsed = (currTime.tv_sec - startTime.tv_sec);
 380         if (elapsed > waittime)
 381         {
 382             return OC_STACK_OK;
 383         }
 384         if (waitForStackResponse)
 385         {
 386             res = OCProcess();
 387         }
 388     }
 389     return res;
 390 }
 391
 392 /**
 393  * Extract secure port information from payload of discovery response.
 394  *
 395  * @param[in] jsonStr response payload of /oic/res discovery.
 396  *
 397  * @return Secure port
 398  */
 399 uint16_t GetSecurePortFromJSON(char* jsonStr)
 400 {
 401     // TODO: Modify error handling
 402     if (NULL == jsonStr)
 403     {
 404         return 0;
 405     }
 406     cJSON *jsonProp = NULL;
 407     cJSON *jsonP = NULL;
 408     cJSON *jsonPort = NULL;
 409
 410     cJSON *jsonRoot = cJSON_Parse(jsonStr);
 411     if(!jsonRoot)
 412     {
 413         // TODO: Add error log & return default secure port
 414         return 0;
 415     }
 416
 417     jsonProp = cJSON_GetObjectItem(jsonRoot, "prop");
 418     if(!jsonProp)
 419     {
 420         // TODO: Add error log & return default secure port
 421         return 0;
 422     }
 423
 424     jsonP = cJSON_GetObjectItem(jsonProp, "p");
 425     if(!jsonP)
 426     {
 427         // TODO: Add error log & return default secure port
 428         return 0;
 429     }
 430
 431     jsonPort = cJSON_GetObjectItem(jsonP, "port");
 432     if(!jsonPort)
 433     {
 434         // TODO: Add error log & return default secure port
 435         return 0;
 436     }
 437
 438     return (uint16_t)jsonPort->valueint;
 439 }
 440
 441 bool PMGenerateQuery(bool isSecure,
 442                      const char* address, const uint16_t port,
 443                      const OCConnectivityType connType,
 444                      char* buffer, size_t bufferSize, const char* uri)
 445 {
 446     if(!address || !buffer || !uri)
 447     {
 448         OIC_LOG(ERROR, TAG, "PMGenerateQuery : Invalid parameters.");
 449         return false;
 450     }
 451
 452     int snRet = 0;
 453     char* prefix = (isSecure == true) ? COAPS_PREFIX : COAP_PREFIX;
 454
 455     switch(connType & CT_MASK_ADAPTER)
 456     {
 457         case CT_ADAPTER_IP:
 458             switch(connType & CT_MASK_FLAGS & ~CT_FLAG_SECURE)
 459             {
 460                 case CT_IP_USE_V4:
 461                         snRet = snprintf(buffer, bufferSize, "%s%s:%d%s",
 462                                          prefix, address, port, uri);
 463                     break;
 464                 case CT_IP_USE_V6:
 465                         snRet = snprintf(buffer, bufferSize, "%s[%s]:%d%s",
 466                                          prefix, address, port, uri);
 467                     break;
 468                 default:
 469                     OIC_LOG(ERROR, TAG, "Unknown address format.");
 470                     return false;
 471             }
 472             // snprintf return value check
 473             if (snRet < 0)
 474             {
 475                 OIC_LOG_V(ERROR, TAG, "PMGenerateQuery : Error (snprintf) %d\n", snRet);
 476                 return false;
 477             }
 478             else if ((size_t)snRet >= bufferSize)
 479             {
 480                 OIC_LOG_V(ERROR, TAG, "PMGenerateQuery : Truncated (snprintf) %d\n", snRet);
 481                 return false;
 482             }
 483
 484             break;
 485         // TODO: We need to verify tinyDTLS in below cases
 486         case CT_ADAPTER_GATT_BTLE:
 487         case CT_ADAPTER_RFCOMM_BTEDR:
 488             OIC_LOG(ERROR, TAG, "Not supported connectivity adapter.");
 489             return false;
 490             break;
 491         default:
 492             OIC_LOG(ERROR, TAG, "Unknown connectivity adapter.");
 493             return false;
 494     }
 495
 496     return true;
 497 }
 498
 499 static OCStackApplicationResult SecurityVersionDiscoveryHandler(void *ctx, OCDoHandle UNUSED,
 500                                 OCClientResponse *clientResponse)
 501 {
 502     if (ctx == NULL)
 503     {
 504         OIC_LOG(ERROR, TAG, "Lost List of device information");
 505         return OC_STACK_KEEP_TRANSACTION;
 506     }
 507     (void)UNUSED;
 508     if (clientResponse)
 509     {
 510         if  (NULL == clientResponse->payload)
 511         {
 512             OIC_LOG(INFO, TAG, "Skiping Null payload");
 513             return OC_STACK_KEEP_TRANSACTION;
 514         }
 515         if (OC_STACK_OK != clientResponse->result)
 516         {
 517             OIC_LOG(INFO, TAG, "Error in response");
 518             return OC_STACK_KEEP_TRANSACTION;
 519         }
 520         else
 521         {
 522             if (PAYLOAD_TYPE_SECURITY != clientResponse->payload->type)
 523             {
 524                 OIC_LOG(INFO, TAG, "Unknown payload type");
 525                 return OC_STACK_KEEP_TRANSACTION;
 526             }
 527
 528             OicSecVer_t *ptrVer = NULL;
 529             uint8_t *payload = ((OCSecurityPayload*)clientResponse->payload)->securityData;
 530             size_t size = ((OCSecurityPayload*)clientResponse->payload)->payloadSize;
 531
 532             OCStackResult res = CBORPayloadToVer(payload, size, &ptrVer);
 533             if ((NULL == ptrVer) && (OC_STACK_OK != res))
 534             {
 535                 OIC_LOG(INFO, TAG, "Ignoring malformed CBOR");
 536                 return OC_STACK_KEEP_TRANSACTION;
 537             }
 538             else
 539             {
 540                 OIC_LOG(DEBUG, TAG, "Successfully converted ver cbor to bin.");
 541
 542                 //If this is owend device discovery we have to filter out the responses.
 543                 DiscoveryInfo* pDInfo = (DiscoveryInfo*)ctx;
 544                 res = UpdateSecVersionOfDevice(pDInfo->ppDevicesList, clientResponse->devAddr.addr,
 545                                                          clientResponse->devAddr.port, ptrVer->secv);
 546                 if (OC_STACK_OK != res)
 547                 {
 548                     OIC_LOG(ERROR, TAG, "Error while getting security version.");
 549                     DeleteVerBinData(ptrVer);
 550                     return OC_STACK_KEEP_TRANSACTION;
 551                 }
 552
 553                 OIC_LOG(INFO, TAG, "= Discovered security version =");
 554                 OIC_LOG_V(DEBUG, TAG, "IP %s", clientResponse->devAddr.addr);
 555                 OIC_LOG_V(DEBUG, TAG, "PORT %d", clientResponse->devAddr.port);
 556                 OIC_LOG_V(DEBUG, TAG, "VERSION %s", ptrVer->secv);
 557
 558                 OIC_LOG(INFO, TAG, "Exiting SecVersionDiscoveryHandler.");
 559                 DeleteVerBinData(ptrVer);
 560             }
 561         }
 562     }
 563     else
 564     {
 565         OIC_LOG(INFO, TAG, "Skiping Null response");
 566         return OC_STACK_KEEP_TRANSACTION;
 567     }
 568
 569     return  OC_STACK_DELETE_TRANSACTION;
 570 }
 571
 572 static OCStackApplicationResult SecurePortDiscoveryHandler(void *ctx, OCDoHandle UNUSED,
 573                                  OCClientResponse *clientResponse)
 574 {
 575     if (ctx == NULL)
 576     {
 577         OIC_LOG(ERROR, TAG, "Lost List of device information");
 578         return OC_STACK_DELETE_TRANSACTION;
 579     }
 580     (void)UNUSED;
 581     if (clientResponse)
 582     {
 583         if  (NULL == clientResponse->payload)
 584         {
 585             OIC_LOG(INFO, TAG, "Skiping Null payload");
 586         }
 587         else
 588         {
 589             if (PAYLOAD_TYPE_DISCOVERY != clientResponse->payload->type)
 590             {
 591                 OIC_LOG(INFO, TAG, "Wrong payload type");
 592                 return OC_STACK_DELETE_TRANSACTION;
 593             }
 594
 595             uint16_t securePort = 0;
 596             OCResourcePayload* resPayload = ((OCDiscoveryPayload*)clientResponse->payload)->resources;
 597
 598             if (resPayload && resPayload->secure)
 599             {
 600                 securePort = resPayload->port;
 601             }
 602             else
 603             {
 604                 OIC_LOG(INFO, TAG, "Can not find secure port information.");
 605                 return OC_STACK_DELETE_TRANSACTION;
 606             }
 607
 608             DiscoveryInfo* pDInfo = (DiscoveryInfo*)ctx;
 609             OCStackResult res = UpdateSecurePortOfDevice(pDInfo->ppDevicesList,
 610                                                          clientResponse->devAddr.addr,
 611                                                          clientResponse->devAddr.port, securePort);
 612             if (OC_STACK_OK != res)
 613             {
 614                 OIC_LOG(ERROR, TAG, "Error while getting secure port.");
 615                 return OC_STACK_DELETE_TRANSACTION;
 616             }
 617
 618             res = SecurityVersionDiscovery(pDInfo, clientResponse);
 619             if(OC_STACK_OK != res)
 620             {
 621                 OIC_LOG(ERROR, TAG, "Failed to SecurityVersionDiscovery");
 622                 return OC_STACK_DELETE_TRANSACTION;
 623             }
 624
 625             OIC_LOG(INFO, TAG, "Exiting SecurePortDiscoveryHandler.");
 626         }
 627
 628         return  OC_STACK_DELETE_TRANSACTION;
 629     }
 630     else
 631     {
 632         OIC_LOG(INFO, TAG, "Skiping Null response");
 633     }
 634
 635     return  OC_STACK_DELETE_TRANSACTION;
 636 }
 637
 638 static OCStackApplicationResult DeviceDiscoveryHandler(void *ctx, OCDoHandle UNUSED,
 639                                 OCClientResponse *clientResponse)
 640 {
 641     if (ctx == NULL)
 642     {
 643         OIC_LOG(ERROR, TAG, "Lost List of device information");
 644         return OC_STACK_KEEP_TRANSACTION;
 645     }
 646     (void)UNUSED;
 647     if (clientResponse)
 648     {
 649         if  (NULL == clientResponse->payload)
 650         {
 651             OIC_LOG(INFO, TAG, "Skiping Null payload");
 652             return OC_STACK_KEEP_TRANSACTION;
 653         }
 654         if (OC_STACK_OK != clientResponse->result)
 655         {
 656             OIC_LOG(INFO, TAG, "Error in response");
 657             return OC_STACK_KEEP_TRANSACTION;
 658         }
 659         else
 660         {
 661             if (PAYLOAD_TYPE_SECURITY != clientResponse->payload->type)
 662             {
 663                 OIC_LOG(INFO, TAG, "Unknown payload type");
 664                 return OC_STACK_KEEP_TRANSACTION;
 665             }
 666
 667             OicSecDoxm_t *ptrDoxm = NULL;
 668             uint8_t *payload = ((OCSecurityPayload*)clientResponse->payload)->securityData;
 669             size_t size = ((OCSecurityPayload*)clientResponse->payload)->payloadSize;
 670
 671             OCStackResult res = CBORPayloadToDoxm(payload, size, &ptrDoxm);
 672             if ((NULL == ptrDoxm) || (OC_STACK_OK != res))
 673             {
 674                 OIC_LOG(INFO, TAG, "Ignoring malformed CBOR");
 675                 return OC_STACK_KEEP_TRANSACTION;
 676             }
 677             else
 678             {
 679                 OIC_LOG(DEBUG, TAG, "Successfully converted doxm cbor to bin.");
 680
 681                 //If this is owend device discovery we have to filter out the responses.
 682                 DiscoveryInfo* pDInfo = (DiscoveryInfo*)ctx;
 683                 OCProvisionDev_t **ppDevicesList = pDInfo->ppDevicesList;
 684
 685                 // Get my device ID from doxm resource
 686                 OicUuid_t myId;
 687                 memset(&myId, 0, sizeof(myId));
 688                 OCStackResult res = GetDoxmDevOwnerId(&myId);
 689                 if(OC_STACK_OK != res)
 690                 {
 691                     OIC_LOG(ERROR, TAG, "Error while getting my device ID.");
 692                     DeleteDoxmBinData(ptrDoxm);
 693                     return OC_STACK_KEEP_TRANSACTION;
 694                 }
 695
 696                 // If this is owned discovery response but owner is not me then discard it.
 697                 if( (pDInfo->isOwnedDiscovery) &&
 698                     (0 != memcmp(&ptrDoxm->owner.id, &myId.id, sizeof(myId.id))) )
 699                 {
 700                     OIC_LOG(DEBUG, TAG, "Discovered device is not owend by me");
 701                     DeleteDoxmBinData(ptrDoxm);
 702                     return OC_STACK_KEEP_TRANSACTION;
 703                 }
 704
 705                 res = AddDevice(ppDevicesList, clientResponse->devAddr.addr,
 706                         clientResponse->devAddr.port,
 707                         clientResponse->devAddr.adapter,
 708                         clientResponse->connType, ptrDoxm);
 709                 if (OC_STACK_OK != res)
 710                 {
 711                     OIC_LOG(ERROR, TAG, "Error while adding data to linkedlist.");
 712                     DeleteDoxmBinData(ptrDoxm);
 713                     return OC_STACK_KEEP_TRANSACTION;
 714                 }
 715
 716                 res = SecurePortDiscovery(pDInfo, clientResponse);
 717                 if(OC_STACK_OK != res)
 718                 {
 719                     OIC_LOG(ERROR, TAG, "Failed to SecurePortDiscovery");
 720                     DeleteDoxmBinData(ptrDoxm);
 721                     return OC_STACK_KEEP_TRANSACTION;
 722                 }
 723
 724                 OIC_LOG(INFO, TAG, "Exiting ProvisionDiscoveryHandler.");
 725             }
 726
 727             return  OC_STACK_KEEP_TRANSACTION;
 728         }
 729     }
 730     else
 731     {
 732         OIC_LOG(INFO, TAG, "Skiping Null response");
 733         return OC_STACK_KEEP_TRANSACTION;
 734     }
 735
 736     return  OC_STACK_DELETE_TRANSACTION;
 737 }
 738
 739 /**
 740  * Discover owned/unowned devices in the same IP subnet. .
 741  *
 742  * @param[in] waittime      Timeout in seconds.
 743  * @param[in] isOwned       bool flag for owned / unowned discovery
 744  * @param[in] ppDevicesList        List of OCProvisionDev_t.
 745  *
 746  * @return OC_STACK_OK on success otherwise error.
 747  */
 748 OCStackResult PMDeviceDiscovery(unsigned short waittime, bool isOwned, OCProvisionDev_t **ppDevicesList)
 749 {
 750     OIC_LOG(DEBUG, TAG, "IN PMDeviceDiscovery");
 751
 752     if (NULL != *ppDevicesList)
 753     {
 754         OIC_LOG(ERROR, TAG, "List is not null can cause memory leak");
 755         return OC_STACK_INVALID_PARAM;
 756     }
 757
 758     const char DOXM_OWNED_FALSE_MULTICAST_QUERY[] = "/oic/sec/doxm?Owned=FALSE";
 759     const char DOXM_OWNED_TRUE_MULTICAST_QUERY[] = "/oic/sec/doxm?Owned=TRUE";
 760
 761     DiscoveryInfo *pDInfo = OICCalloc(1, sizeof(DiscoveryInfo));
 762     if(NULL == pDInfo)
 763     {
 764         OIC_LOG(ERROR, TAG, "PMDeviceDiscovery : Memory allocation failed.");
 765         return OC_STACK_NO_MEMORY;
 766     }
 767
 768     pDInfo->ppDevicesList = ppDevicesList;
 769     pDInfo->isOwnedDiscovery = isOwned;
 770
 771     OCCallbackData cbData;
 772     cbData.cb = &DeviceDiscoveryHandler;
 773     cbData.context = (void *)pDInfo;
 774     cbData.cd = NULL;
 775     OCStackResult res = OC_STACK_ERROR;
 776
 777     const char* query = isOwned ? DOXM_OWNED_TRUE_MULTICAST_QUERY :
 778                                   DOXM_OWNED_FALSE_MULTICAST_QUERY;
 779
 780     OCDoHandle handle = NULL;
 781     res = OCDoResource(&handle, OC_REST_DISCOVER, query, 0, 0,
 782                                      CT_DEFAULT, OC_LOW_QOS, &cbData, NULL, 0);
 783     if (res != OC_STACK_OK)
 784     {
 785         OIC_LOG(ERROR, TAG, "OCStack resource error");
 786         OICFree(pDInfo);
 787         return res;
 788     }
 789
 790     //Waiting for each response.
 791     res = PMTimeout(waittime, true);
 792     if(OC_STACK_OK != res)
 793     {
 794         OIC_LOG(ERROR, TAG, "Failed to wait response for secure discovery.");
 795         OICFree(pDInfo);
 796         OCStackResult resCancel = OCCancel(handle, OC_LOW_QOS, NULL, 0);
 797         if(OC_STACK_OK !=  resCancel)
 798         {
 799             OIC_LOG(ERROR, TAG, "Failed to remove registered callback");
 800         }
 801         return res;
 802     }
 803     res = OCCancel(handle,OC_LOW_QOS,NULL,0);
 804     if (OC_STACK_OK != res)
 805     {
 806         OIC_LOG(ERROR, TAG, "Failed to remove registered callback");
 807         OICFree(pDInfo);
 808         return res;
 809     }
 810     OIC_LOG(DEBUG, TAG, "OUT PMDeviceDiscovery");
 811     OICFree(pDInfo);
 812     return res;
 813 }
 814
 815 static OCStackResult SecurePortDiscovery(DiscoveryInfo* discoveryInfo,
 816                                          const OCClientResponse *clientResponse)
 817 {
 818     OIC_LOG(DEBUG, TAG, "IN SecurePortDiscovery");
 819
 820     if(NULL == discoveryInfo || NULL == clientResponse)
 821     {
 822         return OC_STACK_INVALID_PARAM;
 823     }
 824
 825     char rsrc_uri[MAX_URI_LENGTH+1] = {0};
 826     int wr_len = snprintf(rsrc_uri, sizeof(rsrc_uri), "%s?%s=%s",
 827               OC_RSRVD_WELL_KNOWN_URI, OC_RSRVD_RESOURCE_TYPE, OIC_RSRC_TYPE_SEC_DOXM);
 828     if(wr_len <= 0 || (size_t)wr_len >= sizeof(rsrc_uri))
 829     {
 830         OIC_LOG(ERROR, TAG, "rsrc_uri_string_print failed");
 831         return OC_STACK_ERROR;
 832     }
 833     //Try to the unicast discovery to getting secure port
 834     char query[MAX_URI_LENGTH+MAX_QUERY_LENGTH+1] = {0};
 835     if(!PMGenerateQuery(false,
 836                         clientResponse->devAddr.addr, clientResponse->devAddr.port,
 837                         clientResponse->connType,
 838                         query, sizeof(query), rsrc_uri))
 839     {
 840         OIC_LOG(ERROR, TAG, "SecurePortDiscovery : Failed to generate query");
 841         return OC_STACK_ERROR;
 842     }
 843     OIC_LOG_V(DEBUG, TAG, "Query=%s", query);
 844
 845     OCCallbackData cbData;
 846     cbData.cb = &SecurePortDiscoveryHandler;
 847     cbData.context = (void*)discoveryInfo;
 848     cbData.cd = NULL;
 849     OCStackResult ret = OCDoResource(NULL, OC_REST_DISCOVER, query, 0, 0,
 850             clientResponse->connType, OC_LOW_QOS, &cbData, NULL, 0);
 851     if(OC_STACK_OK != ret)
 852     {
 853         OIC_LOG(ERROR, TAG, "Failed to Secure Port Discovery");
 854         return ret;
 855     }
 856     else
 857     {
 858         OIC_LOG_V(INFO, TAG, "OCDoResource with [%s] Success", query);
 859     }
 860
 861     OIC_LOG(DEBUG, TAG, "OUT SecurePortDiscovery");
 862
 863     return ret;
 864 }
 865
 866 static OCStackResult SecurityVersionDiscovery(DiscoveryInfo* discoveryInfo,
 867                                               const OCClientResponse *clientResponse)
 868 {
 869     OIC_LOG(DEBUG, TAG, "IN SecurityVersionDiscovery");
 870
 871     if(NULL == discoveryInfo || NULL == clientResponse)
 872     {
 873         return OC_STACK_INVALID_PARAM;
 874     }
 875
 876     //Try to the unicast discovery to getting security version
 877     char query[MAX_URI_LENGTH+MAX_QUERY_LENGTH+1] = {0};
 878     if(!PMGenerateQuery(false,
 879                         clientResponse->devAddr.addr, clientResponse->devAddr.port,
 880                         clientResponse->connType,
 881                         query, sizeof(query), OIC_RSRC_VER_URI))
 882     {
 883         OIC_LOG(ERROR, TAG, "SecurityVersionDiscovery : Failed to generate query");
 884         return OC_STACK_ERROR;
 885     }
 886     OIC_LOG_V(DEBUG, TAG, "Query=%s", query);
 887
 888     OCCallbackData cbData;
 889     cbData.cb = &SecurityVersionDiscoveryHandler;
 890     cbData.context = (void*)discoveryInfo;
 891     cbData.cd = NULL;
 892     OCStackResult ret = OCDoResource(NULL, OC_REST_DISCOVER, query, 0, 0,
 893             clientResponse->connType, OC_LOW_QOS, &cbData, NULL, 0);
 894     if(OC_STACK_OK != ret)
 895     {
 896         OIC_LOG(ERROR, TAG, "Failed to Security Version Discovery");
 897         return ret;
 898     }
 899     else
 900     {
 901         OIC_LOG_V(INFO, TAG, "OCDoResource with [%s] Success", query);
 902     }
 903
 904     OIC_LOG(DEBUG, TAG, "OUT SecurityVersionDiscovery");
 905
 906     return ret;
 907 }
 908
 909 /**
 910  * Function to print OCProvisionDev_t for debug purpose.
 911  *
 912  * @param[in] pDev Pointer to OCProvisionDev_t. It's information will be printed by OIC_LOG_XX
 913  *
 914  */
 915 void PMPrintOCProvisionDev(const OCProvisionDev_t* pDev)
 916 {
 917     if (pDev)
 918     {
 919         OIC_LOG(DEBUG, TAG, "+++++ OCProvisionDev_t Information +++++");
 920         OIC_LOG_V(DEBUG, TAG, "IP %s", pDev->endpoint.addr);
 921         OIC_LOG_V(DEBUG, TAG, "PORT %d", pDev->endpoint.port);
 922         OIC_LOG_V(DEBUG, TAG, "S-PORT %d", pDev->securePort);
 923         OIC_LOG(DEBUG, TAG, "++++++++++++++++++++++++++++++++++++++++");
 924     }
 925     else
 926     {
 927         OIC_LOG(DEBUG, TAG, "+++++ OCProvisionDev_t is NULL +++++");
 928     }
 929 }
 930
 931 bool PMDeleteFromUUIDList(OCUuidList_t *pUuidList, OicUuid_t *targetId)
 932 {
 933     if(pUuidList == NULL || targetId == NULL)
 934     {
 935         return false;
 936     }
 937     OCUuidList_t *tmp1 = NULL,*tmp2=NULL;
 938     LL_FOREACH_SAFE(pUuidList, tmp1, tmp2)
 939     {
 940         if(0 == memcmp(tmp1->dev.id, targetId->id, sizeof(targetId->id)))
 941         {
 942             LL_DELETE(pUuidList, tmp1);
 943             OICFree(tmp1);
 944             return true;
 945         }
 946     }
 947     return false;
 948 }