zydronium/SMBLibrary
1
0
Fork 0
mirror of https://github.com/TalAloni/SMBLibrary.git synced 2025-08-12 10:19:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add missing usings

Browse source
This commit is contained in:
Peter Van Dender 2021-04-01 08:45:19 +02:00
parent 0b98e1f754
commit 513566a45f
3 changed files with 38 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

23
SMBLibrary/Client/NameServiceClient.cs
View file

@ -8,6 +8,7 @@ using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Net; using System.Net;
using System.Net.Sockets; using System.Net.Sockets;
using System.Threading;
using SMBLibrary.NetBios; using SMBLibrary.NetBios;
namespace SMBLibrary.Client namespace SMBLibrary.Client
@ -43,14 +44,20 @@ namespace SMBLibrary.Client
private NodeStatusResponse SendNodeStatusRequest(NodeStatusRequest request) private NodeStatusResponse SendNodeStatusRequest(NodeStatusRequest request)
{ {
UdpClient client = new UdpClient(); using (UdpClient client = new UdpClient())
IPEndPoint serverEndPoint = new IPEndPoint(m_serverAddress, NetBiosNameServicePort); {
client.Connect(serverEndPoint); IPEndPoint serverEndPoint = new IPEndPoint(m_serverAddress, NetBiosNameServicePort);
client.Connect(serverEndPoint);
byte[] requestBytes = request.GetBytes(); byte[] requestBytes = request.GetBytes();
client.Send(requestBytes, requestBytes.Length); client.Send(requestBytes, requestBytes.Length);
byte[] responseBytes = client.Receive(ref serverEndPoint); Thread.Sleep(100);
return new NodeStatusResponse(responseBytes, 0); if (client.Available > 0)
{
var responseBytes = client.Receive(ref serverEndPoint);
return new NodeStatusResponse(responseBytes, 0);
}
}
return new NodeStatusResponse();
} }
} }
} }

10
SMBLibrary/NetBios/NameServicePackets/NodeStatusRequest.cs
View file

@ -36,10 +36,12 @@ namespace SMBLibrary.NetBios
public byte[] GetBytes() public byte[] GetBytes()
{ {
MemoryStream stream = new MemoryStream(); using (MemoryStream stream = new MemoryStream())
Header.WriteBytes(stream); {
Question.WriteBytes(stream); Header.WriteBytes(stream);
return stream.ToArray(); Question.WriteBytes(stream);
return stream.ToArray();
}
} }
} }
} }

30
SMBLibrary/NetBios/NameServicePackets/NodeStatusResponse.cs
View file

@ -54,25 +54,29 @@ namespace SMBLibrary.NetBios
{ {
Resource.Data = GetData(); Resource.Data = GetData();
MemoryStream stream = new MemoryStream(); using (MemoryStream stream = new MemoryStream())
Header.WriteBytes(stream); {
Resource.WriteBytes(stream); Header.WriteBytes(stream);
return stream.ToArray(); Resource.WriteBytes(stream);
return stream.ToArray();
}
} }
private byte[] GetData() private byte[] GetData()
{ {
MemoryStream stream = new MemoryStream(); using (MemoryStream stream = new MemoryStream())
stream.WriteByte((byte)Names.Count);
foreach (KeyValuePair<string, NameFlags> entry in Names)
{ {
ByteWriter.WriteAnsiString(stream, entry.Key); stream.WriteByte((byte)Names.Count);
BigEndianWriter.WriteUInt16(stream, (ushort)entry.Value); foreach (KeyValuePair<string, NameFlags> entry in Names)
{
ByteWriter.WriteAnsiString(stream, entry.Key);
BigEndianWriter.WriteUInt16(stream, (ushort)entry.Value);
}
ByteWriter.WriteBytes(stream, Statistics.GetBytes());
return stream.ToArray();
} }
ByteWriter.WriteBytes(stream, Statistics.GetBytes());
return stream.ToArray();
} }
} }
} }