DhcpEnumSubnetClientsV5中的内存泄漏通过P / Invoke

时间:2013-08-23 10:23:31

标签: c# .net memory-leaks pinvoke dhcp

我想通过C#(.NET 3.5,x86)应用程序枚举DHCP服务器(Windows 2008 R2)中的所有DHCP租约。

由于遗憾的是,.NET中没有DHCP服务器类,我使用P / Invoke(dhcpsapi.dll)从DHCP服务器获取数据。

如果我将DhcpEnumSubnetClients函数与DHCP_CLIENT_INFO和DHCP_CLIENT_INFO_ARRAY结构一起使用,那么一切都很好。

但是如果我使用相应的V5函数和结构(DhcpEnumSubnetClientsV5,DHCP_CLIENT_INFO_V5和DHCP_CLIENT_INFO_ARRAY_V5),我会得到一个讨厌的内存泄漏(每个枚举租约大约280-290字节)。

我需要使用V5版本,因为我需要来自DHCP_CLIENT_INFO_V5结构的信息,这在DHCP_CLIENT_INFO结构(bClientType和AddressState)中是不可用的。

在调用DhcpEnumSubnetClientsV5之后,我根据需要调用了DhcpRpcFreeMemory,但是当与DhcpEnumSubnetClientsV5一起使用时,DhcpRpcFreeMemory似乎没有做任何事情。

这是.NET 3.5或dhcpsapi.dll中的错误还是我的代码中缺少的内容?

非常感谢任何帮助!

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;


using System.Data;
using System.Runtime.InteropServices;
using System.Net.NetworkInformation;
using System.Text.RegularExpressions;


class Program
{
    [DllImport("dhcpsapi.dll", CharSet = CharSet.Unicode)]
    static extern UInt32 DhcpEnumSubnetClients(
    string ServerIpAddress,
    UInt32 SubnetAddress,
    ref UInt32 ResumeHandle,
    UInt32 PreferredMaximum,
    out IntPtr ClientInfo,
    out UInt32 ElementsRead,
    out UInt32 ElementsTotal);

    [DllImport("dhcpsapi.dll", CharSet = CharSet.Unicode)]
    static extern UInt32 DhcpEnumSubnetClientsV5(
    string ServerIpAddress,
    UInt32 SubnetAddress,
    ref UInt32 ResumeHandle,
    UInt32 PreferredMaximum,
    out IntPtr ClientInfo,
    out UInt32 ElementsRead,
    out UInt32 ElementsTotal);

    [DllImport("dhcpsapi.dll")]
    static extern void DhcpRpcFreeMemory(IntPtr BuffPtr);

    [StructLayout(LayoutKind.Sequential)]
    private struct DHCP_CLIENT_INFO_ARRAY_V5
    {
        public UInt32 NumElements;
        public IntPtr Elements;
    }

