/* gpgcempr.c - Manager for GPG CE devices
   Copyright (C) 2010 Free Software Foundation, Inc.

   This file is part of Assuan.

   Assuan 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 3 of
   the License, or (at your option) any later version.

   Assuan 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 program; if not, see <http://www.gnu.org/licenses/>.
 */

#define _WIN32_WCE 0x0500

#include <stdio.h>
#include <windows.h>

#define PGM "gpgcemgr"

#define GPGCEDEV_KEY_NAME  L"Drivers\\GnuPG_Device"
#define GPGCEDEV_KEY_NAME2 L"Drivers\\GnuPG_Log"
#define GPGCEDEV_DLL_NAME  L"gpgcedev.dll"
#define GPGCEDEV_PREFIX    L"GPG"


static char *
wchar_to_utf8 (const wchar_t *string)
{
  int n;
  size_t length = wcslen (string);
  char *result;

  n = WideCharToMultiByte (CP_UTF8, 0, string, length, NULL, 0, NULL, NULL);
  if (n < 0 || (n+1) <= 0)
    abort ();

  result = malloc (n+1);
  if (!result)
    abort ();
  n = WideCharToMultiByte (CP_ACP, 0, string, length, result, n, NULL, NULL);
  if (n < 0)
    abort ();
  
  result[n] = 0;
  return result;
}


static wchar_t *
utf8_to_wchar (const char *string)
{
  int n;
  size_t nbytes;
  wchar_t *result;

  if (!string)
    abort ();

  n = MultiByteToWideChar (CP_UTF8, 0, string, -1, NULL, 0);
  if (n < 0)
    abort ();
  nbytes = (size_t)(n+1) * sizeof(*result);
  if (nbytes / sizeof(*result) != (n+1)) 
    abort ();
  result = malloc (nbytes);
  if (!result)
    abort ();

  n = MultiByteToWideChar (CP_UTF8, 0, string, -1, result, n);
  if (n < 0)
    abort ();
  return result;
}


static int
install (void)
{
  HKEY handle;
  DWORD disp, dw;
  int rc;
  
  if ((rc=RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPGCEDEV_KEY_NAME, 0, NULL, 0,
                          KEY_WRITE, NULL, &handle, &disp)))
    {
      fprintf (stderr, PGM": error creating registry key 1: rc=%d\n", rc);
      return 1;
    }

  RegSetValueEx (handle, L"dll", 0, REG_SZ, 
                 (void*)GPGCEDEV_DLL_NAME, sizeof (GPGCEDEV_DLL_NAME));
  RegSetValueEx (handle, L"prefix", 0, REG_SZ,
                 (void*)GPGCEDEV_PREFIX, sizeof (GPGCEDEV_PREFIX));

  dw = 1;
  RegSetValueEx (handle, L"Index", 0, REG_DWORD, (void*)&dw, sizeof dw);
  
  RegCloseKey (handle);

  fprintf (stderr, PGM": registry key 1 created\n");

  if ((rc=RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPGCEDEV_KEY_NAME2, 0, NULL, 0,
                          KEY_WRITE, NULL, &handle, &disp)))
    {
      fprintf (stderr, PGM": error creating registry key 2: rc=%d\n", rc);
      return 1;
    }

  RegSetValueEx (handle, L"dll", 0, REG_SZ, 
                 (void*)GPGCEDEV_DLL_NAME, sizeof (GPGCEDEV_DLL_NAME));
  RegSetValueEx (handle, L"prefix", 0, REG_SZ,
                 (void*)GPGCEDEV_PREFIX, sizeof (GPGCEDEV_PREFIX));

  dw = 2;
  RegSetValueEx (handle, L"Index", 0, REG_DWORD, (void*)&dw, sizeof dw);
  
  RegCloseKey (handle);

  fprintf (stderr, PGM": registry key 2 created\n");


  return 0;
}


