LCOV - code coverage report
Current view: top level - src/wps - wps_dev_attr.c (source / functions) Hit Total Coverage
Test: wpa_supplicant/hostapd combined for hwsim test run 1388613141 Lines: 151 196 77.0 %
Date: 2014-01-02 Functions: 23 23 100.0 %
Branches: 43 72 59.7 %

           Branch data     Line data    Source code
       1                 :            : /*
       2                 :            :  * Wi-Fi Protected Setup - device attributes
       3                 :            :  * Copyright (c) 2008, Jouni Malinen <j@w1.fi>
       4                 :            :  *
       5                 :            :  * This software may be distributed under the terms of the BSD license.
       6                 :            :  * See README for more details.
       7                 :            :  */
       8                 :            : 
       9                 :            : #include "includes.h"
      10                 :            : 
      11                 :            : #include "common.h"
      12                 :            : #include "wps_i.h"
      13                 :            : #include "wps_dev_attr.h"
      14                 :            : 
      15                 :            : 
      16                 :       1461 : int wps_build_manufacturer(struct wps_device_data *dev, struct wpabuf *msg)
      17                 :            : {
      18                 :            :         size_t len;
      19                 :       1461 :         wpa_printf(MSG_DEBUG, "WPS:  * Manufacturer");
      20                 :       1461 :         wpabuf_put_be16(msg, ATTR_MANUFACTURER);
      21         [ +  + ]:       1461 :         len = dev->manufacturer ? os_strlen(dev->manufacturer) : 0;
      22                 :            : #ifndef CONFIG_WPS_STRICT
      23                 :            :         if (len == 0) {
      24                 :            :                 /*
      25                 :            :                  * Some deployed WPS implementations fail to parse zero-length
      26                 :            :                  * attributes. As a workaround, send a space character if the
      27                 :            :                  * device attribute string is empty.
      28                 :            :                  */
      29                 :            :                 wpabuf_put_be16(msg, 1);
      30                 :            :                 wpabuf_put_u8(msg, ' ');
      31                 :            :                 return 0;
      32                 :            :         }
      33                 :            : #endif /* CONFIG_WPS_STRICT */
      34                 :       1461 :         wpabuf_put_be16(msg, len);
      35                 :       1461 :         wpabuf_put_data(msg, dev->manufacturer, len);
      36                 :       1461 :         return 0;
      37                 :            : }
      38                 :            : 
      39                 :            : 
      40                 :       1461 : int wps_build_model_name(struct wps_device_data *dev, struct wpabuf *msg)
      41                 :            : {
      42                 :            :         size_t len;
      43                 :       1461 :         wpa_printf(MSG_DEBUG, "WPS:  * Model Name");
      44                 :       1461 :         wpabuf_put_be16(msg, ATTR_MODEL_NAME);
      45         [ +  + ]:       1461 :         len = dev->model_name ? os_strlen(dev->model_name) : 0;
      46                 :            : #ifndef CONFIG_WPS_STRICT
      47                 :            :         if (len == 0) {
      48                 :            :                 /*
      49                 :            :                  * Some deployed WPS implementations fail to parse zero-length
      50                 :            :                  * attributes. As a workaround, send a space character if the
      51                 :            :                  * device attribute string is empty.
      52                 :            :                  */
      53                 :            :                 wpabuf_put_be16(msg, 1);
      54                 :            :                 wpabuf_put_u8(msg, ' ');
      55                 :            :                 return 0;
      56                 :            :         }
      57                 :            : #endif /* CONFIG_WPS_STRICT */
      58                 :       1461 :         wpabuf_put_be16(msg, len);
      59                 :       1461 :         wpabuf_put_data(msg, dev->model_name, len);
      60                 :       1461 :         return 0;
      61                 :            : }
      62                 :            : 
      63                 :            : 
      64                 :       1461 : int wps_build_model_number(struct wps_device_data *dev, struct wpabuf *msg)
      65                 :            : {
      66                 :            :         size_t len;
      67                 :       1461 :         wpa_printf(MSG_DEBUG, "WPS:  * Model Number");
      68                 :       1461 :         wpabuf_put_be16(msg, ATTR_MODEL_NUMBER);
      69         [ +  + ]:       1461 :         len = dev->model_number ? os_strlen(dev->model_number) : 0;
      70                 :            : #ifndef CONFIG_WPS_STRICT
      71                 :            :         if (len == 0) {
      72                 :            :                 /*
      73                 :            :                  * Some deployed WPS implementations fail to parse zero-length
      74                 :            :                  * attributes. As a workaround, send a space character if the
      75                 :            :                  * device attribute string is empty.
      76                 :            :                  */
      77                 :            :                 wpabuf_put_be16(msg, 1);
      78                 :            :                 wpabuf_put_u8(msg, ' ');
      79                 :            :                 return 0;
      80                 :            :         }
      81                 :            : #endif /* CONFIG_WPS_STRICT */
      82                 :       1461 :         wpabuf_put_be16(msg, len);
      83                 :       1461 :         wpabuf_put_data(msg, dev->model_number, len);
      84                 :       1461 :         return 0;
      85                 :            : }
      86                 :            : 
      87                 :            : 
      88                 :        522 : static int wps_build_serial_number(struct wps_device_data *dev,
      89                 :            :                                    struct wpabuf *msg)
      90                 :            : {
      91                 :            :         size_t len;
      92                 :        522 :         wpa_printf(MSG_DEBUG, "WPS:  * Serial Number");
      93                 :        522 :         wpabuf_put_be16(msg, ATTR_SERIAL_NUMBER);
      94         [ +  + ]:        522 :         len = dev->serial_number ? os_strlen(dev->serial_number) : 0;
      95                 :            : #ifndef CONFIG_WPS_STRICT
      96                 :            :         if (len == 0) {
      97                 :            :                 /*
      98                 :            :                  * Some deployed WPS implementations fail to parse zero-length
      99                 :            :                  * attributes. As a workaround, send a space character if the
     100                 :            :                  * device attribute string is empty.
     101                 :            :                  */
     102                 :            :                 wpabuf_put_be16(msg, 1);
     103                 :            :                 wpabuf_put_u8(msg, ' ');
     104                 :            :                 return 0;
     105                 :            :         }
     106                 :            : #endif /* CONFIG_WPS_STRICT */
     107                 :        522 :         wpabuf_put_be16(msg, len);
     108                 :        522 :         wpabuf_put_data(msg, dev->serial_number, len);
     109                 :        522 :         return 0;
     110                 :            : }
     111                 :            : 
     112                 :            : 
     113                 :       1678 : int wps_build_primary_dev_type(struct wps_device_data *dev, struct wpabuf *msg)
     114                 :            : {
     115                 :       1678 :         wpa_printf(MSG_DEBUG, "WPS:  * Primary Device Type");
     116                 :       1678 :         wpabuf_put_be16(msg, ATTR_PRIMARY_DEV_TYPE);
     117                 :       1678 :         wpabuf_put_be16(msg, WPS_DEV_TYPE_LEN);
     118                 :       1678 :         wpabuf_put_data(msg, dev->pri_dev_type, WPS_DEV_TYPE_LEN);
     119                 :       1678 :         return 0;
     120                 :            : }
     121                 :            : 
     122                 :            : 
     123                 :        939 : int wps_build_secondary_dev_type(struct wps_device_data *dev,
     124                 :            :                                   struct wpabuf *msg)
     125                 :            : {
     126         [ +  - ]:        939 :         if (!dev->num_sec_dev_types)
     127                 :        939 :                 return 0;
     128                 :            : 
     129                 :          0 :         wpa_printf(MSG_DEBUG, "WPS:  * Secondary Device Type");
     130                 :          0 :         wpabuf_put_be16(msg, ATTR_SECONDARY_DEV_TYPE_LIST);
     131                 :          0 :         wpabuf_put_be16(msg, WPS_DEV_TYPE_LEN * dev->num_sec_dev_types);
     132                 :          0 :         wpabuf_put_data(msg, dev->sec_dev_type,
     133                 :          0 :                         WPS_DEV_TYPE_LEN * dev->num_sec_dev_types);
     134                 :            : 
     135                 :        939 :         return 0;
     136                 :            : }
     137                 :            : 
     138                 :            : 
     139                 :        939 : int wps_build_req_dev_type(struct wps_device_data *dev, struct wpabuf *msg,
     140                 :            :                            unsigned int num_req_dev_types,
     141                 :            :                            const u8 *req_dev_types)
     142                 :            : {
     143                 :            :         unsigned int i;
     144                 :            : 
     145         [ -  + ]:        939 :         for (i = 0; i < num_req_dev_types; i++) {
     146                 :          0 :                 wpa_hexdump(MSG_DEBUG, "WPS: * Requested Device Type",
     147                 :          0 :                             req_dev_types + i * WPS_DEV_TYPE_LEN,
     148                 :            :                             WPS_DEV_TYPE_LEN);
     149                 :          0 :                 wpabuf_put_be16(msg, ATTR_REQUESTED_DEV_TYPE);
     150                 :          0 :                 wpabuf_put_be16(msg, WPS_DEV_TYPE_LEN);
     151                 :          0 :                 wpabuf_put_data(msg, req_dev_types + i * WPS_DEV_TYPE_LEN,
     152                 :            :                                 WPS_DEV_TYPE_LEN);
     153                 :            :         }
     154                 :            : 
     155                 :        939 :         return 0;
     156                 :            : }
     157                 :            : 
     158                 :            : 
     159                 :       1678 : int wps_build_dev_name(struct wps_device_data *dev, struct wpabuf *msg)
     160                 :            : {
     161                 :            :         size_t len;
     162                 :       1678 :         wpa_printf(MSG_DEBUG, "WPS:  * Device Name");
     163                 :       1678 :         wpabuf_put_be16(msg, ATTR_DEV_NAME);
     164         [ +  + ]:       1678 :         len = dev->device_name ? os_strlen(dev->device_name) : 0;
     165                 :            : #ifndef CONFIG_WPS_STRICT
     166                 :            :         if (len == 0) {
     167                 :            :                 /*
     168                 :            :                  * Some deployed WPS implementations fail to parse zero-length
     169                 :            :                  * attributes. As a workaround, send a space character if the
     170                 :            :                  * device attribute string is empty.
     171                 :            :                  */
     172                 :            :                 wpabuf_put_be16(msg, 1);
     173                 :            :                 wpabuf_put_u8(msg, ' ');
     174                 :            :                 return 0;
     175                 :            :         }
     176                 :            : #endif /* CONFIG_WPS_STRICT */
     177                 :       1678 :         wpabuf_put_be16(msg, len);
     178                 :       1678 :         wpabuf_put_data(msg, dev->device_name, len);
     179                 :       1678 :         return 0;
     180                 :            : }
     181                 :            : 
     182                 :            : 
     183                 :        522 : int wps_build_device_attrs(struct wps_device_data *dev, struct wpabuf *msg)
     184                 :            : {
     185   [ +  -  +  - ]:       1044 :         if (wps_build_manufacturer(dev, msg) ||
     186         [ +  - ]:       1044 :             wps_build_model_name(dev, msg) ||
     187         [ +  - ]:       1044 :             wps_build_model_number(dev, msg) ||
     188         [ +  - ]:       1044 :             wps_build_serial_number(dev, msg) ||
     189         [ -  + ]:       1044 :             wps_build_primary_dev_type(dev, msg) ||
     190                 :        522 :             wps_build_dev_name(dev, msg))
     191                 :          0 :                 return -1;
     192                 :        522 :         return 0;
     193                 :            : }
     194                 :            : 
     195                 :            : 
     196                 :        193 : int wps_build_os_version(struct wps_device_data *dev, struct wpabuf *msg)
     197                 :            : {
     198                 :        193 :         wpa_printf(MSG_DEBUG, "WPS:  * OS Version");
     199                 :        193 :         wpabuf_put_be16(msg, ATTR_OS_VERSION);
     200                 :        193 :         wpabuf_put_be16(msg, 4);
     201                 :        193 :         wpabuf_put_be32(msg, 0x80000000 | dev->os_version);
     202                 :        193 :         return 0;
     203                 :            : }
     204                 :            : 
     205                 :            : 
     206                 :         97 : int wps_build_vendor_ext_m1(struct wps_device_data *dev, struct wpabuf *msg)
     207                 :            : {
     208         [ -  + ]:         97 :         if (dev->vendor_ext_m1 != NULL) {
     209                 :          0 :                 wpa_hexdump(MSG_DEBUG, "WPS:  * Vendor Extension M1",
     210                 :          0 :                             wpabuf_head_u8(dev->vendor_ext_m1),
     211                 :          0 :                             wpabuf_len(dev->vendor_ext_m1));
     212                 :          0 :                 wpabuf_put_be16(msg, ATTR_VENDOR_EXT);
     213                 :          0 :                 wpabuf_put_be16(msg, wpabuf_len(dev->vendor_ext_m1));
     214                 :          0 :                 wpabuf_put_buf(msg, dev->vendor_ext_m1);
     215                 :            :         }
     216                 :         97 :         return 0;
     217                 :            : }
     218                 :            : 
     219                 :            : 
     220                 :       1154 : int wps_build_rf_bands(struct wps_device_data *dev, struct wpabuf *msg,
     221                 :            :                        u8 rf_band)
     222                 :            : {
     223                 :       1154 :         wpa_printf(MSG_DEBUG, "WPS:  * RF Bands (%x)", dev->rf_bands);
     224                 :       1154 :         wpabuf_put_be16(msg, ATTR_RF_BANDS);
     225                 :       1154 :         wpabuf_put_be16(msg, 1);
     226         [ +  + ]:       1154 :         wpabuf_put_u8(msg, rf_band ? rf_band : dev->rf_bands);
     227                 :       1154 :         return 0;
     228                 :            : }
     229                 :            : 
     230                 :            : 
     231                 :        658 : int wps_build_vendor_ext(struct wps_device_data *dev, struct wpabuf *msg)
     232                 :            : {
     233                 :            :         int i;
     234                 :            : 
     235         [ +  + ]:       7238 :         for (i = 0; i < MAX_WPS_VENDOR_EXTENSIONS; i++) {
     236         [ +  - ]:       6580 :                 if (dev->vendor_ext[i] == NULL)
     237                 :       6580 :                         continue;
     238                 :          0 :                 wpa_hexdump(MSG_DEBUG, "WPS:  * Vendor Extension",
     239                 :          0 :                             wpabuf_head_u8(dev->vendor_ext[i]),
     240                 :          0 :                             wpabuf_len(dev->vendor_ext[i]));
     241                 :          0 :                 wpabuf_put_be16(msg, ATTR_VENDOR_EXT);
     242                 :          0 :                 wpabuf_put_be16(msg, wpabuf_len(dev->vendor_ext[i]));
     243                 :          0 :                 wpabuf_put_buf(msg, dev->vendor_ext[i]);
     244                 :            :         }
     245                 :            : 
     246                 :        658 :         return 0;
     247                 :            : }
     248                 :            : 
     249                 :            : 
     250                 :        182 : static int wps_process_manufacturer(struct wps_device_data *dev, const u8 *str,
     251                 :            :                                     size_t str_len)
     252                 :            : {
     253         [ -  + ]:        182 :         if (str == NULL) {
     254                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No Manufacturer received");
     255                 :          0 :                 return -1;
     256                 :            :         }
     257                 :            : 
     258                 :        182 :         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Manufacturer", str, str_len);
     259                 :            : 
     260                 :        182 :         os_free(dev->manufacturer);
     261                 :        182 :         dev->manufacturer = dup_binstr(str, str_len);
     262         [ -  + ]:        182 :         if (dev->manufacturer == NULL)
     263                 :          0 :                 return -1;
     264                 :            : 
     265                 :        182 :         return 0;
     266                 :            : }
     267                 :            : 
     268                 :            : 
     269                 :        182 : static int wps_process_model_name(struct wps_device_data *dev, const u8 *str,
     270                 :            :                                   size_t str_len)
     271                 :            : {
     272         [ -  + ]:        182 :         if (str == NULL) {
     273                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No Model Name received");
     274                 :          0 :                 return -1;
     275                 :            :         }
     276                 :            : 
     277                 :        182 :         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Model Name", str, str_len);
     278                 :            : 
     279                 :        182 :         os_free(dev->model_name);
     280                 :        182 :         dev->model_name = dup_binstr(str, str_len);
     281         [ -  + ]:        182 :         if (dev->model_name == NULL)
     282                 :          0 :                 return -1;
     283                 :            : 
     284                 :        182 :         return 0;
     285                 :            : }
     286                 :            : 
     287                 :            : 
     288                 :        182 : static int wps_process_model_number(struct wps_device_data *dev, const u8 *str,
     289                 :            :                                     size_t str_len)
     290                 :            : {
     291         [ -  + ]:        182 :         if (str == NULL) {
     292                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No Model Number received");
     293                 :          0 :                 return -1;
     294                 :            :         }
     295                 :            : 
     296                 :        182 :         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Model Number", str, str_len);
     297                 :            : 
     298                 :        182 :         os_free(dev->model_number);
     299                 :        182 :         dev->model_number = dup_binstr(str, str_len);
     300         [ -  + ]:        182 :         if (dev->model_number == NULL)
     301                 :          0 :                 return -1;
     302                 :            : 
     303                 :        182 :         return 0;
     304                 :            : }
     305                 :            : 
     306                 :            : 
     307                 :        182 : static int wps_process_serial_number(struct wps_device_data *dev,
     308                 :            :                                      const u8 *str, size_t str_len)
     309                 :            : {
     310         [ -  + ]:        182 :         if (str == NULL) {
     311                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No Serial Number received");
     312                 :          0 :                 return -1;
     313                 :            :         }
     314                 :            : 
     315                 :        182 :         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Serial Number", str, str_len);
     316                 :            : 
     317                 :        182 :         os_free(dev->serial_number);
     318                 :        182 :         dev->serial_number = dup_binstr(str, str_len);
     319         [ -  + ]:        182 :         if (dev->serial_number == NULL)
     320                 :          0 :                 return -1;
     321                 :            : 
     322                 :        182 :         return 0;
     323                 :            : }
     324                 :            : 
     325                 :            : 
     326                 :        182 : static int wps_process_dev_name(struct wps_device_data *dev, const u8 *str,
     327                 :            :                                 size_t str_len)
     328                 :            : {
     329         [ -  + ]:        182 :         if (str == NULL) {
     330                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No Device Name received");
     331                 :          0 :                 return -1;
     332                 :            :         }
     333                 :            : 
     334                 :        182 :         wpa_hexdump_ascii(MSG_DEBUG, "WPS: Device Name", str, str_len);
     335                 :            : 
     336                 :        182 :         os_free(dev->device_name);
     337                 :        182 :         dev->device_name = dup_binstr(str, str_len);
     338         [ -  + ]:        182 :         if (dev->device_name == NULL)
     339                 :          0 :                 return -1;
     340                 :            : 
     341                 :        182 :         return 0;
     342                 :            : }
     343                 :            : 
     344                 :            : 
     345                 :        182 : static int wps_process_primary_dev_type(struct wps_device_data *dev,
     346                 :            :                                         const u8 *dev_type)
     347                 :            : {
     348                 :            : #ifndef CONFIG_NO_STDOUT_DEBUG
     349                 :            :         char devtype[WPS_DEV_TYPE_BUFSIZE];
     350                 :            : #endif /* CONFIG_NO_STDOUT_DEBUG */
     351                 :            : 
     352         [ -  + ]:        182 :         if (dev_type == NULL) {
     353                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No Primary Device Type received");
     354                 :          0 :                 return -1;
     355                 :            :         }
     356                 :            : 
     357                 :        182 :         os_memcpy(dev->pri_dev_type, dev_type, WPS_DEV_TYPE_LEN);
     358                 :        182 :         wpa_printf(MSG_DEBUG, "WPS: Primary Device Type: %s",
     359                 :        182 :                    wps_dev_type_bin2str(dev->pri_dev_type, devtype,
     360                 :            :                                         sizeof(devtype)));
     361                 :            : 
     362                 :        182 :         return 0;
     363                 :            : }
     364                 :            : 
     365                 :            : 
     366                 :        182 : int wps_process_device_attrs(struct wps_device_data *dev,
     367                 :            :                              struct wps_parse_attr *attr)
     368                 :            : {
     369         [ +  - ]:        182 :         if (wps_process_manufacturer(dev, attr->manufacturer,
     370         [ +  - ]:        182 :                                      attr->manufacturer_len) ||
     371                 :        182 :             wps_process_model_name(dev, attr->model_name,
     372         [ +  - ]:        182 :                                    attr->model_name_len) ||
     373                 :        182 :             wps_process_model_number(dev, attr->model_number,
     374         [ +  - ]:        182 :                                      attr->model_number_len) ||
     375                 :        182 :             wps_process_serial_number(dev, attr->serial_number,
     376         [ +  - ]:        182 :                                       attr->serial_number_len) ||
     377         [ -  + ]:        364 :             wps_process_primary_dev_type(dev, attr->primary_dev_type) ||
     378                 :        182 :             wps_process_dev_name(dev, attr->dev_name, attr->dev_name_len))
     379                 :          0 :                 return -1;
     380                 :        182 :         return 0;
     381                 :            : }
     382                 :            : 
     383                 :            : 
     384                 :         96 : int wps_process_os_version(struct wps_device_data *dev, const u8 *ver)
     385                 :            : {
     386         [ -  + ]:         96 :         if (ver == NULL) {
     387                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No OS Version received");
     388                 :          0 :                 return -1;
     389                 :            :         }
     390                 :            : 
     391                 :         96 :         dev->os_version = WPA_GET_BE32(ver);
     392                 :         96 :         wpa_printf(MSG_DEBUG, "WPS: OS Version %08x", dev->os_version);
     393                 :            : 
     394                 :         96 :         return 0;
     395                 :            : }
     396                 :            : 
     397                 :            : 
     398                 :         96 : int wps_process_rf_bands(struct wps_device_data *dev, const u8 *bands)
     399                 :            : {
     400         [ -  + ]:         96 :         if (bands == NULL) {
     401                 :          0 :                 wpa_printf(MSG_DEBUG, "WPS: No RF Bands received");
     402                 :          0 :                 return -1;
     403                 :            :         }
     404                 :            : 
     405                 :         96 :         dev->rf_bands = *bands;
     406                 :         96 :         wpa_printf(MSG_DEBUG, "WPS: Enrollee RF Bands 0x%x", dev->rf_bands);
     407                 :            : 
     408                 :         96 :         return 0;
     409                 :            : }
     410                 :            : 
     411                 :            : 
     412                 :        368 : void wps_device_data_free(struct wps_device_data *dev)
     413                 :            : {
     414                 :        368 :         os_free(dev->device_name);
     415                 :        368 :         dev->device_name = NULL;
     416                 :        368 :         os_free(dev->manufacturer);
     417                 :        368 :         dev->manufacturer = NULL;
     418                 :        368 :         os_free(dev->model_name);
     419                 :        368 :         dev->model_name = NULL;
     420                 :        368 :         os_free(dev->model_number);
     421                 :        368 :         dev->model_number = NULL;
     422                 :        368 :         os_free(dev->serial_number);
     423                 :        368 :         dev->serial_number = NULL;
     424                 :        368 : }

Generated by: LCOV version 1.9