added disconnect tool

2 files changed, 70 insertions, 1 deletions

diff --git a/sesman/tools/Makefile.am b/sesman/tools/Makefile.am
index 8596e77e..a48b0f11 100644
--- a/sesman/tools/Makefile.am
+++ b/sesman/tools/Makefile.am
@@ -14,7 +14,8 @@ INCLUDES = \
 bin_PROGRAMS = \
   xrdp-sesrun \
   xrdp-sestest \
-  xrdp-sesadmin
+  xrdp-sesadmin \
+  xrdp-dis
 
 xrdp_sesrun_SOURCES = \
   sesrun.c \
@@ -27,6 +28,9 @@ xrdp_sestest_SOURCES = \
 xrdp_sesadmin_SOURCES = \
   sesadmin.c
 
+xrdp_dis_SOURCES = \
+  dis.c
+
 xrdp_sesrun_LDADD = \
   $(top_srcdir)/common/libcommon.la
 
diff --git a/sesman/tools/dis.c b/sesman/tools/dis.c
new file mode 100644
index 00000000..4a1ef78c
--- /dev/null
+++ b/sesman/tools/dis.c
@@ -0,0 +1,65 @@
+/*
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+
+   This program 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 General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+   xrdp: A Remote Desktop Protocol server.
+   Copyright (C) Jay Sorg 2011
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+
+int main(int argc, char** argv)
+{
+  int sck;
+  int dis;
+  struct sockaddr_un sa;
+  size_t len;
+  char* p;
+  char* display;
+
+  if (argc != 1)
+  {
+    printf("xrdp disconnect utility\n");
+    printf("run with no parameters to disconnect you xrdp session\n");
+    return 0;
+  }
+
+  display = getenv("DISPLAY");
+  if (display == 0)
+  {
+    printf("display not set\n");
+    return 1;
+  }
+  dis = strtol(display + 1, &p, 10);
+  memset(&sa, 0, sizeof(sa));
+  sa.sun_family = AF_UNIX;
+  sprintf(sa.sun_path, "/tmp/xrdp_disconnect_display_%d", dis);
+  if (access(sa.sun_path, F_OK) != 0)
+  {
+    printf("not in an xrdp session\n");
+    return 1;
+  }
+  sck = socket(PF_UNIX, SOCK_DGRAM, 0);
+  len = sizeof(sa);
+  if (sendto(sck, "sig", 4, 0, (struct sockaddr*)&sa, len) > 0)
+  {
+    printf("message sent ok\n");
+  }
+  return 0;
+}