TestSetCommState.c

 
#include <stdio.h>
#include <sys/stat.h>
 
#include <winpr/comm.h>
#include <winpr/crt.h>
 
#include "../comm.h"
 
static void init_empty_dcb(DCB* pDcb)
{
  WINPR_ASSERT(pDcb);
 
  ZeroMemory(pDcb, sizeof(DCB));
  pDcb->DCBlength = sizeof(DCB);
  pDcb->XonChar = 1;
  pDcb->XoffChar = 2;
}
 
static BOOL test_fParity(HANDLE hComm)
{
  DCB dcb = { 0 };
  BOOL result = 0;
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  /* test 1 */
  dcb.fParity = TRUE;
  result = SetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "SetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  if (!dcb.fParity)
  {
    (void)fprintf(stderr, "unexpected fParity: %" PRIu32 " instead of TRUE\n", dcb.fParity);
    return FALSE;
  }
 
  /* test 2 */
  dcb.fParity = FALSE;
  result = SetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "SetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  if (dcb.fParity)
  {
    (void)fprintf(stderr, "unexpected fParity: %" PRIu32 " instead of FALSE\n", dcb.fParity);
    return FALSE;
  }
 
  /* test 3 (redo test 1) */
  dcb.fParity = TRUE;
  result = SetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "SetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%08" PRIx32 "\n", GetLastError());
    return FALSE;
  }
 
  if (!dcb.fParity)
  {
    (void)fprintf(stderr, "unexpected fParity: %" PRIu32 " instead of TRUE\n", dcb.fParity);
    return FALSE;
  }
 
  return TRUE;
}
 
static BOOL test_SerialSys(HANDLE hComm)
{
  DCB dcb = { 0 };
  BOOL result = 0;
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%x\n", GetLastError());
    return FALSE;
  }
 
  /* Test 1 */
  dcb.BaudRate = CBR_115200;
  result = SetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "SetCommState failure: 0x%08x\n", GetLastError());
    return FALSE;
  }
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%x\n", GetLastError());
    return FALSE;
  }
  if (dcb.BaudRate != CBR_115200)
  {
    (void)fprintf(stderr, "SetCommState failure: could not set BaudRate=%d (CBR_115200)\n",
                  CBR_115200);
    return FALSE;
  }
 
  /* Test 2 using a defferent baud rate */
 
  dcb.BaudRate = CBR_57600;
  result = SetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "SetCommState failure: 0x%x\n", GetLastError());
    return FALSE;
  }
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%x\n", GetLastError());
    return FALSE;
  }
  if (dcb.BaudRate != CBR_57600)
  {
    (void)fprintf(stderr, "SetCommState failure: could not set BaudRate=%d (CBR_57600)\n",
                  CBR_57600);
    return FALSE;
  }
 
  /* Test 3 using an unsupported baud rate  on Linux */
  dcb.BaudRate = CBR_128000;
  result = SetCommState(hComm, &dcb);
  if (result)
  {
    (void)fprintf(stderr,
                  "SetCommState failure: unexpected support of BaudRate=%d (CBR_128000)\n",
                  CBR_128000);
    return FALSE;
  }
 
  return TRUE;
}
 
static BOOL test_SerCxSys(HANDLE hComm)
{
  /* as of today there is no difference */
  return test_SerialSys(hComm);
}
 
static BOOL test_SerCx2Sys(HANDLE hComm)
{
  /* as of today there is no difference */
  return test_SerialSys(hComm);
}
 
static BOOL test_generic(HANDLE hComm)
{
  DCB dcb = { 0 };
  DCB dcb2 = { 0 };
  BOOL result = 0;
 
  init_empty_dcb(&dcb);
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%x\n", GetLastError());
    return FALSE;
  }
 
  /* Checks whether we get the same information before and after SetCommState */
  memcpy(&dcb2, &dcb, sizeof(DCB));
  result = SetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "SetCommState failure: 0x%08x\n", GetLastError());
    return FALSE;
  }
 
  result = GetCommState(hComm, &dcb);
  if (!result)
  {
    (void)fprintf(stderr, "GetCommState failure: 0x%x\n", GetLastError());
    return FALSE;
  }
 
  if (memcmp(&dcb, &dcb2, sizeof(DCB)) != 0)
  {
    (void)fprintf(stderr,
                  "DCB is different after SetCommState() whereas it should have not changed\n");
    return FALSE;
  }
 
  // TODO: a more complete and generic test using GetCommProperties()
 
  /* TMP: TODO: fBinary tests */
 
  /* fParity tests */
  if (!test_fParity(hComm))
  {
    (void)fprintf(stderr, "test_fParity failure\n");
    return FALSE;
  }
 
  return TRUE;
}
 
int TestSetCommState(int argc, char* argv[])
{
  struct stat statbuf = { 0 };
  BOOL result = 0;
  HANDLE hComm = NULL;
 
  if (stat("/dev/ttyS0", &statbuf) < 0)
  {
    (void)fprintf(stderr, "/dev/ttyS0 not available, making the test to succeed though\n");
    return EXIT_SUCCESS;
  }
 
  result = DefineCommDevice("COM1", "/dev/ttyS0");
  if (!result)
  {
    (void)fprintf(stderr, "DefineCommDevice failure: 0x%x\n", GetLastError());
    return EXIT_FAILURE;
  }
 
  hComm = CreateFile("COM1", GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
  if (hComm == INVALID_HANDLE_VALUE)
  {
    (void)fprintf(stderr, "CreateFileA failure: 0x%x\n", GetLastError());
    return EXIT_FAILURE;
  }
 
  if (!test_generic(hComm))
  {
    (void)fprintf(stderr, "test_generic failure (SerialDriverUnknown)\n");
    return EXIT_FAILURE;
  }
 
  _comm_setServerSerialDriver(hComm, SerialDriverSerialSys);
  if (!test_generic(hComm))
  {
    (void)fprintf(stderr, "test_generic failure (SerialDriverSerialSys)\n");
    return EXIT_FAILURE;
  }
  if (!test_SerialSys(hComm))
  {
    (void)fprintf(stderr, "test_SerialSys failure\n");
    return EXIT_FAILURE;
  }
 
  _comm_setServerSerialDriver(hComm, SerialDriverSerCxSys);
  if (!test_generic(hComm))
  {
    (void)fprintf(stderr, "test_generic failure (SerialDriverSerCxSys)\n");
    return EXIT_FAILURE;
  }
  if (!test_SerCxSys(hComm))
  {
    (void)fprintf(stderr, "test_SerCxSys failure\n");
    return EXIT_FAILURE;
  }
 
  _comm_setServerSerialDriver(hComm, SerialDriverSerCx2Sys);
  if (!test_generic(hComm))
  {
    (void)fprintf(stderr, "test_generic failure (SerialDriverSerCx2Sys)\n");
    return EXIT_FAILURE;
  }
  if (!test_SerCx2Sys(hComm))
  {
    (void)fprintf(stderr, "test_SerCx2Sys failure\n");
    return EXIT_FAILURE;
  }
 
  if (!CloseHandle(hComm))
  {
    (void)fprintf(stderr, "CloseHandle failure, GetLastError()=%08x\n", GetLastError());
    return EXIT_FAILURE;
  }
 
  return EXIT_SUCCESS;
}