From 0201f61d19eb3bee5a97af6e20f139c36b57b7d1 Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@gmail.com>
Date: Sat, 14 Mar 2015 00:34:44 +0100
Subject: [PATCH] Fixed missing callbacks for named pipes.

---
 winpr/libwinpr/pipe/pipe.c | 67 ++++++++++++++++++++++++++++++++++----
 winpr/libwinpr/pipe/pipe.h |  6 ++--
 2 files changed, 63 insertions(+), 10 deletions(-)

diff --git a/winpr/libwinpr/pipe/pipe.c b/winpr/libwinpr/pipe/pipe.c
index 6b95cf8ef..dd7c0ba7c 100644
--- a/winpr/libwinpr/pipe/pipe.c
+++ b/winpr/libwinpr/pipe/pipe.c
@@ -111,6 +111,62 @@ BOOL PipeCloseHandle(HANDLE handle) {
 	return TRUE;
 }
 
+static BOOL NamedPipeCloseHandle(HANDLE handle);
+
+static BOOL NamedPipeIsHandled(HANDLE handle)
+{
+	WINPR_NAMED_PIPE* pPipe = (WINPR_NAMED_PIPE*) handle;
+
+	if (!pPipe || pPipe->Type != HANDLE_TYPE_NAMED_PIPE)
+	{
+		SetLastError(ERROR_INVALID_HANDLE);
+		return FALSE;
+	}
+
+	return TRUE;
+}
+
+static int NamedPipeGetFd(HANDLE handle)
+{
+	WINPR_NAMED_PIPE *pipe = (WINPR_NAMED_PIPE *)handle;
+
+	if (!NamedPipeIsHandled(handle))
+		return -1;
+
+	if (pipe->ServerMode)
+		return pipe->serverfd;
+	return pipe->clientfd;
+}
+
+BOOL NamedPipeCloseHandle(HANDLE handle) {
+	WINPR_NAMED_PIPE* pNamedPipe = (WINPR_NAMED_PIPE *)handle;
+
+	if (!NamedPipeIsHandled(handle))
+		return FALSE;
+
+	if (pNamedPipe->pfnUnrefNamedPipe)
+		pNamedPipe->pfnUnrefNamedPipe(pNamedPipe);
+
+	if (pNamedPipe->name)
+		free(pNamedPipe->name);
+
+	if (pNamedPipe->lpFileName)
+		free((void*)pNamedPipe->lpFileName);
+
+	if (pNamedPipe->lpFilePath)
+		free((void*)pNamedPipe->lpFilePath);
+
+	if (pNamedPipe->serverfd != -1)
+		close(pNamedPipe->serverfd);
+	
+	if (pNamedPipe->clientfd != -1)
+		close(pNamedPipe->clientfd);
+	
+	free(handle);
+
+	return TRUE;
+}
+
 static void InitWinPRPipeModule()
 {
 	if (g_NamedPipeServerSockets)
@@ -252,6 +308,9 @@ HANDLE CreateNamedPipeA(LPCSTR lpName, DWORD dwOpenMode, DWORD dwPipeMode, DWORD
 	pNamedPipe->dwFlagsAndAttributes = dwOpenMode;
 	pNamedPipe->clientfd = -1;
 	pNamedPipe->ServerMode = TRUE;
+	pNamedPipe->cb.GetFd = NamedPipeGetFd;
+	pNamedPipe->cb.CloseHandle = NamedPipeCloseHandle;
+	pNamedPipe->cb.IsHandled = NamedPipeIsHandled;
 	ArrayList_Lock(g_NamedPipeServerSockets);
 
 	for (index = 0; index < ArrayList_Count(g_NamedPipeServerSockets); index++)
@@ -347,13 +406,7 @@ out:
 
 	if (hNamedPipe == INVALID_HANDLE_VALUE)
 	{
-		if (pNamedPipe)
-		{
-			free((void*)pNamedPipe->name);
-			free((void*)pNamedPipe->lpFileName);
-			free((void*)pNamedPipe->lpFilePath);
-			free(pNamedPipe);
-		}
+		NamedPipeCloseHandle(hNamedPipe);
 
 		if (serverfd != -1)
 			close(serverfd);
diff --git a/winpr/libwinpr/pipe/pipe.h b/winpr/libwinpr/pipe/pipe.h
index 4cf3f2650..3084a8b48 100644
--- a/winpr/libwinpr/pipe/pipe.h
+++ b/winpr/libwinpr/pipe/pipe.h
@@ -47,9 +47,9 @@ struct winpr_named_pipe
 	int clientfd;
 	int serverfd;
 
-	const char* name;
-	const char* lpFileName;
-	const char* lpFilePath;
+	char* name;
+	char* lpFileName;
+	char* lpFilePath;
 
 	BOOL ServerMode;
 	DWORD dwOpenMode;