static int
deinstall (wchar_t *name)
{
  int result = 0;
  HANDLE shd;
  DEVMGR_DEVICE_INFORMATION dinfo;

  memset (&dinfo, 0, sizeof dinfo);
  dinfo.dwSize = sizeof dinfo;
  shd = FindFirstDevice (DeviceSearchByLegacyName, name, &dinfo);
  if (shd == INVALID_HANDLE_VALUE)
    {
      if (GetLastError () == 18)
        fprintf (stderr, PGM": device not found\n");
      else
        {
          fprintf (stderr, PGM": FindFirstDevice failed: rc=%d\n", 
                   (int)GetLastError ());
          result = 1;
        }
    }
  else
    {
      fprintf (stderr, PGM": ActivateDevice handle is %p\n", dinfo.hDevice);
      if (dinfo.hDevice && dinfo.hDevice != INVALID_HANDLE_VALUE)
        {
          if (!DeactivateDevice (dinfo.hDevice))
            {
              fprintf (stderr, PGM": DeactivateDevice failed: rc=%d\n",
                       (int)GetLastError ());
              result = 1;
            }
          else
            fprintf (stderr, PGM": DeactivateDevice succeeded\n");
        }
      FindClose (shd);
    }

  return result;
}


static int
enable_debug (int yes)
{
  HKEY handle;
  DWORD disp;
  int rc;
  
  if ((rc=RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPGCEDEV_KEY_NAME, 0, NULL, 0,
                          KEY_WRITE, NULL, &handle, &disp)))
    {
      fprintf (stderr, PGM": error creating debug registry key: rc=%d\n", rc);
      return 1;
    }
  
  RegSetValueEx (handle, L"debugDriver", 0, REG_SZ, 
                 (void*)(yes? L"1":L"0"), sizeof L"0");
  RegCloseKey (handle);
  return 0;
}


static int
enable_log (int yes)
{
  HKEY handle;
  DWORD disp;
  int rc;
  
  if ((rc=RegCreateKeyEx (HKEY_LOCAL_MACHINE, GPGCEDEV_KEY_NAME2, 0, NULL, 0,
                          KEY_WRITE, NULL, &handle, &disp)))
    {
      fprintf (stderr, PGM": error creating debug registry key: rc=%d\n", rc);
      return 1;
    }
  
  RegSetValueEx (handle, L"enableLog", 0, REG_SZ, 
                 (void*)(yes? L"1":L"0"), sizeof L"0");
  RegCloseKey (handle);
  return 0;
}



/* Kudos to Scott Seligman <scott@scottandmichelle.net> for his work
   on the reverse engineering.  */
struct htc_sensor_s
{
  SHORT   tilt_x;         // From -1000 to 1000 (about), 0 is flat
  SHORT   tilt_y;         // From -1000 to 1000 (about), 0 is flat
  SHORT   tilt_z;         // From -1000 to 1000 (about)
  DWORD   angle_x;        // 0 .. 359
  DWORD   angle_y;        // From 0 to 359
  DWORD   orientation;    // 0.. 5?
  DWORD   unknown;        // Handle?
};
typedef struct htc_sensor_s *htc_sensor_t;

static HANDLE (WINAPI *HTCSensorOpen) (DWORD);
static void   (WINAPI *HTCSensorClose) (HANDLE);
static DWORD  (WINAPI *HTCSensorGetDataOutput) (HANDLE, htc_sensor_t);

