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: SetUp()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 usage(char * me,char * arg)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 initFromOptions(int argc,char ** argv)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 }; 94