merge common implementations and server implementations

2026-05-25 18:16:56 +01:00 · 2017-08-16 02:43:41 -04:00
parent e3531534b8
commit bfcd1b520f
389 changed files with 1212 additions and 1626 deletions
--- a/Emby.Server.Implementations/IO/SharpCifs/Dcerpc/DcerpcPipeHandle.cs
+++ b/Emby.Server.Implementations/IO/SharpCifs/Dcerpc/DcerpcPipeHandle.cs
@@ -0,0 +1,135 @@
+// This code is derived from jcifs smb client library <jcifs at samba dot org>
+// Ported by J. Arturo <webmaster at komodosoft dot net>
+//  
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// 
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+// 
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+using System;
+using System.IO;
+using SharpCifs.Smb;
+using SharpCifs.Util;
+using SharpCifs.Util.Sharpen;
+
+namespace SharpCifs.Dcerpc
+{
+	public class DcerpcPipeHandle : DcerpcHandle
+	{
+		internal SmbNamedPipe Pipe;
+
+		internal SmbFileInputStream In;
+
+		internal SmbFileOutputStream Out;
+
+		internal bool IsStart = true;
+
+		/// <exception cref="UnknownHostException"></exception>
+		/// <exception cref="System.UriFormatException"></exception>
+		/// <exception cref="SharpCifs.Dcerpc.DcerpcException"></exception>
+		public DcerpcPipeHandle(string url, NtlmPasswordAuthentication auth)
+		{
+			Binding = ParseBinding(url);
+			url = "smb://" + Binding.Server + "/IPC$/" + Runtime.Substring(Binding.Endpoint
+				, 6);
+			string @params = string.Empty;
+			string server;
+			string address;
+			server = (string)Binding.GetOption("server");
+			if (server != null)
+			{
+				@params += "&server=" + server;
+			}
+			address = (string)Binding.GetOption("address");
+			if (server != null)
+			{
+				@params += "&address=" + address;
+			}
+			if (@params.Length > 0)
+			{
+				url += "?" + Runtime.Substring(@params, 1);
+			}
+			Pipe = new SmbNamedPipe(url, (unchecked(0x2019F) << 16) | SmbNamedPipe.PipeTypeRdwr
+				 | SmbNamedPipe.PipeTypeDceTransact, auth);
+		}
+
+		/// <exception cref="System.IO.IOException"></exception>
+		protected internal override void DoSendFragment(byte[] buf, int off, int length, 
+			bool isDirect)
+		{
+			if (Out != null && Out.IsOpen() == false)
+			{
+				throw new IOException("DCERPC pipe is no longer open");
+			}
+			if (In == null)
+			{
+				In = (SmbFileInputStream)Pipe.GetNamedPipeInputStream();
+			}
+			if (Out == null)
+			{
+				Out = (SmbFileOutputStream)Pipe.GetNamedPipeOutputStream();
+			}
+			if (isDirect)
+			{
+				Out.WriteDirect(buf, off, length, 1);
+				return;
+			}
+			Out.Write(buf, off, length);
+		}
+
+		/// <exception cref="System.IO.IOException"></exception>
+		protected internal override void DoReceiveFragment(byte[] buf, bool isDirect)
+		{
+			int off;
+			int flags;
+			int length;
+			if (buf.Length < MaxRecv)
+			{
+				throw new ArgumentException("buffer too small");
+			}
+			if (IsStart && !isDirect)
+			{
+				// start of new frag, do trans
+				off = In.Read(buf, 0, 1024);
+			}
+			else
+			{
+				off = In.ReadDirect(buf, 0, buf.Length);
+			}
+			if (buf[0] != 5 && buf[1] != 0)
+			{
+				throw new IOException("Unexpected DCERPC PDU header");
+			}
+			flags = buf[3] & unchecked(0xFF);
+			// next read is start of new frag
+            IsStart = (flags & DcerpcConstants.DcerpcLastFrag) == DcerpcConstants.DcerpcLastFrag;
+			length = Encdec.Dec_uint16le(buf, 8);
+			if (length > MaxRecv)
+			{
+				throw new IOException("Unexpected fragment length: " + length);
+			}
+			while (off < length)
+			{
+				off += In.ReadDirect(buf, off, length - off);
+			}
+		}
+
+		/// <exception cref="System.IO.IOException"></exception>
+		public override void Close()
+		{
+			State = 0;
+			if (Out != null)
+			{
+				Out.Close();
+			}
+		}
+	}
+}