    [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
    private struct DHCP_CLIENT_INFO_V5
    {
        public UInt32 ClientIpAddress;
        public UInt32 SubnetMask;
        public DHCP_CLIENT_UID ClientHardwareAddress;
        public string ClientName;
        public string ClientComment;
        public DATE_TIME ClientLeaseExpires;
        public DHCP_HOST_INFO OwnerHost;
        public byte bClientType;
        public byte AddressState;
    }

    [StructLayout(LayoutKind.Sequential)]
    private struct DHCP_CLIENT_INFO_ARRAY
    {
        public UInt32 NumElements;
        public IntPtr Elements;
    }

    [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
    private struct DHCP_CLIENT_INFO
    {
        public UInt32 ClientIpAddress;
        public UInt32 SubnetMask;
        public DHCP_CLIENT_UID ClientHardwareAddress;
        public string ClientName;
        public string ClientComment;
        public DATE_TIME ClientLeaseExpires;
        public DHCP_HOST_INFO OwnerHost;
    }

    [StructLayout(LayoutKind.Sequential)]
    private struct DHCP_CLIENT_UID
    {
        public UInt32 DataLength;
        public IntPtr Data;
    }

    [StructLayout(LayoutKind.Sequential)]
    private struct DATE_TIME
    {
        public UInt32 dwLowDateTime;
        public UInt32 dwHighDateTime;
        public DATE_TIME(DateTime date)
        {
            if (date == DateTime.MaxValue)
            {
                this.dwLowDateTime = this.dwHighDateTime = UInt32.MaxValue;
                return;
            }
            if (date == DateTime.MinValue)
            {
                this.dwLowDateTime = this.dwHighDateTime = UInt32.MinValue;
                return;
            }
            this.dwLowDateTime = (UInt32)(date.ToFileTime() & 0xFFFFFFFF);
            this.dwHighDateTime = (UInt32)((date.ToFileTime() & 0x7FFFFFFF00000000) >> 32);
        }
        public DateTime Convert()
        {
            if (this.dwHighDateTime == 0 && this.dwLowDateTime == 0)
                return DateTime.MinValue;
            if (this.dwHighDateTime == int.MaxValue && this.dwLowDateTime == UInt32.MaxValue)
                return DateTime.MaxValue;
            return DateTime.FromFileTime((((long)this.dwHighDateTime) << 32) | (UInt32)this.dwLowDateTime);
        }
        public override string ToString()
        {
            return this.Convert().ToString();
        }
    }

    [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
    private struct DHCP_HOST_INFO
    {
        public UInt32 IpAddress;
        public string NetBiosName;
        public string HostName;
    }




    /////////////////////////////////////////////////////////////////////////////////
    //
    // Main
    //
    /////////////////////////////////////////////////////////////////////////////////
    static void Main(string[] args)
    {
        Console.Write("Server: ");
        string server = Console.ReadLine();

        Console.Write("Subnet: ");
        string subnet = Console.ReadLine();


        // 10 iterations for subnet with 150 leases:    5 MB RAM (DhcpEnumSubnetClients),     5 MB RAM (DhcpEnumSubnetClientsV5)
        // 100 iterations for subnet with 150 leases:   7 MB RAM (DhcpEnumSubnetClients),    11 MB RAM (DhcpEnumSubnetClientsV5)
        // 200 iterations for subnet with 150 leases:   7 MB RAM (DhcpEnumSubnetClients),    16 MB RAM (DhcpEnumSubnetClientsV5)
        // 500 iterations for subnet with 150 leases:   7 MB RAM (DhcpEnumSubnetClients),    28 MB RAM (DhcpEnumSubnetClientsV5)
        // -> 40-45 KB leaked per iteration for DhcpEnumSubnetClientsV5; 273-307 bytes per lease

        // 10 iterations for subnet with 1100 leases:   8 MB RAM (DhcpEnumSubnetClients),   11 MB RAM (DhcpEnumSubnetClientsV5)
        // 100 iterations for subnet with 1100 leases:  8 MB RAM (DhcpEnumSubnetClients),   38 MB RAM (DhcpEnumSubnetClientsV5)
        // 200 iterations for subnet with 1100 leases:  8 MB RAM (DhcpEnumSubnetClients),   68 MB RAM (DhcpEnumSubnetClientsV5)
        // 500 iterations for subnet with 1100 leases:  8 MB RAM (DhcpEnumSubnetClients),   157 MB RAM (DhcpEnumSubnetClientsV5)
        // -> 305-307 KB leaked per iteration for DhcpEnumSubnetClientsV5; 283-285 bytes per lease


        // -> around 280-290 bytes leaked per lease enumerated by DhcpEnumSubnetClientsV5

        for (int k = 0; k < 10; k++)
        {
            DataTable dhcpLeases = new DataTable();
            dhcpLeases = GetSubnetLeases(server, subnet);

            if (dhcpLeases == null || dhcpLeases.Rows.Count < 1)
            {
                Console.WriteLine("No Leases found");
                Environment.Exit(0);
            }


            for (int i = 0; i < dhcpLeases.Rows.Count; i++)
            {
                Console.WriteLine("MAC Address: " + dhcpLeases.Rows[i]["macAddress"] + ", IP Address: " + dhcpLeases.Rows[i]["ipAddress"]);
            }

            Console.WriteLine(dhcpLeases.Rows.Count + " Leases found");
        }

        Console.Write("Any key to exit");
        Console.ReadLine();
    }




    /////////////////////////////////////////////////////////////////////////////////
    //
    // Get all Leases for "subnet" on "server"
    // Returns DataTable containing Leases or null if there was a problem
    //
    /////////////////////////////////////////////////////////////////////////////////
    public static DataTable GetSubnetLeases(string server, string subnet)
    {
        UInt32 response = 0, resumeHandle = 0;
        UInt32 nRead = 0, nTotal = 0;

        DataTable leases = new DataTable();
        leases.Columns.Add("ipAddress");
        leases.Columns.Add("macAddress");
        leases.Columns.Add("subnet");
        leases.Columns.Add("clientName");

        for (; ; )
        {
            IntPtr retPtr = IntPtr.Zero;

            response = DhcpEnumSubnetClientsV5(server, ConvertIP(subnet), ref resumeHandle, 65536, out retPtr, out nRead, out nTotal);      // ---- Memory Leak ----
//            response = DhcpEnumSubnetClients(server, ConvertIP(subnet), ref resumeHandle, 65536, out retPtr, out nRead, out nTotal);      // No Memory Leak

            if (response == 259)                                                //ERROR_NO_MORE_ITEMS = 259
            {
                if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
                break;
            }

            if (response != 0 && response != 234 || retPtr == IntPtr.Zero)      //ERROR_MORE_DATA = 234
            {
                if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
                return null;
            }

            DHCP_CLIENT_INFO_ARRAY_V5 nativeArray = (DHCP_CLIENT_INFO_ARRAY_V5)Marshal.PtrToStructure(retPtr, typeof(DHCP_CLIENT_INFO_ARRAY_V5));   // ---- Memory Leak ----
//            DHCP_CLIENT_INFO_ARRAY nativeArray = (DHCP_CLIENT_INFO_ARRAY)Marshal.PtrToStructure(retPtr, typeof(DHCP_CLIENT_INFO_ARRAY));          // No Memory Leak

            for (int i = 0; i < nativeArray.NumElements; i++)
            {
                IntPtr iPtr = Marshal.ReadIntPtr(nativeArray.Elements, (i * Marshal.SizeOf(typeof(IntPtr))));

                DHCP_CLIENT_INFO_V5 nativeClient = (DHCP_CLIENT_INFO_V5)Marshal.PtrToStructure(iPtr, typeof(DHCP_CLIENT_INFO_V5));  // ---- Memory Leak ----
//                DHCP_CLIENT_INFO nativeClient = (DHCP_CLIENT_INFO)Marshal.PtrToStructure(iPtr, typeof(DHCP_CLIENT_INFO));         // No Memory Leak

                string ipAddress = ConvertIP(nativeClient.ClientIpAddress);
                string clientName = nativeClient.ClientName;
                byte[] macBytes = new byte[nativeClient.ClientHardwareAddress.DataLength];
                Marshal.Copy(nativeClient.ClientHardwareAddress.Data, macBytes, 0, macBytes.Length);
                PhysicalAddress mac = new PhysicalAddress(macBytes);
                string macAddress = mac.ToString().ToUpper();
                macAddress = Regex.Replace(macAddress, "(.{2})(.{2})(.{2})(.{2})(.{2})(.{2})", "$1:$2:$3:$4:$5:$6");

                leases.Rows.Add(ipAddress, macAddress, subnet, clientName);
            }

            if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
            if (response == 0) break;
        }
        return leases;
    }


    /////////////////////////////////////////////////////////////////////////////////
    //
    // Convert IP Address in w.x.y.z format to UInt32
    //
    /////////////////////////////////////////////////////////////////////////////////
    public static UInt32 ConvertIP(string ipString)
    {
        string[] ipArray = ipString.Split('.');
        UInt32 ipUint32 = (Convert.ToUInt32(ipArray[0]) * 16777216) + (Convert.ToUInt32(ipArray[1]) * 65536) + (Convert.ToUInt32(ipArray[2]) * 256) + (Convert.ToUInt32(ipArray[3]));
        return ipUint32;
    }


    /////////////////////////////////////////////////////////////////////////////////
    //
    // Convert IP Address from UInt32 to string in w.x.y.z format
    //
    /////////////////////////////////////////////////////////////////////////////////
    public static string ConvertIP(UInt32 ipUInt32)
    {
        string ipString = "";
        UInt32 octet = 0;

        octet = ipUInt32 / 16777216;
        ipString = ipString + octet.ToString() + ".";
        ipUInt32 = ipUInt32 - (octet * 16777216);

        octet = ipUInt32 / 65536;
        ipString = ipString + octet.ToString() + ".";
        ipUInt32 = ipUInt32 - (octet * 65536);

        octet = ipUInt32 / 256;
        ipString = ipString + octet.ToString() + ".";
        ipUInt32 = ipUInt32 - (octet * 256);

        ipString = ipString + ipUInt32.ToString();
        return ipString;
    }


}

修改 即使我删除了使用从DhcpEnumSubnetClientsV5返回的DHCP_CLIENT_INFO_ARRAY_V5的所有代码(即我只调用DhcpEnumSubnetClientsV5并且不关心返回的内容),内存泄漏仍然是相同的。

    /////////////////////////////////////////////////////////////////////////////////
    //
    // Get all Leases for "subnet" on "server"
    // Returns DataTable containing Leases or null if there was a problem
    //
    /////////////////////////////////////////////////////////////////////////////////
    public static DataTable GetSubnetLeases(string server, string subnet)
    {
        UInt32 response = 0, resumeHandle = 0;
        UInt32 nRead = 0, nTotal = 0;

        DataTable leases = new DataTable();
        leases.Columns.Add("ipAddress");
        leases.Columns.Add("macAddress");
        leases.Columns.Add("subnet");
        leases.Columns.Add("clientName");

        for (; ; )
        {
            IntPtr retPtr = IntPtr.Zero;

            response = DhcpEnumSubnetClientsV5(server, ConvertIP(subnet), ref resumeHandle, 65536, out retPtr, out nRead, out nTotal);      // ---- Memory Leak ----
//            response = DhcpEnumSubnetClients(server, ConvertIP(subnet), ref resumeHandle, 65536, out retPtr, out nRead, out nTotal);      // No Memory Leak

            if (response == 259)                                                //ERROR_NO_MORE_ITEMS = 259
            {
                if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
                break;
            }

            if (response != 0 && response != 234 || retPtr == IntPtr.Zero)      //ERROR_MORE_DATA = 234
            {
                if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
                return null;
            }

            if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
            if (response == 0) break;
        }

        leases.Rows.Add("1.1.1.1", "11:11:11:11:11:11", subnet, "clientName");
        return leases;
    }

编辑:根据David Heffernan的输入,我还尝试在释放整个结构使用的内存之前释放Clients数组使用的内存,但它没有帮助。

    /////////////////////////////////////////////////////////////////////////////////
    //
    // Get all Leases for "subnet" on "server"
    // Returns DataTable containing Leases or null if there was a problem
    //
    /////////////////////////////////////////////////////////////////////////////////
    public static DataTable GetSubnetLeases(string server, string subnet)
    {
        UInt32 response = 0, resumeHandle = 0;
        UInt32 nRead = 0, nTotal = 0;

        DataTable leases = new DataTable();
        leases.Columns.Add("ipAddress");
        leases.Columns.Add("macAddress");
        leases.Columns.Add("subnet");
        leases.Columns.Add("clientName");

        for (; ; )
        {
            IntPtr retPtr = IntPtr.Zero;

            response = DhcpEnumSubnetClientsV5(server, ConvertIP(subnet), ref resumeHandle, 65536, out retPtr, out nRead, out nTotal);      // ---- Memory Leak ----
//            response = DhcpEnumSubnetClients(server, ConvertIP(subnet), ref resumeHandle, 65536, out retPtr, out nRead, out nTotal);      // No Memory Leak

            if (response == 259)                                                //ERROR_NO_MORE_ITEMS = 259
            {
                if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
                break;
            }

            if (response != 0 && response != 234 || retPtr == IntPtr.Zero)      //ERROR_MORE_DATA = 234
            {
                if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);
                return null;
            }


            IntPtr clientsArrayPtr = Marshal.ReadIntPtr(retPtr, sizeof(UInt32));    // Get Pointer to the Clients Array

            if (clientsArrayPtr != IntPtr.Zero) DhcpRpcFreeMemory(clientsArrayPtr); // Release Memory used by Cients Array
            if (retPtr != IntPtr.Zero) DhcpRpcFreeMemory(retPtr);                   // Release Memory used by the DHCP_CLIENT_INFO_ARRAY_V5 struct
            if (response == 0) break;
        }

        leases.Rows.Add("1.1.1.1", "11:11:11:11:11:11", subnet, "clientName");
        return leases;
    }

0 个答案:

没有答案