Home | History | Annotate | Download | only in functional 
      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 #include <VtsHalHidlTargetTestEnvBase.h>
     30 // Helper functions to obtain references to the various HIDL interface objects.
     31 // Note: We only have a single instance of each of these objects currently.
     32 // These helper functions should be modified to return vectors if we support
     33 // multiple instances.
     34 android::sp<android::hardware::wifi::V1_0::IWifi> getWifi();
     35 android::sp<android::hardware::wifi::V1_0::IWifiChip> getWifiChip();
     36 android::sp<android::hardware::wifi::V1_0::IWifiApIface> getWifiApIface();
     37 android::sp<android::hardware::wifi::V1_0::IWifiNanIface> getWifiNanIface();
     38 android::sp<android::hardware::wifi::V1_0::IWifiP2pIface> getWifiP2pIface();
     39 android::sp<android::hardware::wifi::V1_0::IWifiStaIface> getWifiStaIface();
     40 android::sp<android::hardware::wifi::V1_0::IWifiRttController>
     41 getWifiRttController();
     42 // Configure the chip in a mode to support the creation of the provided
     43 // iface type.
     44 bool configureChipToSupportIfaceType(
     45     const android::sp<android::hardware::wifi::V1_0::IWifiChip>& wifi_chip,
     46     android::hardware::wifi::V1_0::IfaceType type,
     47     android::hardware::wifi::V1_0::ChipModeId* configured_mode_id);
     48 // Used to trigger IWifi.stop() at the end of every test.
     49 void stopWifi();
     50 
     51 class WifiHidlEnvironment : public ::testing::VtsHalHidlTargetTestEnvBase {
     52    protected:
     53     virtual void HidlSetUp() override {
     54         stopWifi();
     55         sleep(5);
     56     }
     57 
     58    public:
     59     // Whether NaN feature is supported on the device.
     60     bool isNanOn = false;
     61 
     62     void usage(char* me, char* arg) {
     63         fprintf(stderr,
     64                 "unrecognized option: %s\n\n"
     65                 "usage: %s <gtest options> <test options>\n\n"
     66                 "test options are:\n\n"
     67                 "-N, --nan_on: Whether NAN feature is supported\n",
     68                 arg, me);
     69     }
     70 
     71     int initFromOptions(int argc, char** argv) {
     72         static struct option options[] = {{"nan_on", no_argument, 0, 'N'},
     73                                           {0, 0, 0, 0}};
     74 
     75         int c;
     76         while ((c = getopt_long(argc, argv, "N", options, NULL)) >= 0) {
     77             switch (c) {
     78                 case 'N':
     79                     isNanOn = true;
     80                     break;
     81                 default:
     82                     usage(argv[0], argv[optind]);
     83                     return 2;
     84             }
     85         }
     86 
     87         if (optind < argc) {
     88             usage(argv[0], argv[optind]);
     89             return 2;
     90         }
     91 
     92         return 0;
     93     }
     94 };
     95