summaryrefslogtreecommitdiffstats
path: root/xrdpapi/xrdpapi.c
diff options
context:
space:
mode:
Diffstat (limited to 'xrdpapi/xrdpapi.c')
-rw-r--r--xrdpapi/xrdpapi.c383
1 files changed, 383 insertions, 0 deletions
diff --git a/xrdpapi/xrdpapi.c b/xrdpapi/xrdpapi.c
new file mode 100644
index 00000000..694b3800
--- /dev/null
+++ b/xrdpapi/xrdpapi.c
@@ -0,0 +1,383 @@
+/**
+ * xrdp: A Remote Desktop Protocol server.
+ *
+ * Copyright (C) Thomas Goddard 2012
+ * Copyright (C) Jay Sorg 2012
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/* do not use os_calls in here */
+
+#define LOG_LEVEL 1
+#define LLOG(_level, _args) \
+ do { if (_level < LOG_LEVEL) { ErrorF _args ; } } while (0)
+#define LLOGLN(_level, _args) \
+ do { if (_level < LOG_LEVEL) { printf _args ; printf("\n"); } } while (0)
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+
+#include "xrdpapi.h"
+
+struct wts_obj
+{
+ int fd;
+ int status;
+ char name[8];
+ char dname[128];
+ int display_num;
+ int flags;
+};
+
+/*****************************************************************************/
+static int
+get_display_num_from_display(char* display_text)
+{
+ int index;
+ int mode;
+ int host_index;
+ int disp_index;
+ int scre_index;
+ char host[256];
+ char disp[256];
+ char scre[256];
+
+ index = 0;
+ host_index = 0;
+ disp_index = 0;
+ scre_index = 0;
+ mode = 0;
+ while (display_text[index] != 0)
+ {
+ if (display_text[index] == ':')
+ {
+ mode = 1;
+ }
+ else if (display_text[index] == '.')
+ {
+ mode = 2;
+ }
+ else if (mode == 0)
+ {
+ host[host_index] = display_text[index];
+ host_index++;
+ }
+ else if (mode == 1)
+ {
+ disp[disp_index] = display_text[index];
+ disp_index++;
+ }
+ else if (mode == 2)
+ {
+ scre[scre_index] = display_text[index];
+ scre_index++;
+ }
+ index++;
+ }
+ host[host_index] = 0;
+ disp[disp_index] = 0;
+ scre[scre_index] = 0;
+ return atoi(disp);
+}
+
+/*****************************************************************************/
+void*
+WTSVirtualChannelOpen(void* hServer, unsigned int SessionId,
+ const char* pVirtualName)
+{
+ if (hServer != WTS_CURRENT_SERVER_HANDLE)
+ {
+ return 0;
+ }
+ return WTSVirtualChannelOpenEx(SessionId, pVirtualName, 0);
+}
+
+/*****************************************************************************/
+static int
+can_send(int sck, int millis)
+{
+ struct timeval time;
+ fd_set wfds;
+ int select_rv;
+
+ FD_ZERO(&wfds);
+ FD_SET(sck, &wfds);
+ time.tv_sec = millis / 1000;
+ time.tv_usec = (millis * 1000) % 1000000;
+ select_rv = select(sck + 1, 0, &wfds, 0, &time);
+ if (select_rv > 0)
+ {
+ return 1;
+ }
+ return 0;
+}
+
+/*****************************************************************************/
+static int
+can_recv(int sck, int millis)
+{
+ struct timeval time;
+ fd_set rfds;
+ int select_rv;
+
+ FD_ZERO(&rfds);
+ FD_SET(sck, &rfds);
+ time.tv_sec = millis / 1000;
+ time.tv_usec = (millis * 1000) % 1000000;
+ select_rv = select(sck + 1, &rfds, 0, 0, &time);
+ if (select_rv > 0)
+ {
+ return 1;
+ }
+ return 0;
+}
+
+/*****************************************************************************/
+static int
+send_init(struct wts_obj* wts)
+{
+ char initmsg[64];
+
+ memset(initmsg, 0, 64);
+ strncpy(initmsg, wts->name, 8);
+ initmsg[16] = (wts->flags >> 0) & 0xff;
+ initmsg[17] = (wts->flags >> 8) & 0xff;
+ initmsg[18] = (wts->flags >> 16) & 0xff;
+ initmsg[19] = (wts->flags >> 24) & 0xff;
+ LLOGLN(10, ("send_init: sending %s", initmsg));
+ if (!can_send(wts->fd, 500))
+ {
+ return 1;
+ }
+ if (send(wts->fd, initmsg, 64, 0) != 64)
+ {
+ return 1;
+ }
+ LLOGLN(10, ("send_init: send ok!"));
+ return 0;
+}
+
+/*****************************************************************************/
+void*
+WTSVirtualChannelOpenEx(unsigned int SessionId,
+ const char* pVirtualName,
+ unsigned int flags)
+{
+ struct wts_obj* wts;
+ char* display_text;
+ struct sockaddr_un s;
+ int bytes;
+ unsigned long llong;
+
+ if (SessionId != WTS_CURRENT_SESSION)
+ {
+ LLOGLN(0, ("WTSVirtualChannelOpenEx: SessionId bad"));
+ return 0;
+ }
+ wts = (struct wts_obj*)malloc(sizeof(struct wts_obj));
+ memset(wts, 0, sizeof(struct wts_obj));
+ wts->fd = -1;
+ wts->flags = flags;
+ display_text = getenv("DISPLAY");
+ if (display_text != 0)
+ {
+ wts->display_num = get_display_num_from_display(display_text);
+ }
+ if (wts->display_num > 0)
+ {
+ wts->fd = socket(AF_UNIX, SOCK_STREAM, 0);
+ /* set non blocking */
+ llong = fcntl(wts->fd, F_GETFL);
+ llong = llong | O_NONBLOCK;
+ fcntl(wts->fd, F_SETFL, llong);
+ /* connect to session chansrv */
+ memset(&s, 0, sizeof(struct sockaddr_un));
+ s.sun_family = AF_UNIX;
+ bytes = sizeof(s.sun_path);
+ snprintf(s.sun_path, bytes - 1, "/tmp/.xrdp/xrdpapi_%d", wts->display_num);
+ s.sun_path[bytes - 1] = 0;
+ bytes = sizeof(struct sockaddr_un);
+ if (connect(wts->fd, (struct sockaddr*)&s, bytes) == 0)
+ {
+ LLOGLN(10, ("WTSVirtualChannelOpenEx: connected ok, name %s", pVirtualName));
+ strncpy(wts->name, pVirtualName, 8);
+ /* wait for connection to complete and send init */
+ if (send_init(wts) == 0)
+ {
+ /* all ok */
+ wts->status = 1;
+ }
+ }
+ }
+ else
+ {
+ LLOGLN(0, ("WTSVirtualChannelOpenEx: display is 0"));
+ }
+ return wts;
+}
+
+/*****************************************************************************/
+int
+WTSVirtualChannelWrite(void* hChannelHandle, const char* Buffer,
+ unsigned int Length, unsigned int* pBytesWritten)
+{
+ struct wts_obj* wts;
+ int error;
+ int lerrno;
+
+ wts = (struct wts_obj*)hChannelHandle;
+ if (wts == 0)
+ {
+ return 0;
+ }
+ if (wts->status != 1)
+ {
+ return 0;
+ }
+ if (can_send(wts->fd, 0))
+ {
+ error = send(wts->fd, Buffer, Length, 0);
+ if (error == -1)
+ {
+ lerrno = errno;
+ if ((lerrno == EWOULDBLOCK) || (lerrno == EAGAIN) ||
+ (lerrno == EINPROGRESS))
+ {
+ *pBytesWritten = 0;
+ return 1;
+ }
+ return 0;
+ }
+ else if (error == 0)
+ {
+ return 0;
+ }
+ else if (error > 0)
+ {
+ *pBytesWritten = error;
+ return 1;
+ }
+ }
+ *pBytesWritten = 0;
+ return 1;
+}
+
+/*****************************************************************************/
+int
+WTSVirtualChannelRead(void* hChannelHandle, unsigned int TimeOut,
+ char* Buffer, unsigned int BufferSize,
+ unsigned int* pBytesRead)
+{
+ struct wts_obj* wts;
+ int error;
+ int lerrno;
+
+ wts = (struct wts_obj*)hChannelHandle;
+ if (wts == 0)
+ {
+ return 0;
+ }
+ if (wts->status != 1)
+ {
+ return 0;
+ }
+ if (can_recv(wts->fd, TimeOut))
+ {
+ error = recv(wts->fd, Buffer, BufferSize, 0);
+ if (error == -1)
+ {
+ lerrno = errno;
+ if ((lerrno == EWOULDBLOCK) || (lerrno == EAGAIN) ||
+ (lerrno == EINPROGRESS))
+ {
+ *pBytesRead = 0;
+ return 1;
+ }
+ return 0;
+ }
+ else if (error == 0)
+ {
+ return 0;
+ }
+ else if (error > 0)
+ {
+ *pBytesRead = error;
+ return 1;
+ }
+ }
+ *pBytesRead = 0;
+ return 1;
+}
+
+/*****************************************************************************/
+int
+WTSVirtualChannelClose(void* hChannelHandle)
+{
+ struct wts_obj* wts;
+
+ wts = (struct wts_obj*)hChannelHandle;
+ if (wts == 0)
+ {
+ return 0;
+ }
+ if (wts->fd != -1)
+ {
+ close(wts->fd);
+ }
+ free(wts);
+ return 1;
+}
+
+/*****************************************************************************/
+int
+WTSVirtualChannelQuery(void* hChannelHandle, WTS_VIRTUAL_CLASS WtsVirtualClass,
+ void** ppBuffer, unsigned int* pBytesReturned)
+{
+ struct wts_obj* wts;
+
+ wts = (struct wts_obj*)hChannelHandle;
+ if (wts == 0)
+ {
+ return 0;
+ }
+ if (wts->status != 1)
+ {
+ return 0;
+ }
+ if (WtsVirtualClass == WTSVirtualFileHandle)
+ {
+ *pBytesReturned = 4;
+ *ppBuffer = malloc(4);
+ memcpy(*ppBuffer, &(wts->fd), 4);
+ }
+ return 1;
+}
+
+/*****************************************************************************/
+void
+WTSFreeMemory(void* pMemory)
+{
+ if (pMemory != 0)
+ {
+ free(pMemory);
+ }
+}
generated by cgit v1.2.1 (git 2.18.0) at 2025-01-18 13:48:13 +0000