wifi/1.0/vts/functional/wifi_hidl_test_utils.h

   1 /*
   2  * Copyright (C) 2016 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #pragma once
  18
  19 #include <android/hardware/wifi/1.0/IWifi.h>
  20 #include <android/hardware/wifi/1.0/IWifiApIface.h>
  21 #include <android/hardware/wifi/1.0/IWifiChip.h>
  22 #include <android/hardware/wifi/1.0/IWifiNanIface.h>
  23 #include <android/hardware/wifi/1.0/IWifiP2pIface.h>
  24 #include <android/hardware/wifi/1.0/IWifiRttController.h>
  25 #include <android/hardware/wifi/1.0/IWifiStaIface.h>
  26
  27 #include <getopt.h>
  28
  29 // Helper functions to obtain references to the various HIDL interface objects.
  30 // Note: We only have a single instance of each of these objects currently.
  31 // These helper functions should be modified to return vectors if we support
  32 // multiple instances.
  33 android::sp<android::hardware::wifi::V1_0::IWifi> getWifi();
  34 android::sp<android::hardware::wifi::V1_0::IWifiChip> getWifiChip();
  35 android::sp<android::hardware::wifi::V1_0::IWifiApIface> getWifiApIface();
  36 android::sp<android::hardware::wifi::V1_0::IWifiNanIface> getWifiNanIface();
  37 android::sp<android::hardware::wifi::V1_0::IWifiP2pIface> getWifiP2pIface();
  38 android::sp<android::hardware::wifi::V1_0::IWifiStaIface> getWifiStaIface();
  39 android::sp<android::hardware::wifi::V1_0::IWifiRttController>
  40 getWifiRttController();
  41 // Configure the chip in a mode to support the creation of the provided
  42 // iface type.
  43 bool configureChipToSupportIfaceType(
  44     const android::sp<android::hardware::wifi::V1_0::IWifiChip>& wifi_chip,
  45     android::hardware::wifi::V1_0::IfaceType type,
  46     android::hardware::wifi::V1_0::ChipModeId* configured_mode_id);
  47 // Used to trigger IWifi.stop() at the end of every test.
  48 void stopWifi();
  49
  50 class WifiHidlEnvironment : public ::testing::Environment {
  51    protected:
  52     virtual void SetUp() override {
  53         stopWifi();
  54         sleep(5);
  55     }
  56
  57    public:
  58     // Whether NaN feature is supported on the device.
  59     bool isNanOn = false;
  60
  61     void usage(char* me, char* arg) {
  62         fprintf(stderr,
  63                 "unrecognized option: %s\n\n"
  64                 "usage: %s <gtest options> <test options>\n\n"
  65                 "test options are:\n\n"
  66                 "-N, --nan_on: Whether NAN feature is supported\n",
  67                 arg, me);
  68     }
  69
  70     int initFromOptions(int argc, char** argv) {
  71         static struct option options[] = {{"nan_on", no_argument, 0, 'N'},
  72                                           {0, 0, 0, 0}};
  73
  74         int c;
  75         while ((c = getopt_long(argc, argv, "N", options, NULL)) >= 0) {
  76             switch (c) {
  77                 case 'N':
  78                     isNanOn = true;
  79                     break;
  80                 default:
  81                     usage(argv[0], argv[optind]);
  82                     return 2;
  83             }
  84         }
  85
  86         if (optind < argc) {
  87             usage(argv[0], argv[optind]);
  88             return 2;
  89         }
  90
  91         return 0;
  92     }
  93 };