387 lines
11 KiB
C#
387 lines
11 KiB
C#
|
//
|
||
|
|
// PipeWin32.cs
|
||
|
|
//
|
||
|
|
// Author:
|
||
|
|
// Atsushi Enomoto <atsushi@ximian.com>
|
||
|
|
//
|
||
|
|
// Copyright (C) 2009 Novell, Inc. http://www.novell.com
|
||
|
|
//
|
||
|
|
// Permission is hereby granted, free of charge, to any person obtaining
|
||
|
|
// a copy of this software and associated documentation files (the
|
||
|
|
// "Software"), to deal in the Software without restriction, including
|
||
|
|
// without limitation the rights to use, copy, modify, merge, publish,
|
||
|
|
// distribute, sublicense, and/or sell copies of the Software, and to
|
||
|
|
// permit persons to whom the Software is furnished to do so, subject to
|
||
|
|
// the following conditions:
|
||
|
|
//
|
||
|
|
// The above copyright notice and this permission notice shall be
|
||
|
|
// included in all copies or substantial portions of the Software.
|
||
|
|
//
|
||
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||
|
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||
|
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
||
|
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
||
|
|
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
||
|
|
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
||
|
|
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
|
|
//
|
||
|
|
|
||
|
|
#if !BOOTSTRAP_BASIC
|
||
|
|
|
||
|
|
using System;
|
||
|
|
using System.ComponentModel;
|
||
|
|
using System.IO;
|
||
|
|
using System.Linq;
|
||
|
|
using System.Runtime.InteropServices;
|
||
|
|
using System.Security.AccessControl;
|
||
|
|
using System.Security.Permissions;
|
||
|
|
using System.Security.Principal;
|
||
|
|
using System.Text;
|
||
|
|
using Microsoft.Win32;
|
||
|
|
using Microsoft.Win32.SafeHandles;
|
||
|
|
|
||
|
|
namespace System.IO.Pipes
|
||
|
|
{
|
||
|
|
static class Win32PipeError
|
||
|
|
{
|
||
|
|
public static Exception GetException ()
|
||
|
|
{
|
||
|
|
return GetException (Marshal.GetLastWin32Error ());
|
||
|
|
}
|
||
|
|
|
||
|
|
public static Exception GetException (int errorCode)
|
||
|
|
{
|
||
|
|
switch (errorCode) {
|
||
|
|
case 5: return new UnauthorizedAccessException ();
|
||
|
|
default: return new Win32Exception (errorCode);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
abstract class Win32AnonymousPipe : IPipe
|
||
|
|
{
|
||
|
|
protected Win32AnonymousPipe ()
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
public abstract SafePipeHandle Handle { get; }
|
||
|
|
|
||
|
|
public void WaitForPipeDrain ()
|
||
|
|
{
|
||
|
|
throw new NotImplementedException ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class Win32AnonymousPipeClient : Win32AnonymousPipe, IAnonymousPipeClient
|
||
|
|
{
|
||
|
|
// AnonymousPipeClientStream owner;
|
||
|
|
|
||
|
|
public Win32AnonymousPipeClient (AnonymousPipeClientStream owner, SafePipeHandle handle)
|
||
|
|
{
|
||
|
|
// this.owner = owner;
|
||
|
|
|
||
|
|
this.handle = handle;
|
||
|
|
}
|
||
|
|
|
||
|
|
SafePipeHandle handle;
|
||
|
|
|
||
|
|
public override SafePipeHandle Handle {
|
||
|
|
get { return handle; }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class Win32AnonymousPipeServer : Win32AnonymousPipe, IAnonymousPipeServer
|
||
|
|
{
|
||
|
|
// AnonymousPipeServerStream owner;
|
||
|
|
|
||
|
|
public unsafe Win32AnonymousPipeServer (AnonymousPipeServerStream owner, PipeDirection direction,
|
||
|
|
HandleInheritability inheritability, int bufferSize,
|
||
|
|
PipeSecurity pipeSecurity)
|
||
|
|
{
|
||
|
|
IntPtr r, w;
|
||
|
|
|
||
|
|
byte[] securityDescriptor = null;
|
||
|
|
if (pipeSecurity != null)
|
||
|
|
securityDescriptor = pipeSecurity.GetSecurityDescriptorBinaryForm ();
|
||
|
|
|
||
|
|
fixed (byte* securityDescriptorPtr = securityDescriptor) {
|
||
|
|
SecurityAttributes att = new SecurityAttributes (inheritability, (IntPtr)securityDescriptorPtr);
|
||
|
|
if (!Win32Marshal.CreatePipe (out r, out w, ref att, bufferSize))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
}
|
||
|
|
|
||
|
|
var rh = new SafePipeHandle (r, true);
|
||
|
|
var wh = new SafePipeHandle (w, true);
|
||
|
|
|
||
|
|
if (direction == PipeDirection.Out) {
|
||
|
|
server_handle = wh;
|
||
|
|
client_handle = rh;
|
||
|
|
} else {
|
||
|
|
server_handle = rh;
|
||
|
|
client_handle = wh;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public Win32AnonymousPipeServer (AnonymousPipeServerStream owner, SafePipeHandle serverHandle, SafePipeHandle clientHandle)
|
||
|
|
{
|
||
|
|
// this.owner = owner;
|
||
|
|
this.server_handle = serverHandle;
|
||
|
|
this.client_handle = clientHandle;
|
||
|
|
}
|
||
|
|
|
||
|
|
SafePipeHandle server_handle, client_handle;
|
||
|
|
|
||
|
|
public override SafePipeHandle Handle {
|
||
|
|
get { return server_handle; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public SafePipeHandle ClientHandle {
|
||
|
|
get { return client_handle; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public void DisposeLocalCopyOfClientHandle ()
|
||
|
|
{
|
||
|
|
throw new NotImplementedException ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
abstract class Win32NamedPipe : IPipe
|
||
|
|
{
|
||
|
|
string name_cache;
|
||
|
|
|
||
|
|
public string Name {
|
||
|
|
get {
|
||
|
|
if (name_cache != null)
|
||
|
|
return name_cache;
|
||
|
|
|
||
|
|
int s, c, m, t;
|
||
|
|
byte [] un = new byte [200];
|
||
|
|
while (true) {
|
||
|
|
if (!Win32Marshal.GetNamedPipeHandleState (Handle, out s, out c, out m, out t, un, un.Length))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
|
||
|
|
if (un [un.Length - 1] == 0)
|
||
|
|
break;
|
||
|
|
un = new byte [un.Length * 10];
|
||
|
|
}
|
||
|
|
name_cache = Encoding.Default.GetString (un);
|
||
|
|
return name_cache;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public abstract SafePipeHandle Handle { get; }
|
||
|
|
|
||
|
|
public void WaitForPipeDrain ()
|
||
|
|
{
|
||
|
|
throw new NotImplementedException ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class Win32NamedPipeClient : Win32NamedPipe, INamedPipeClient
|
||
|
|
{
|
||
|
|
NamedPipeClientStream owner;
|
||
|
|
|
||
|
|
// .ctor with existing handle
|
||
|
|
public Win32NamedPipeClient (NamedPipeClientStream owner, SafePipeHandle safePipeHandle)
|
||
|
|
{
|
||
|
|
this.handle = safePipeHandle;
|
||
|
|
this.owner = owner;
|
||
|
|
|
||
|
|
// FIXME: retrieve is_async from state?
|
||
|
|
}
|
||
|
|
|
||
|
|
// .ctor without handle - create new
|
||
|
|
public Win32NamedPipeClient (NamedPipeClientStream owner, string serverName, string pipeName,
|
||
|
|
PipeAccessRights desiredAccessRights, PipeOptions options,
|
||
|
|
HandleInheritability inheritability)
|
||
|
|
{
|
||
|
|
name = String.Format ("\\\\{0}\\pipe\\{1}", serverName, pipeName);
|
||
|
|
var att = new SecurityAttributes (inheritability, IntPtr.Zero);
|
||
|
|
is_async = (options & PipeOptions.Asynchronous) != PipeOptions.None;
|
||
|
|
|
||
|
|
opener = delegate {
|
||
|
|
var ret = Win32Marshal.CreateFile (name, desiredAccessRights, 0, ref att, 3, 0, IntPtr.Zero);
|
||
|
|
if (ret == new IntPtr (-1L))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
|
||
|
|
return new SafePipeHandle (ret, true);
|
||
|
|
};
|
||
|
|
this.owner = owner;
|
||
|
|
}
|
||
|
|
|
||
|
|
Func<SafePipeHandle> opener;
|
||
|
|
bool is_async;
|
||
|
|
string name;
|
||
|
|
SafePipeHandle handle;
|
||
|
|
|
||
|
|
public override SafePipeHandle Handle {
|
||
|
|
get { return handle; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public bool IsAsync {
|
||
|
|
get { return is_async; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Connect ()
|
||
|
|
{
|
||
|
|
if (owner.IsConnected)
|
||
|
|
throw new InvalidOperationException ("The named pipe is already connected");
|
||
|
|
|
||
|
|
handle = opener ();
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Connect (int timeout)
|
||
|
|
{
|
||
|
|
if (owner.IsConnected)
|
||
|
|
throw new InvalidOperationException ("The named pipe is already connected");
|
||
|
|
|
||
|
|
if (!Win32Marshal.WaitNamedPipe (name, timeout))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
Connect ();
|
||
|
|
}
|
||
|
|
|
||
|
|
public int NumberOfServerInstances {
|
||
|
|
get {
|
||
|
|
int s, c, m, t;
|
||
|
|
byte [] un = null;
|
||
|
|
if (!Win32Marshal.GetNamedPipeHandleState (Handle, out s, out c, out m, out t, un, 0))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
return c;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
class Win32NamedPipeServer : Win32NamedPipe, INamedPipeServer
|
||
|
|
{
|
||
|
|
//NamedPipeServerStream owner;
|
||
|
|
|
||
|
|
// .ctor with existing handle
|
||
|
|
public Win32NamedPipeServer (NamedPipeServerStream owner, SafePipeHandle safePipeHandle)
|
||
|
|
{
|
||
|
|
handle = safePipeHandle;
|
||
|
|
//this.owner = owner;
|
||
|
|
}
|
||
|
|
|
||
|
|
// .ctor without handle - create new
|
||
|
|
public unsafe Win32NamedPipeServer (NamedPipeServerStream owner, string pipeName, int maxNumberOfServerInstances,
|
||
|
|
PipeTransmissionMode transmissionMode, PipeAccessRights rights,
|
||
|
|
PipeOptions options, int inBufferSize, int outBufferSize,
|
||
|
|
PipeSecurity pipeSecurity, HandleInheritability inheritability)
|
||
|
|
{
|
||
|
|
string name = String.Format ("\\\\.\\pipe\\{0}", pipeName);
|
||
|
|
|
||
|
|
uint openMode;
|
||
|
|
openMode = (uint)rights | (uint)options; // Enum values match Win32 flags exactly.
|
||
|
|
|
||
|
|
int pipeMode = 0;
|
||
|
|
if ((owner.TransmissionMode & PipeTransmissionMode.Message) != 0)
|
||
|
|
pipeMode |= 4;
|
||
|
|
//if ((readTransmissionMode & PipeTransmissionMode.Message) != 0)
|
||
|
|
// pipeMode |= 2;
|
||
|
|
if ((options & PipeOptions.Asynchronous) != 0)
|
||
|
|
pipeMode |= 1;
|
||
|
|
|
||
|
|
byte[] securityDescriptor = null;
|
||
|
|
if (pipeSecurity != null)
|
||
|
|
securityDescriptor = pipeSecurity.GetSecurityDescriptorBinaryForm ();
|
||
|
|
|
||
|
|
fixed (byte* securityDescriptorPtr = securityDescriptor) {
|
||
|
|
// FIXME: is nDefaultTimeout = 0 ok?
|
||
|
|
var att = new SecurityAttributes (inheritability, (IntPtr)securityDescriptorPtr);
|
||
|
|
var ret = Win32Marshal.CreateNamedPipe (name, openMode, pipeMode, maxNumberOfServerInstances,
|
||
|
|
outBufferSize, inBufferSize, 0, ref att, IntPtr.Zero);
|
||
|
|
if (ret == new IntPtr (-1L))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
handle = new SafePipeHandle (ret, true);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
SafePipeHandle handle;
|
||
|
|
|
||
|
|
public override SafePipeHandle Handle {
|
||
|
|
get { return handle; }
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Disconnect ()
|
||
|
|
{
|
||
|
|
Win32Marshal.DisconnectNamedPipe (Handle);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void WaitForConnection ()
|
||
|
|
{
|
||
|
|
if (!Win32Marshal.ConnectNamedPipe (Handle, IntPtr.Zero))
|
||
|
|
throw Win32PipeError.GetException ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
[StructLayout (LayoutKind.Sequential)]
|
||
|
|
struct SecurityAttributes
|
||
|
|
{
|
||
|
|
public readonly int Length;
|
||
|
|
public readonly IntPtr SecurityDescriptor;
|
||
|
|
public readonly bool Inheritable;
|
||
|
|
|
||
|
|
public SecurityAttributes (HandleInheritability inheritability, IntPtr securityDescriptor)
|
||
|
|
{
|
||
|
|
Length = Marshal.SizeOf (typeof (SecurityAttributes));
|
||
|
|
SecurityDescriptor = securityDescriptor;
|
||
|
|
Inheritable = inheritability == HandleInheritability.Inheritable;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static class Win32Marshal
|
||
|
|
{
|
||
|
|
internal static bool IsWindows {
|
||
|
|
get {
|
||
|
|
switch (Environment.OSVersion.Platform) {
|
||
|
|
case PlatformID.Win32S:
|
||
|
|
case PlatformID.Win32Windows:
|
||
|
|
case PlatformID.Win32NT:
|
||
|
|
case PlatformID.WinCE:
|
||
|
|
return true;
|
||
|
|
default:
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa365152%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern bool CreatePipe (out IntPtr readHandle, out IntPtr writeHandle,
|
||
|
|
ref SecurityAttributes pipeAtts, int size);
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa365150%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern IntPtr CreateNamedPipe (string name, uint openMode, int pipeMode, int maxInstances,
|
||
|
|
int outBufferSize, int inBufferSize, int defaultTimeout,
|
||
|
|
ref SecurityAttributes securityAttributes, IntPtr atts);
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa365146%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern bool ConnectNamedPipe (SafePipeHandle handle, IntPtr overlapped);
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa365166%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern bool DisconnectNamedPipe (SafePipeHandle handle);
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa365443%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern bool GetNamedPipeHandleState (SafePipeHandle handle,
|
||
|
|
out int state, out int curInstances,
|
||
|
|
out int maxCollectionCount, out int collectDateTimeout,
|
||
|
|
byte [] userName, int maxUserNameSize);
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa365800%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern bool WaitNamedPipe (string name, int timeout);
|
||
|
|
|
||
|
|
// http://msdn.microsoft.com/en-us/library/aa363858%28VS.85%29.aspx
|
||
|
|
[DllImport ("kernel32", SetLastError=true)]
|
||
|
|
internal static extern IntPtr CreateFile (string name, PipeAccessRights desiredAccess, FileShare fileShare,
|
||
|
|
ref SecurityAttributes atts, int creationDisposition, int flags, IntPtr templateHandle);
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|