text: /* TEMPLATE GENERATED TESTCASE FILE
  Filename: CWE78_OS_Command_Injection__wchar_t_listen_socket_w32spawnl_06.c
  Label Definition File: CWE78_OS_Command_Injection.no_path.label.xml
  Template File: sources-sink-06.tmpl.c
  */
  /*
   * @description
   * CWE: 78 OS Command Injection
   * BadSource: listen_socket Read data using a listen socket (server side)
   * GoodSource: Benign input
   * Sink: w32spawnl
   *    BadSink : execute command with wspawnl
   * Flow Variant: 06 Control flow: if(static_const_five==5) and if(static_const_five!=5)
   *
   * */
  
  #include "std_testcase.h"
  
  #include <wchar.h>
  
  #ifdef _WIN32
  # define COMMAND_INT_PATH L"%WINDIR%\\system32\\cmd.exe"
  # define COMMAND_INT L"cmd.exe"
  # define COMMAND_ARG1 L"/c"
  # define COMMAND_ARG2 L"dir"
  # define COMMAND_ARG3 data
  #else /* NOT _WIN32 */
  # define COMMAND_INT_PATH L"/bin/sh"
  # define COMMAND_INT L"sh"
  # define COMMAND_ARG1 L"ls"
  # define COMMAND_ARG2 data
  # define COMMAND_ARG3 NULL
  #endif
  
  #ifdef _WIN32
  # include <winsock2.h>
  # include <windows.h>
  # include <direct.h>
  # define PATH_SZ 100
  # pragma comment(lib, "ws2_32") /* include ws2_32.lib when linking */
  # define CLOSE_SOCKET closesocket
  #else
  # define PATH_SZ PATH_MAX
  # define INVALID_SOCKET -1
  # define SOCKET_ERROR -1
  # define CLOSE_SOCKET close
  # define SOCKET int
  #endif
  
  #define TCP_PORT 27015
  #define LISTEN_BACKLOG 5
  
  #include <process.h>
  
  /* The variable below is declared "const", so a tool should be able
     to identify that reads of this will always give its initialized
     value. */
  static const int static_const_five = 5;
  
  #ifndef OMITBAD
  
  void CWE78_OS_Command_Injection__wchar_t_listen_socket_w32spawnl_06_bad()
  {
      wchar_t * data;
      wchar_t data_buf[100] = L"";
      data = data_buf;
      if(static_const_five==5)
      {
          {
  #ifdef _WIN32
              WSADATA wsa_data;
              int wsa_data_init = 0;
  #endif
              int recv_rv;
              struct sockaddr_in s_in;
              wchar_t *replace;
              SOCKET listen_socket = INVALID_SOCKET;
              SOCKET accept_socket = INVALID_SOCKET;
              size_t data_len = wcslen(data);
              do
              {
  #ifdef _WIN32
                  if (WSAStartup(MAKEWORD(2,2), &wsa_data) != NO_ERROR) break;
                  wsa_data_init = 1;
  #endif
                  listen_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
                  if (listen_socket == INVALID_SOCKET) break;
                  memset(&s_in, 0, sizeof(s_in));
                  s_in.sin_family = AF_INET;
                  s_in.sin_addr.s_addr = INADDR_ANY;
                  s_in.sin_port = htons(TCP_PORT);
                  if (bind(listen_socket, (struct sockaddr*)&s_in, sizeof(s_in)) == SOCKET_ERROR) break;
                  if (listen(listen_socket, LISTEN_BACKLOG) == SOCKET_ERROR) break;
                  accept_socket = accept(listen_socket, NULL, NULL);
                  if (accept_socket == SOCKET_ERROR) break;
                  /* Abort on error or the connection was closed */
                  recv_rv = recv(accept_socket, (char *)data+data_len, (int)(100-data_len-1), 0);
                  if (recv_rv == SOCKET_ERROR || recv_rv == 0) break;
                  /* Append null terminator */
                  data[recv_rv] = L'\0';
                  /* Eliminate CRLF */
                  replace = wcschr(data, L'\r');
                  if (replace) *replace = L'\0';
                  replace = wcschr(data, L'\n');
                  if (replace) *replace = L'\0';
              }
              while (0);
              if (listen_socket != INVALID_SOCKET) CLOSE_SOCKET(listen_socket);
              if (accept_socket != INVALID_SOCKET) CLOSE_SOCKET(accept_socket);
  #ifdef _WIN32
              if (wsa_data_init) WSACleanup();
  #endif
          }
      }
      else
      {
          /* INCIDENTAL: CWE 561 Dead Code, the code below will never run */
          /* FIX: Benign input preventing command injection */
          wcscat(data, L"*.*");
      }
      /* wspawnl - specify the path where the command is located */
      /* POSSIBLE FLAW: Execute command without validating input possibly leading to command injection */
      _wspawnl(_P_WAIT, COMMAND_INT_PATH, COMMAND_INT_PATH, COMMAND_ARG1, COMMAND_ARG2, COMMAND_ARG3, NULL);
  }
  
  #endif /* OMITBAD */
  
  #ifndef OMITGOOD
  
  /* goodG2B1() - use goodsource and badsink by changing the static_const_five==5 to static_const_five!=5 */
  static void goodG2B1()
  {
      wchar_t * data;
      wchar_t data_buf[100] = L"";
      data = data_buf;
      if(static_const_five!=5)
      {
          /* INCIDENTAL: CWE 561 Dead Code, the code below will never run */
          {
  #ifdef _WIN32
              WSADATA wsa_data;
              int wsa_data_init = 0;
  #endif
              int recv_rv;
              struct sockaddr_in s_in;
              wchar_t *replace;
              SOCKET listen_socket = INVALID_SOCKET;
              SOCKET accept_socket = INVALID_SOCKET;
              size_t data_len = wcslen(data);
              do
              {
  #ifdef _WIN32
                  if (WSAStartup(MAKEWORD(2,2), &wsa_data) != NO_ERROR) break;
                  wsa_data_init = 1;
  #endif
                  listen_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
                  if (listen_socket == INVALID_SOCKET) break;
                  memset(&s_in, 0, sizeof(s_in));
                  s_in.sin_family = AF_INET;
                  s_in.sin_addr.s_addr = INADDR_ANY;
                  s_in.sin_port = htons(TCP_PORT);
                  if (bind(listen_socket, (struct sockaddr*)&s_in, sizeof(s_in)) == SOCKET_ERROR) break;
                  if (listen(listen_socket, LISTEN_BACKLOG) == SOCKET_ERROR) break;
                  accept_socket = accept(listen_socket, NULL, NULL);
                  if (accept_socket == SOCKET_ERROR) break;
                  /* Abort on error or the connection was closed */
                  recv_rv = recv(accept_socket, (char *)data+data_len, (int)(100-data_len-1), 0);
                  if (recv_rv == SOCKET_ERROR || recv_rv == 0) break;
                  /* Append null terminator */
                  data[recv_rv] = L'\0';
                  /* Eliminate CRLF */
                  replace = wcschr(data, L'\r');
                  if (replace) *replace = L'\0';
                  replace = wcschr(data, L'\n');
                  if (replace) *replace = L'\0';
              }
              while (0);
              if (listen_socket != INVALID_SOCKET) CLOSE_SOCKET(listen_socket);
              if (accept_socket != INVALID_SOCKET) CLOSE_SOCKET(accept_socket);
  #ifdef _WIN32
              if (wsa_data_init) WSACleanup();
  #endif
          }
      }
      else
      {
          /* FIX: Benign input preventing command injection */
          wcscat(data, L"*.*");
      }
      /* wspawnl - specify the path where the command is located */
      /* POSSIBLE FLAW: Execute command without validating input possibly leading to command injection */
      _wspawnl(_P_WAIT, COMMAND_INT_PATH, COMMAND_INT_PATH, COMMAND_ARG1, COMMAND_ARG2, COMMAND_ARG3, NULL);
  }
  
  /* goodG2B2() - use goodsource and badsink by reversing the blocks in the if statement */
  static void goodG2B2()
  {
      wchar_t * data;
      wchar_t data_buf[100] = L"";
      data = data_buf;
      if(static_const_five==5)
      {
          /* FIX: Benign input preventing command injection */
          wcscat(data, L"*.*");
      }
      else
      {
          /* INCIDENTAL: CWE 561 Dead Code, the code below will never run */
          {
  #ifdef _WIN32
              WSADATA wsa_data;
              int wsa_data_init = 0;
  #endif
              int recv_rv;
              struct sockaddr_in s_in;
              wchar_t *replace;
              SOCKET listen_socket = INVALID_SOCKET;
              SOCKET accept_socket = INVALID_SOCKET;
              size_t data_len = wcslen(data);
              do
              {
  #ifdef _WIN32
                  if (WSAStartup(MAKEWORD(2,2), &wsa_data) != NO_ERROR) break;
                  wsa_data_init = 1;
  #endif
                  listen_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
                  if (listen_socket == INVALID_SOCKET) break;
                  memset(&s_in, 0, sizeof(s_in));
                  s_in.sin_family = AF_INET;
                  s_in.sin_addr.s_addr = INADDR_ANY;
                  s_in.sin_port = htons(TCP_PORT);
                  if (bind(listen_socket, (struct sockaddr*)&s_in, sizeof(s_in)) == SOCKET_ERROR) break;
                  if (listen(listen_socket, LISTEN_BACKLOG) == SOCKET_ERROR) break;
                  accept_socket = accept(listen_socket, NULL, NULL);
                  if (accept_socket == SOCKET_ERROR) break;
                  /* Abort on error or the connection was closed */
                  recv_rv = recv(accept_socket, (char *)data+data_len, (int)(100-data_len-1), 0);
                  if (recv_rv == SOCKET_ERROR || recv_rv == 0) break;
                  /* Append null terminator */
                  data[recv_rv] = L'\0';
                  /* Eliminate CRLF */
                  replace = wcschr(data, L'\r');
                  if (replace) *replace = L'\0';
                  replace = wcschr(data, L'\n');
                  if (replace) *replace = L'\0';
              }
              while (0);
              if (listen_socket != INVALID_SOCKET) CLOSE_SOCKET(listen_socket);
              if (accept_socket != INVALID_SOCKET) CLOSE_SOCKET(accept_socket);
  #ifdef _WIN32
              if (wsa_data_init) WSACleanup();
  #endif
          }
      }
      /* wspawnl - specify the path where the command is located */
      /* POSSIBLE FLAW: Execute command without validating input possibly leading to command injection */
      _wspawnl(_P_WAIT, COMMAND_INT_PATH, COMMAND_INT_PATH, COMMAND_ARG1, COMMAND_ARG2, COMMAND_ARG3, NULL);
  }
  
  void CWE78_OS_Command_Injection__wchar_t_listen_socket_w32spawnl_06_good()
  {
      goodG2B1();
      goodG2B2();
  }
  
  #endif /* OMITGOOD */
  
  /* Below is the main(). It is only used when building this testcase on
     its own for testing or for building a binary to use in testing binary
     analysis tools. It is not used when compiling all the testcases as one
     application, which is how source code analysis tools are tested. */
  
  #ifdef INCLUDEMAIN
  
  int main(int argc, char * argv[])
  {
      /* seed randomness */
      srand( (unsigned)time(NULL) );
  #ifndef OMITGOOD
      printLine("Calling good()...");
      CWE78_OS_Command_Injection__wchar_t_listen_socket_w32spawnl_06_good();
      printLine("Finished good()");
  #endif /* OMITGOOD */
  #ifndef OMITBAD
      printLine("Calling bad()...");
      CWE78_OS_Command_Injection__wchar_t_listen_socket_w32spawnl_06_bad();
      printLine("Finished bad()");
  #endif /* OMITBAD */
      return 0;
  }
  
  #endif
  