245 lines
4.6 KiB
C++
245 lines
4.6 KiB
C++
#ifndef _MIPADDRESSADMIN_HH
|
|
#define _MIPADDRESSADMIN_HH
|
|
|
|
|
|
// Copyright (c) Microsoft. All Rights Reserved
|
|
//
|
|
// THIS IS UNPUBLISHED PROPRIETARY SOURCE CODE OF Microsoft.
|
|
// The copyright notice above does not evidence any
|
|
// actual or intendd publication of such source code.
|
|
//
|
|
// OneLiner : Interface for MIPAddressAdmin.
|
|
// DevUnit : wlbstest
|
|
// Author : Murtaza Hakim
|
|
//
|
|
// Description:
|
|
// ------------
|
|
// Manages adding, deleting and querying for IP addresses on
|
|
// a nic using wmi.
|
|
|
|
|
|
// Include Files
|
|
|
|
#include "MWmiObject.h"
|
|
#include "MWmiInstance.h"
|
|
|
|
#include <vector>
|
|
#include <wbemidl.h>
|
|
#include <comdef.h>
|
|
|
|
using namespace std;
|
|
|
|
|
|
|
|
// Class Definition
|
|
class MIPAddressAdmin
|
|
{
|
|
public:
|
|
|
|
enum MIPAddressAdmin_Error
|
|
{
|
|
MIPAddressAdmin_SUCCESS = 0,
|
|
|
|
COM_FAILURE = 1,
|
|
|
|
UNCONSTRUCTED = 2,
|
|
|
|
NO_SUCH_NIC = 3,
|
|
|
|
NO_SUCH_IP = 4,
|
|
|
|
NOT_SUPPORTED = 5,
|
|
};
|
|
|
|
// Description
|
|
// -----------
|
|
// Constructor.
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// machineIP IN : ip of machine to administer.
|
|
// nicName IN : name of nic to administer.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// none.
|
|
//
|
|
|
|
MIPAddressAdmin( const _bstr_t& machineIP,
|
|
const _bstr_t& nicName );
|
|
|
|
// use this when doing locally.
|
|
//
|
|
MIPAddressAdmin( const _bstr_t& nicName );
|
|
|
|
|
|
// Description
|
|
// -----------
|
|
// Copy Constructor.
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// obj : object to copy.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// none.
|
|
//
|
|
|
|
MIPAddressAdmin( const MIPAddressAdmin& obj );
|
|
|
|
|
|
// Description
|
|
// -----------
|
|
// Assignment operator
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// rhs : object to assign.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// self.
|
|
//
|
|
MIPAddressAdmin&
|
|
operator=(const MIPAddressAdmin& rhs );
|
|
|
|
// Description
|
|
// -----------
|
|
// Destructor
|
|
//
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// none
|
|
//
|
|
// Returns
|
|
// -------
|
|
// none.
|
|
//
|
|
|
|
~MIPAddressAdmin();
|
|
|
|
|
|
// Description
|
|
// -----------
|
|
// Adds IP address to interface.
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// ipAddrToAdd in : ipAddr to add in dotted dec notation.
|
|
// subnetMask in : subnet mask in dotted dec notation.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// success else error code.
|
|
|
|
MIPAddressAdmin_Error
|
|
addIPAddress(const _bstr_t& ipAddrToAdd,
|
|
const _bstr_t& subnetMask);
|
|
|
|
|
|
// Description
|
|
// -----------
|
|
// Deletes IP address from interface.
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// none
|
|
//
|
|
// Returns
|
|
// -------
|
|
// SUCCESS else error code.
|
|
//
|
|
|
|
MIPAddressAdmin_Error
|
|
deleteIPAddress( const _bstr_t& ipAddrToDelete );
|
|
|
|
|
|
// Description
|
|
// -----------
|
|
// Gets all ip addresses on nic
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// ipAddress OUT : list of ip addresses on nic.
|
|
// subnetMask OUT : list of subnet mask for each ip.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// SUCCESS else error code.
|
|
//
|
|
|
|
MIPAddressAdmin_Error
|
|
getIPAddresses( vector<_bstr_t>* ipAddress,
|
|
vector<_bstr_t>* subnetMask );
|
|
|
|
// Description
|
|
// -----------
|
|
// Checks whether the nic we are administering
|
|
// has dhcp enabled or not.
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// dhcpEnabled OUT : is set to true if dhcp, else set to false.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// SUCCESS else error code.
|
|
//
|
|
|
|
MIPAddressAdmin_Error
|
|
isDHCPEnabled( bool& dhcpEnabled );
|
|
|
|
|
|
// Description
|
|
// -----------
|
|
// Enables dhcp on this nic.
|
|
//
|
|
// Parameters
|
|
// ----------
|
|
// none.
|
|
//
|
|
// Returns
|
|
// -------
|
|
// SUCCESS else error code.
|
|
//
|
|
|
|
MIPAddressAdmin_Error
|
|
enableDHCP();
|
|
|
|
|
|
// refresh connection
|
|
//
|
|
|
|
MIPAddressAdmin_Error
|
|
refreshConnection();
|
|
|
|
private:
|
|
|
|
_bstr_t _machineIP;
|
|
_bstr_t _nicName;
|
|
|
|
|
|
MIPAddressAdmin_Error status;
|
|
|
|
MWmiObject machine;
|
|
|
|
MIPAddressAdmin_Error
|
|
checkStatus( vector<MWmiInstance>* nicInstance );
|
|
|
|
};
|
|
//------------------------------------------------------
|
|
//
|
|
//------------------------------------------------------
|
|
// Inline Functions
|
|
//------------------------------------------------------
|
|
//
|
|
//------------------------------------------------------
|
|
// Ensure Type Safety
|
|
//------------------------------------------------------
|
|
typedef class MIPAddressAdmin MIPAddressAdmin;
|
|
//------------------------------------------------------
|
|
//
|
|
#endif _MIPADDRESSADMIN_HH
|