C# newbie :이 코드를 어떻게 수정하여 DNS 조회를 수행합니까?
-
10-07-2019 - |
문제
C#을 사용한 것은 이번이 처음이므로 내 요소가 많이 벗어났습니다. 그러나 나는 Stack Overflow의 멋진 사람들이 나를 도울 수 있다는 믿음을 가지고 있습니다! 인터넷에서 떠 다니는 다른 코드를 기반으로 다음 코드 (아래)를 생각해 냈습니다. 내가하려는 것은 주어진 도메인의 모든 "텍스트"DNS 레코드 (TXT)를 찾는 것입니다. Visual C# 2008 Express의 콘솔 애플리케이션으로 Google.com의 레코드를 찾아 간단하게 시작했습니다. 나중에 명령 줄 매개 변수를 기반으로 도메인을 사용자 정의하는 것에 대해 걱정합니다. 지금은이 작업을 수행하고 싶습니다. 모든 DNS 조회 도구는 Google.com 에이 DNS 텍스트 레코드가 DNS에 있다고 말할 수 있습니다.
v=spf1 include:_netblocks.google.com ip4:216.73.93.70/31 ip4:216.73.93.72/31 ~all
그러나 불행히도 그것은 내가 얻는 것이 아닙니다. 내 코드는 다음과 같습니다.
namespace DnsUtils
{
using System;
using System.Collections;
using System.ComponentModel;
using System.Runtime.InteropServices;
using System.Text;
public class DnsTxt
{
[DllImport("dnsapi", EntryPoint = "DnsQuery_W", CharSet = CharSet.Unicode, SetLastError = true, ExactSpelling = true)]
private static extern int DnsQuery([MarshalAs(UnmanagedType.VBByRefStr)]ref string pszName, QueryTypes wType, QueryOptions options, int aipServers, ref IntPtr ppQueryResults, int pReserved);
[DllImport("dnsapi", CharSet = CharSet.Auto, SetLastError = true)]
private static extern void DnsRecordListFree(IntPtr pRecordList, int FreeType);
public static string[] GetTXTRecords(string domain)
{
IntPtr ptr1 = IntPtr.Zero;
IntPtr ptr2 = IntPtr.Zero;
TXTRecord recTxt;
if (Environment.OSVersion.Platform != PlatformID.Win32NT)
{
throw new NotSupportedException();
}
ArrayList list1 = new ArrayList();
UnicodeEncoding encoding = new UnicodeEncoding();
int num1 = DnsTxt.DnsQuery(ref domain, QueryTypes.DNS_TYPE_TEXT, QueryOptions.DNS_QUERY_BYPASS_CACHE, 0, ref ptr1, 0);
if (num1 != 0)
{
throw new Win32Exception(num1);
}
for (ptr2 = ptr1; !ptr2.Equals(IntPtr.Zero); ptr2 = recTxt.pNext)
{
recTxt = (TXTRecord)Marshal.PtrToStructure(ptr2, typeof(TXTRecord));
if (recTxt.wType == 16)
{
IntPtr pointerToAddressStringArray = Marshal.AllocHGlobal(IntPtr.Size);
IntPtr addressStringArray = Marshal.ReadIntPtr(pointerToAddressStringArray);
for (int i = 0; i < recTxt.dwStringCount; i++)
{
IntPtr addressCharArray = Marshal.ReadIntPtr(recTxt.pStringArray, i * 4);
int offset = 0;
ArrayList bytesList = new ArrayList();
byte newByte = Marshal.ReadByte(addressCharArray, offset++);
while (newByte != 0)
{
bytesList.Add(newByte);
newByte = Marshal.ReadByte(addressCharArray, offset++);
}
byte[] bytesArray = new byte[offset];
bytesList.CopyTo(bytesArray);
string textValue = encoding.GetString(bytesArray);
list1.Add(textValue);
}
Marshal.FreeHGlobal(pointerToAddressStringArray);
}
}
DnsTxt.DnsRecordListFree(ptr2, 0);
return (string[])list1.ToArray(typeof(string));
}
private enum QueryOptions
{
DNS_QUERY_ACCEPT_TRUNCATED_RESPONSE = 1,
DNS_QUERY_BYPASS_CACHE = 8,
DNS_QUERY_DONT_RESET_TTL_VALUES = 0x100000,
DNS_QUERY_NO_HOSTS_FILE = 0x40,
DNS_QUERY_NO_LOCAL_NAME = 0x20,
DNS_QUERY_NO_NETBT = 0x80,
DNS_QUERY_NO_RECURSION = 4,
DNS_QUERY_NO_WIRE_QUERY = 0x10,
DNS_QUERY_RESERVED = -16777216,
DNS_QUERY_RETURN_MESSAGE = 0x200,
DNS_QUERY_STANDARD = 0,
DNS_QUERY_TREAT_AS_FQDN = 0x1000,
DNS_QUERY_USE_TCP_ONLY = 2,
DNS_QUERY_WIRE_ONLY = 0x100
}
private enum QueryTypes
{
DNS_TYPE_TEXT = 16
}
[StructLayout(LayoutKind.Sequential)]
private struct TXTRecord
{
public IntPtr pNext;
public string pName;
public short wType;
public short wDataLength;
public int flags;
public int dwTtl;
public int dwReserved;
public int dwStringCount;
public IntPtr pStringArray;
}
static void Main(string[] args)
{
try
{
string[] s = DnsUtils.DnsTxt.GetTXTRecords("google.com");
foreach (string st in s)
{
Console.WriteLine("Value: {0}", st);
}
}
catch (Win32Exception e)
{
Console.WriteLine(e.Message);
}
Console.ReadLine();
}
}
}
이 코드를 실행하면 인쇄합니다 Value: ?
. 이제 이것이 실제로 DNS 쿼리를 수행하고 예상대로 결과를 얻었음을 의미하지만 intptr을 문자열 배열로 변환하려고하는 부분에서 무언가를 망쳤습니다 (더 어렵다고 생각하기 때문에 더 어려워요. 전자를 디버깅하려면). 어쨌든 제안이 있습니까? 내가 어디로 갔는지 볼 수 있습니까? 미리 감사드립니다.
해결책
이 줄은 나에게 약간 꺼져 있습니다.
IntPtr addressCharArray = Marshal.ReadIntPtr(recTxt.pStringArray, i * 4);
TXT 레코드의 첫 번째 문자열 항목의 첫 4 바이트를 INTPTR로 읽은 것 같습니다. 나는 다음과 같은 생각을한다.
string s = Marshal.PtrToStringAuto(recTxt.pStringArray);
첫 번째 항목을 얻을 수 있습니다. 그 후, 나는 다음과 같은 생각을한다.
IntPtr p = new IntPtr(recTxt.pStringArray.ToInt32() + sizeof(uint) * i);
string s = Marshal.PtrToStringAuto(p);
나머지는 얻을 것입니다.
다른 팁
더 간단한 대안으로 새로운 nslookup 프로세스를 만들고 필요한 것을 잡기 위해 StandardOUput를 구문 분석하여 모든 COM Interop을 다루지 않을 수 있습니다.
using System;
using System.Diagnostics;
namespace ConsoleApplication1
{
class Program
{
static void Main(string[] args)
{
var startInfo = new ProcessStartInfo("nslookup");
startInfo.Arguments = "-type=TXT google.com";
startInfo.RedirectStandardOutput = true;
startInfo.UseShellExecute = false;
startInfo.WindowStyle = ProcessWindowStyle.Hidden;
using (var cmd = Process.Start(startInfo))
{
// This is where you grab the output from nslookup.
Console.WriteLine(cmd.StandardOutput.ReadToEnd());
}
Console.Read();
}
}
}