text: /* TEMPLATE GENERATED TESTCASE FILE
  Filename: CWE789_Uncontrolled_Mem_Alloc__new_char_connect_socket_54b.cpp
  Label Definition File: CWE789_Uncontrolled_Mem_Alloc__new.label.xml
  Template File: sources-sinks-54b.tmpl.cpp
  */
  /*
   * @description
   * CWE: 789 Uncontrolled Memory Allocation
   * BadSource: connect_socket Read data using a connect socket (client side)
   * GoodSource: Small number greater than zero
   * Sinks:
   *    GoodSink: Allocate memory with new [] and check the size of the memory to be allocated
   *    BadSink : Allocate memory with new [], but incorrectly check the size of the memory to be allocated
   * Flow Variant: 54 Data flow: data passed as an argument from one function through three others to a fifth; all five functions are in different source files
   *
   * */
  
  #include "std_testcase.h"
  
  #ifndef _WIN32
  #include <wchar.h>
  #endif
  
  #ifdef _WIN32
  #include <winsock2.h>
  #include <windows.h>
  #include <direct.h>
  #pragma comment(lib, "ws2_32") /* include ws2_32.lib when linking */
  #define CLOSE_SOCKET closesocket
  #else /* NOT _WIN32 */
  #include <sys/types.h>
  #include <sys/socket.h>
  #include <netinet/in.h>
  #include <arpa/inet.h>
  #include <unistd.h>
  #define INVALID_SOCKET -1
  #define SOCKET_ERROR -1
  #define CLOSE_SOCKET close
  #define SOCKET int
  #endif
  
  #define TCP_PORT 27015
  #define IP_ADDRESS "127.0.0.1"
  #define CHAR_ARRAY_SIZE (3 * sizeof(data) + 2)
  
  #define HELLO_STRING "hello"
  
  namespace CWE789_Uncontrolled_Mem_Alloc__new_char_connect_socket_54
  {
  
  #ifndef OMITBAD
  
  /* bad function declaration */
  void badSink_c(size_t data);
  
  void badSink_b(size_t data)
  {
      badSink_c(data);
  }
  
  #endif /* OMITBAD */
  
  #ifndef OMITGOOD
  
  /* goodG2B uses the GoodSource with the BadSink */
  void goodG2BSink_c(size_t data);
  
  void goodG2BSink_b(size_t data)
  {
      goodG2BSink_c(data);
  }
  
  /* goodB2G uses the BadSource with the GoodSink */
  void goodB2GSink_c(size_t data);
  
  void goodB2GSink_b(size_t data)
  {
      goodB2GSink_c(data);
  }
  
  #endif /* OMITGOOD */
  
  } /* close namespace */
  