server/Windows/cli/wfreerdp.c

 
#include <freerdp/config.h>
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
 
#include <winpr/tchar.h>
#include <winpr/windows.h>
 
#include "wf_interface.h"
 
#include "wfreerdp.h"
 
#include <freerdp/server/server-common.h>
#include <freerdp/log.h>
#define TAG SERVER_TAG("windows")
 
int IDcount = 0;
 
BOOL CALLBACK moncb(HMONITOR hMonitor, HDC hdcMonitor, LPRECT lprcMonitor, LPARAM dwData)
{
  WLog_DBG(TAG, "%d\t(%ld, %ld), (%ld, %ld)", IDcount, lprcMonitor->left, lprcMonitor->top,
           lprcMonitor->right, lprcMonitor->bottom);
  IDcount++;
  return TRUE;
}
 
int main(int argc, char* argv[])
{
  freerdp_server_warn_unmaintained(argc, argv);
 
  BOOL screen_selected = FALSE;
  int index = 1;
  wfServer* server;
  server = wfreerdp_server_new();
  set_screen_id(0);
  // handle args
  errno = 0;
 
  while (index < argc)
  {
    // first the args that will cause the program to terminate
    if (strcmp("--list-screens", argv[index]) == 0)
    {
      int width;
      int height;
      int bpp;
      WLog_INFO(TAG, "Detecting screens...");
      WLog_INFO(TAG, "ID\tResolution\t\tName (Interface)");
 
      for (int i = 0;; i++)
      {
        _TCHAR name[128] = { 0 };
        if (get_screen_info(i, name, ARRAYSIZE(name), &width, &height, &bpp) != 0)
        {
          if ((width * height * bpp) == 0)
            continue;
 
          WLog_INFO(TAG, "%d\t%dx%dx%d\t", i, width, height, bpp);
          WLog_INFO(TAG, "%s", name);
        }
        else
        {
          break;
        }
      }
 
      {
        int vscreen_w;
        int vscreen_h;
        vscreen_w = GetSystemMetrics(SM_CXVIRTUALSCREEN);
        vscreen_h = GetSystemMetrics(SM_CYVIRTUALSCREEN);
        WLog_INFO(TAG, "");
        EnumDisplayMonitors(NULL, NULL, moncb, 0);
        IDcount = 0;
        WLog_INFO(TAG, "Virtual Screen = %dx%d", vscreen_w, vscreen_h);
      }
 
      return 0;
    }
 
    if (strcmp("--screen", argv[index]) == 0)
    {
      UINT32 val;
      screen_selected = TRUE;
      index++;
 
      if (index == argc)
      {
        WLog_INFO(TAG, "missing screen id parameter");
        return 0;
      }
 
      val = strtoul(argv[index], NULL, 0);
 
      if ((errno != 0) || (val > UINT32_MAX))
        return -1;
 
      set_screen_id(val);
      index++;
    }
 
    if (index == argc - 1)
    {
      UINT32 val = strtoul(argv[index], NULL, 0);
 
      if ((errno != 0) || (val > UINT32_MAX))
        return -1;
 
      server->port = val;
      break;
    }
  }
 
  if (screen_selected == FALSE)
  {
    int width;
    int height;
    int bpp;
    WLog_INFO(TAG, "screen id not provided. attempting to detect...");
    WLog_INFO(TAG, "Detecting screens...");
    WLog_INFO(TAG, "ID\tResolution\t\tName (Interface)");
 
    for (int i = 0;; i++)
    {
      _TCHAR name[128] = { 0 };
      if (get_screen_info(i, name, ARRAYSIZE(name), &width, &height, &bpp) != 0)
      {
        if ((width * height * bpp) == 0)
          continue;
 
        WLog_INFO(TAG, "%d\t%dx%dx%d\t", i, width, height, bpp);
        WLog_INFO(TAG, "%s", name);
        set_screen_id(i);
        break;
      }
      else
      {
        break;
      }
    }
  }
 
  WLog_INFO(TAG, "Starting server");
  wfreerdp_server_start(server);
  (void)WaitForSingleObject(server->thread, INFINITE);
  WLog_INFO(TAG, "Stopping server");
  wfreerdp_server_stop(server);
  wfreerdp_server_free(server);
  return 0;
}