Home | History | Annotate | Download | only in api 
      1 /*
      2  * Copyright (C) 2006 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 /** \file
     18   This file consists of implementation of AdbInterfaceEnumObject class that
     19   encapsulates enumerator of USB interfaces available through this API.
     20 */
     21 
     22 #include "stdafx.h"
     23 #include "adb_api.h"
     24 #include "adb_interface_enum.h"
     25 #include "adb_helper_routines.h"
     26 
     27 AdbInterfaceEnumObject::AdbInterfaceEnumObject()
     28     : AdbObjectHandle(AdbObjectTypeInterfaceEnumerator) {
     29   current_interface_ = interfaces_.begin();
     30 }
     31 
     32 AdbInterfaceEnumObject::~AdbInterfaceEnumObject() {
     33 }
     34 
     35 bool AdbInterfaceEnumObject::InitializeEnum(GUID class_id,
     36                                             bool exclude_not_present,
     37                                             bool exclude_removed,
     38                                             bool active_only) {
     39   // Calc flags for SetupDiGetClassDevs
     40   DWORD flags = DIGCF_DEVICEINTERFACE;
     41   if (exclude_not_present)
     42     flags |= DIGCF_PRESENT;
     43 
     44   // Do the enum
     45   bool ret = EnumerateDeviceInterfaces(class_id,
     46                                        flags,
     47                                        exclude_removed,
     48                                        active_only,
     49                                        &interfaces_);
     50 
     51   // If enum was successfull set current enum pointer
     52   // to the beginning of the array
     53   if (ret)
     54     current_interface_ = interfaces_.begin();
     55 
     56   return ret;
     57 }
     58 
     59 bool AdbInterfaceEnumObject::Next(AdbInterfaceInfo* info, ULONG* size) {
     60   // Make sure that it's opened
     61   if (!IsOpened()) {
     62     SetLastError(ERROR_INVALID_HANDLE);
     63     return false;
     64   }
     65 
     66   ATLASSERT(NULL != size);
     67   if (NULL == size) {
     68     SetLastError(ERROR_INVALID_PARAMETER);
     69     return false;
     70   }
     71 
     72   // Lets see if enum is over
     73   if (interfaces_.end() == current_interface_) {
     74     SetLastError(ERROR_NO_MORE_ITEMS);
     75     return false;
     76   }
     77 
     78   AdbInstanceEnumEntry& entry = *current_interface_;
     79 
     80   // Big enough?
     81   if ((NULL == info) || (*size < entry.GetFlatSize())) {
     82     *size = entry.GetFlatSize();
     83     SetLastError(ERROR_INSUFFICIENT_BUFFER);
     84     return false;
     85   }
     86 
     87   // All checks passed
     88   entry.Save(info);
     89   current_interface_++;
     90   return true;
     91 }
     92 
     93 bool AdbInterfaceEnumObject::Reset() {
     94   // Make sure that it's opened
     95   if (!IsOpened()) {
     96     SetLastError(ERROR_INVALID_HANDLE);
     97     return false;
     98   }
     99 
    100   current_interface_ = interfaces_.begin();
    101 
    102   return true;
    103 }
    104