static int
load_sensor_api (void)
{
  static HMODULE dll_hd;

  if (dll_hd)
    return 0;

  dll_hd = LoadLibrary (L"HTCSensorSDK.dll");
  if (!dll_hd)
    {
      fprintf (stderr, PGM": error loading sensor DLL: rc=%d\n",
               (int)GetLastError ());
      return 1;
    }

  HTCSensorOpen = (void*)GetProcAddress (dll_hd, L"HTCSensorOpen");
  if (HTCSensorOpen)
    HTCSensorClose = (void*)GetProcAddress (dll_hd, L"HTCSensorClose");
  if (HTCSensorClose)
    HTCSensorGetDataOutput = (void*)
      GetProcAddress (dll_hd, L"HTCSensorGetDataOutput");
  if (!HTCSensorGetDataOutput)
    {
      fprintf (stderr, PGM": error loading function from sensor DLL: rc=%d\n",
               (int)GetLastError ());
      CloseHandle (dll_hd);
      return 1;
    }
  return 0;
}


static int 
gravity (void)
{
  int rc;
  HANDLE sensor;
  struct htc_sensor_s lastdata;
  struct htc_sensor_s data;

  rc = load_sensor_api ();
  if (rc)
    return rc;

  sensor = HTCSensorOpen (1 /* tilt sensor */);
  if (!sensor  || sensor == INVALID_HANDLE_VALUE)
    {
      fprintf (stderr, PGM": error opening gravity sensor: rc=%d\n",
               (int)GetLastError ());
      HTCSensorClose (sensor);
      return 1;
    }

  memset (&lastdata, 0, sizeof lastdata);
  while (HTCSensorGetDataOutput (sensor, &data))
    {
      if (lastdata.tilt_x/10 != data.tilt_x/10
          || lastdata.tilt_y/10 != data.tilt_y/10
          || lastdata.tilt_z/10 != data.tilt_z/10
          || lastdata.angle_x/5 != data.angle_x/5
          || lastdata.angle_y/5 != data.angle_y/5
          || lastdata.orientation != data.orientation)
        {
          lastdata = data;
          printf ("tilt: x=%-5d y=%-5d z=%-5d  "
                  "angle: x=%-3d y=%-3d  "
                  "ori: %d\n",
                  (int)data.tilt_x, (int)data.tilt_y, (int)data.tilt_z,
                  (int)data.angle_x, (int)data.angle_y,
                  (int)data.orientation);
        }
      Sleep (200);
    }
  fprintf (stderr, PGM": reading sensor data failed: rc=%d\n",
           (int)GetLastError ());
  HTCSensorClose (sensor);
  return 0;
}



