gfwleak/zhangyang-libzt
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add NetworkStream to C# wrapper. Fix errno bug in C# wrapper

Browse Source
This commit is contained in:
Joseph Henry
2021-05-17 21:28:15 -07:00
parent a37ee00c3a
commit eac36f781d
7 changed files with 487 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/bindings/csharp/CSharpSockets.cxx
View File

@@ -398,6 +398,11 @@ SWIGINTERN void SWIG_CSharpException(int code, const char* msg)
extern "C" {
#endif
SWIGEXPORT int SWIGSTDCALL CSharp_zts_errno_get()
{
return zts_errno;
}
#ifndef ZTS_DISABLE_CENTRAL_API
SWIGEXPORT int SWIGSTDCALL CSharp_zts_central_set_access_mode(char jarg1)
@@ -1574,6 +1579,11 @@ SWIGEXPORT int SWIGSTDCALL CSharp_zts_bsd_shutdown(int jarg1, int jarg2)
return jresult;
}
SWIGEXPORT size_t SWIGSTDCALL CSharp_zts_get_data_available(int fd)
{
return zts_get_data_available(fd);
}
SWIGEXPORT int SWIGSTDCALL CSharp_zts_set_no_delay(int jarg1, int jarg2)
{
int jresult;

372
src/bindings/csharp/NetworkStream.cs Normal file
View File

@@ -0,0 +1,372 @@
/*
* Copyright (c)2013-2021 ZeroTier, Inc.
*
* Use of this software is governed by the Business Source License included
* in the LICENSE.TXT file in the project's root directory.
*
* Change Date: 2026-01-01
*
* On the date above, in accordance with the Business Source License, use
* of this software will be governed by version 2.0 of the Apache License.
*/
/****/
using System;
using System.Threading;
using System.IO;
using System.Runtime.InteropServices;
using System.Net.Sockets;
using ZeroTier;
namespace ZeroTier.Sockets
{
public class NetworkStream : Stream {
private ZeroTier.Sockets.Socket _streamSocket;
private bool _isReadable;
private bool _isWriteable;
private bool _ownsSocket;
private volatile bool _isDisposed = false;
internal NetworkStream()
{
_ownsSocket = true;
}
public NetworkStream(ZeroTier.Sockets.Socket socket)
{
if (socket == null) {
throw new ArgumentNullException("socket");
}
InitNetworkStream(socket, FileAccess.ReadWrite);
}
public NetworkStream(ZeroTier.Sockets.Socket socket, bool ownsSocket)
{
if (socket == null) {
throw new ArgumentNullException("socket");
}
InitNetworkStream(socket, FileAccess.ReadWrite);
_ownsSocket = ownsSocket;
}
public NetworkStream(ZeroTier.Sockets.Socket socket, FileAccess accessMode)
{
if (socket == null) {
throw new ArgumentNullException("socket");
}
InitNetworkStream(socket, accessMode);
}
public NetworkStream(ZeroTier.Sockets.Socket socket, FileAccess accessMode, bool ownsSocket)
{
if (socket == null) {
throw new ArgumentNullException("socket");
}
InitNetworkStream(socket, accessMode);
_ownsSocket = ownsSocket;
}
internal NetworkStream(NetworkStream networkStream, bool ownsSocket)
{
ZeroTier.Sockets.Socket socket = networkStream.Socket;
if (socket == null) {
throw new ArgumentNullException("networkStream");
}
InitNetworkStream(socket, FileAccess.ReadWrite);
_ownsSocket = ownsSocket;
}
protected ZeroTier.Sockets.Socket Socket
{
get {
return _streamSocket;
}
}
internal void ConvertToNotSocketOwner()
{
_ownsSocket = false;
GC.SuppressFinalize(this);
}
public override int ReadTimeout
{
get {
return _streamSocket.ReceiveTimeout;
}
set {
if (value <= 0) {
throw new ArgumentOutOfRangeException("Timeout value must be greater than zero");
}
_streamSocket.ReceiveTimeout = value;
}
}
public override int WriteTimeout
{
get {
return _streamSocket.SendTimeout;
}
set {
if (value <= 0) {
throw new ArgumentOutOfRangeException("Timeout value must be greater than zero");
}
_streamSocket.SendTimeout = value;
}
}
protected bool Readable
{
get {
return _isReadable;
}
set {
_isReadable = value;
}
}
protected bool Writeable
{
get {
return _isWriteable;
}
set {
_isWriteable = value;
}
}
public override bool CanRead
{
get {
return _isReadable;
}
}
public override bool CanSeek
{
get {
return false;
}
}
public override bool CanWrite
{
get {
return _isWriteable;
}
}
public override bool CanTimeout
{
get {
return true;
}
}
public virtual bool DataAvailable
{
get {
if (_streamSocket == null) {
throw new IOException("ZeroTier socket is null");
}
if (_isDisposed) {
throw new ObjectDisposedException("ZeroTier.Sockets.Socket");
}
return _streamSocket.Available != 0;
}
}
internal void InitNetworkStream(ZeroTier.Sockets.Socket socket, FileAccess accessMode)
{
if (! socket.Connected) {
throw new IOException("ZeroTier socket must be connected");
}
if (! socket.Blocking) {
throw new IOException("ZeroTier socket must be in blocking mode");
}
if (socket.SocketType != SocketType.Stream) {
throw new IOException("ZeroTier socket must by stream type");
}
_streamSocket = socket;
switch (accessMode) {
case FileAccess.Write:
_isWriteable = true;
break;
case FileAccess.Read:
_isReadable = true;
break;
case FileAccess.ReadWrite:
default:
_isReadable = true;
_isWriteable = true;
break;
}
}
public override int Read([In, Out] byte[] buffer, int offset, int size)
{
bool canRead = CanRead;
if (_isDisposed) {
throw new ObjectDisposedException("ZeroTier.Sockets.Socket");
}
if (! canRead) {
throw new InvalidOperationException("Cannot read from ZeroTier socket");
}
if (buffer == null) {
throw new ArgumentNullException("buffer");
}
if (offset < 0 || offset > buffer.Length) {
throw new ArgumentOutOfRangeException("offset");
}
if (size < 0 || size > buffer.Length - offset) {
throw new ArgumentOutOfRangeException("size");
}
if (_streamSocket == null) {
throw new IOException("ZeroTier socket is null");
}
try {
int bytesTransferred = _streamSocket.Receive(buffer, offset, size, 0);
return bytesTransferred;
}
catch (Exception exception) {
throw new IOException("Cannot read from ZeroTier socket", exception);
}
}
public override void Write(byte[] buffer, int offset, int size)
{
bool canWrite = CanWrite;
if (_isDisposed) {
throw new ObjectDisposedException("ZeroTier.Sockets.Socket");
}
if (! canWrite) {
throw new InvalidOperationException("Cannot write to ZeroTier socket");
}
if (buffer == null) {
throw new ArgumentNullException("buffer");
}
if (offset < 0 || offset > buffer.Length) {
throw new ArgumentOutOfRangeException("offset");
}
if (size < 0 || size > buffer.Length - offset) {
throw new ArgumentOutOfRangeException("size");
}
if (_streamSocket == null) {
throw new IOException("ZeroTier socket is null");
}
try {
_streamSocket.Send(buffer, offset, size, SocketFlags.None);
}
catch (Exception exception) {
throw new IOException("Cannot write to ZeroTier socket", exception);
}
}
internal bool Poll(int microSeconds, SelectMode mode)
{
if (_streamSocket == null) {
throw new IOException("ZeroTier socket is null");
}
if (_isDisposed) {
throw new ObjectDisposedException("ZeroTier.Sockets.Socket");
}
return _streamSocket.Poll(microSeconds, mode);
}
internal bool PollRead()
{
if (_streamSocket == null) {
return false;
}
if (_isDisposed) {
return false;
}
return _streamSocket.Poll(0, SelectMode.SelectRead);
}
public override void Flush()
{
// Not applicable
}
public override void SetLength(long value)
{
throw new NotSupportedException("Not supported");
}
public override long Length
{
get {
throw new NotSupportedException("Not supported");
}
}
public override long Position
{
get {
throw new NotSupportedException("Not supported");
}
set {
throw new NotSupportedException("Not supported");
}
}
public override long Seek(long offset, SeekOrigin origin)
{
throw new NotSupportedException("Not supported");
}
public void Close(int timeout)
{
if (timeout < 0) {
throw new ArgumentOutOfRangeException("Timeout value must be greater than zero");
}
_streamSocket.Close(timeout);
}
internal bool Connected
{
get {
if (! _isDisposed && _streamSocket != null && _streamSocket.Connected) {
return true;
}
else {
return false;
}
}
}
protected override void Dispose(bool disposing)
{
bool cleanedUp = _isDisposed;
_isDisposed = true;
if (! cleanedUp && disposing) {
if (_streamSocket != null) {
_isWriteable = false;
_isReadable = false;
if (_ownsSocket) {
if (_streamSocket != null) {
_streamSocket.Shutdown(SocketShutdown.Both);
_streamSocket.Close();
}
}
}
}
base.Dispose(disposing);
}
~NetworkStream()
{
Dispose(false);
}
}
}

66
src/bindings/csharp/Socket.cs
View File

@@ -230,6 +230,12 @@ namespace ZeroTier.Sockets
public void Close()
{
Close(0);
}
public void Close(int timeout)
{
// TODO: Timeout needs to be implemented
if (_isClosed) {
throw new ObjectDisposedException("Socket has already been closed");
}
@@ -291,21 +297,10 @@ namespace ZeroTier.Sockets
public Int32 Send(Byte[] buffer)
{
if (_isClosed) {
throw new ObjectDisposedException("Socket has been closed");
}
if (_fd < 0) {
throw new ZeroTier.Sockets.SocketException((int)ZeroTier.Constants.ERR_SOCKET);
}
if (buffer == null) {
throw new ArgumentNullException("buffer");
}
int flags = 0;
IntPtr bufferPtr = Marshal.UnsafeAddrOfPinnedArrayElement(buffer, 0);
return zts_bsd_send(_fd, bufferPtr, (uint)Buffer.ByteLength(buffer), (int)flags);
return Send(buffer, 0, buffer != null ? buffer.Length : 0, SocketFlags.None);
}
public Int32 Receive(Byte[] buffer)
public Int32 Send(Byte[] buffer, int offset, int size, SocketFlags socketFlags)
{
if (_isClosed) {
throw new ObjectDisposedException("Socket has been closed");
@@ -316,9 +311,49 @@ namespace ZeroTier.Sockets
if (buffer == null) {
throw new ArgumentNullException("buffer");
}
if (size < 0 || size > buffer.Length - offset) {
throw new ArgumentOutOfRangeException("size");
}
if (offset < 0 || offset > buffer.Length) {
throw new ArgumentOutOfRangeException("offset");
}
int flags = 0;
IntPtr bufferPtr = Marshal.UnsafeAddrOfPinnedArrayElement(buffer, 0);
return zts_bsd_recv(_fd, bufferPtr, (uint)Buffer.ByteLength(buffer), (int)flags);
return zts_bsd_send(_fd, bufferPtr + offset, (uint)Buffer.ByteLength(buffer), (int)flags);
}
public int Available
{
get {
return zts_get_data_available(_fd);
}
}
public Int32 Receive(Byte[] buffer)
{
return Receive(buffer, 0, buffer != null ? buffer.Length : 0, SocketFlags.None);
}
public Int32 Receive(byte[] buffer, int offset, int size, SocketFlags socketFlags)
{
if (_isClosed) {
throw new ObjectDisposedException("Socket has been closed");
}
if (_fd < 0) {
throw new ZeroTier.Sockets.SocketException((int)ZeroTier.Constants.ERR_SOCKET);
}
if (buffer == null) {
throw new ArgumentNullException("buffer");
}
if (size < 0 || size > buffer.Length - offset) {
throw new ArgumentOutOfRangeException("size");
}
if (offset < 0 || offset > buffer.Length) {
throw new ArgumentOutOfRangeException("offset");
}
int flags = 0;
IntPtr bufferPtr = Marshal.UnsafeAddrOfPinnedArrayElement(buffer, 0);
return zts_bsd_recv(_fd, bufferPtr + offset, (uint)Buffer.ByteLength(buffer), (int)flags);
}
public int ReceiveTimeout
@@ -579,6 +614,9 @@ namespace ZeroTier.Sockets
[DllImport("libzt", EntryPoint = "CSharp_zts_bsd_shutdown")]
static extern int zts_bsd_shutdown(int arg1, int arg2);
[DllImport("libzt", EntryPoint = "CSharp_zts_get_data_available")]
static extern int zts_get_data_available(int fd);
[DllImport("libzt", EntryPoint = "CSharp_zts_set_no_delay")]
static extern int zts_set_no_delay(int fd, int enabled);