/* No GPD1 device on the HTC Touch Pro 2.  */
# if 0
static int
gps_raw (void)
{
  HANDLE hd;
  char buffer[1000];
  unsigned long nread;
  int count;

  hd = CreateFile (L"GPD1:", GENERIC_READ, FILE_SHARE_READ,
                   NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  if (hd == INVALID_HANDLE_VALUE)
    {
      fprintf (stderr, PGM": can't open `GPD1': rc=%d\n",
               (int)GetLastError ());
      return 1;
    }
  fprintf (stderr, PGM": GPS device successfully opened\n");

  for (count=0; count < 100; count++)
    {
      if (!ReadFile (hd, buffer, sizeof buffer-1, &nread, NULL))
        {
          fprintf (stderr, PGM": error reading `GPD1': rc=%d\n",
                   (int)GetLastError ());
          CloseHandle (hd);
          return 1;
        }
      buffer[nread-1] = 0;
      fputs (buffer, stdout);
    }

  CloseHandle (hd);
  return 0;
}
#endif

/* Untested samples for CE6.  */
#if 0
static int
gps (void)
{
  HANDLE hd;
  GPS_POSITION pos;

  hd = GPSOpenDevice (NULL, NULL, NULL, 0);
  if (hd == INVALID_HANDLE_VALUE)
    {
      fprintf (stderr, PGM": GPSOpenDevice failed: rc=%d\n",
               (int)GetLastError ());
      return 1;
    }
  fprintf (stderr, PGM": GPS device successfully opened\n");

  if (GPSGetPosition (hd, &pos, 2000, 0))
    {
      fprintf (stderr, PGM": GPSGetPosition failed: rc=%d\n",
               (int)GetLastError ());
      GPSCloseDevice (hd);
      return 1;
    }


  GPSCloseDevice (hd);
  return 0;
}
#endif


static void
set_show_registry (const wchar_t *key, const wchar_t *name, const char *value)
{
  HKEY handle;
  DWORD disp, nbytes, n1, type;
  int rc;
  
  if ((rc=RegCreateKeyEx (HKEY_LOCAL_MACHINE, key, 0, NULL, 0,
                          KEY_WRITE, NULL, &handle, &disp)))
    {
      fprintf (stderr, PGM": error creating registry key: rc=%d\n", rc);
      return;
    }

  if (value && !stricmp (value, "none"))
    {
      if ((rc=RegDeleteValue (handle, name)))
        fprintf (stderr, PGM": error deleting registry value: rc=%d\n", rc);
    }
  else if (value)
    {
      wchar_t *tmp = utf8_to_wchar (value);
      if ((rc=RegSetValueEx (handle, name, 0, REG_SZ, 
                             (void*)tmp, wcslen (tmp)*sizeof(wchar_t))))
        fprintf (stderr, PGM": error setting registry value: rc=%d\n", rc);
      free (tmp);
    }
  else
    {
      nbytes = 2;
      if ((rc=RegQueryValueEx (handle, name, 0, NULL, NULL, &nbytes)))
        {
          if (rc == ERROR_FILE_NOT_FOUND)
            fprintf (stderr, PGM": registry value not set\n"); 
          else
            fprintf (stderr, PGM": error reading registry value: rc=%d\n", rc); 
        }
      else
        {
          char *result = malloc ((n1=nbytes+2));
          if (!result)
            fprintf (stderr, PGM": malloc failed: rc=%d\n", 
                     (int)GetLastError ());
          else if ((rc=RegQueryValueEx (handle, name, 0, &type,
                                         (void*)result, &n1)))
            {
              fprintf (stderr, PGM": error reading registry value (2): "
                       "rc=%d\n", rc);
              free (result);
            }
          else
            {
              result[nbytes] = 0;   /* Make sure it is a string.  */
              result[nbytes+1] = 0; 
              if (type == REG_SZ)
                {
                  wchar_t *tmp = (void*)result;
                  result = wchar_to_utf8 (tmp);
                  free (tmp);
                  printf ("%s\n", result);
                }
              else
                fprintf (stderr, PGM": registry value is not a string\n");
              free (result);
            }
        }
    }
  
  RegCloseKey (handle);
}



int
main (int argc, char **argv)
{
  int result = 0;

  if (argc > 1 && !strcmp (argv[1], "--register"))
    result = install ();
  else if (argc > 1 && !strcmp (argv[1], "--deactivate"))
    {
      if (deinstall (L"GPG1:"))
        result = 1;
      if (deinstall (L"GPG2:"))
        result = 1;
    }
  else if (argc > 1 && !strcmp (argv[1], "--activate"))
    {
      HANDLE hd;

      /* This is mainly for testing.  The activation is usually done
         right before the device is opened.  */
      if (ActivateDevice (GPGCEDEV_KEY_NAME, 0) == INVALID_HANDLE_VALUE)
        {
          fprintf (stderr, PGM": ActivateDevice 1 failed: rc=%d\n",
                   (int)GetLastError ());
          result = 1;
        }
      else if (ActivateDevice (GPGCEDEV_KEY_NAME2, 0) == INVALID_HANDLE_VALUE)
        {
          fprintf (stderr, PGM": ActivateDevice 2 failed: rc=%d\n",
                   (int)GetLastError ());
          result = 1;
        }
      else
        {
          fprintf (stderr, PGM": devices activated\n");
          hd = CreateFile (L"GPG1:", GENERIC_WRITE,
                           FILE_SHARE_READ | FILE_SHARE_WRITE,
                           NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
          if (hd == INVALID_HANDLE_VALUE)
            {
              fprintf (stderr, PGM": opening `GPG1:' failed: rc=%d\n",
                       (int)GetLastError ());
              result = 1;
            }
          else
            {
              fprintf (stderr, PGM": device `GPG1:' seems to work\n");
              CloseHandle (hd);
            }

          hd = CreateFile (L"GPG2:", GENERIC_WRITE,
                           FILE_SHARE_READ | FILE_SHARE_WRITE,
                           NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
          if (hd == INVALID_HANDLE_VALUE)
            {
              fprintf (stderr, PGM": opening `GPG2:' failed: rc=%d\n",
                       (int)GetLastError ());
              result = 1;
            }
          else
            {
              fprintf (stderr, PGM": device `GPG2:' seems to work\n");
              CloseHandle (hd);
            }
          
        }
    }
  else if (argc > 1 && !strcmp (argv[1], "--gravity"))
    result = gravity ();
  /* else if (argc > 1 && !strcmp (argv[1], "--gps")) */
  /*   result = gps (); */
  else if (argc > 1 && !strcmp (argv[1], "--log"))
    {
      HANDLE hd;

      hd = CreateFile (L"GPG2:", GENERIC_WRITE,
                       FILE_SHARE_READ | FILE_SHARE_WRITE,
                       NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
      if (hd == INVALID_HANDLE_VALUE)
        {
          fprintf (stderr, PGM": opening `GPG2:' failed: rc=%d\n",
                   (int)GetLastError ());
          result = 1;
        }
      else
        {
          char marktwain[] = "I have never let my schooling interfere"
            " with my education.\n";
          DWORD nwritten;
          int i;

          for (i=0; i < 3; i++) 
            {
              if (!WriteFile (hd, marktwain, strlen (marktwain),
                              &nwritten, NULL))
                {
                  fprintf (stderr, PGM": writing `GPG2:' failed: rc=%d\n",
                           (int)GetLastError ());
                  result = 1;
                }
              Sleep (200);
            }
          CloseHandle (hd);
        }
    }
  else if (argc > 1 && !strcmp (argv[1], "--enable-debug"))
    result = enable_debug (1);
  else if (argc > 1 && !strcmp (argv[1], "--disable-debug"))
    result = enable_debug (0);
  else if (argc > 1 && !strcmp (argv[1], "--enable-log"))
    result = enable_log (1);
  else if (argc > 1 && !strcmp (argv[1], "--disable-log"))
    result = enable_log (0);
  else if (argc > 1 && !strcmp (argv[1], "--gpgme-log"))
    set_show_registry (L"Software\\GNU\\gpgme", L"debug", 
                                argc > 2? argv[2] : NULL); 
  else if (argc > 1 && !strcmp (argv[1], "--gnupg-log"))
    set_show_registry (L"Software\\GNU\\GnuPG", L"DefaultLogFile", 
                                argc > 2? argv[2] : NULL);
 else
    {
      fprintf (stderr,
               "usage: " PGM " COMMAND\n"
               "Commands are:\n"
               "  --register        Register the GPGCEDEV device\n"
               "  --deactivate      Deactivate the GPGCEDEV device\n"
               "  --activate        Activate the GPGCEDEV devive\n"
               "  --enable-debug    Enable debugging of GPGCEDEV device\n"
               "  --disable-debug   Disable debugging of GPGCEDEV device\n"
               "  --gravity         Show output of the gravity sensor\n"
               "  --enable-log      Enable logging via \"GPG2:\"\n"
               "  --disable-log     Disable logging via \"GPG2:\"\n"
               "  --log             Write a test string to \"GPG2:\"\n"
               "  --gpgme-log [ARG] Show or set GPGME log output\n"
               "  --gnupg-log [ARG] Show or set GnuPG default log file\n"
               "                    (No ARG shows, \"none\" disables)\n"
               );
      result = 1;
    }

  fflush (stdout);
  fflush (stderr);
  Sleep (1000);
  return result;
}