home *** CD-ROM | disk | FTP | other *** search
/ Tricks of the Windows Gam…ming Gurus (2nd Edition) / Disc2.iso / msdn_vcb / samples / vc98 / sdk / netds / winsock / nwlink / dgrecv / dgrecv.c next >
C/C++ Source or Header  |  1997-10-05  |  7KB  |  313 lines
  1. /****************************************************************************\
  2. *  dgrecv.c -- sample program demonstrating NWLink.
  3. *
  4. *       Microsoft Developer Support
  5. *       Copyright (c) 1992-1997 Microsoft Corporation
  6. *
  7. *  This program is a simple example of opening a socket,
  8. *  binding to the socket, and waiting for a datagram packet.
  9. * 
  10. ****************************************************************************/
  11. #include <windows.h>
  12. #include <stdio.h>
  13. #include <stdlib.h>
  14. #include <string.h>
  15. #include <malloc.h>
  16. #include <wsipx.h>
  17. #include <wsnwlink.h>
  18. #include "../testlib/testlib.h"
  19.  
  20. /*
  21. *   Sockaddr structures 
  22. */
  23.  
  24. SOCKADDR_IPX addr;
  25. SOCKADDR_IPX baddr;
  26. SOCKADDR_IPX raddr;
  27.  
  28. /*
  29. *   Function Prototypes 
  30. */
  31.  
  32. extern int main(int, char **);
  33. extern int net_init(SOCKET *);
  34. extern int dg_recv(SOCKET);
  35.  
  36. /****************************************************************************
  37. *
  38. *    FUNCTION:  main( int argc, char **argv )
  39. *
  40. *    PURPOSE:   This is the main entry for the program
  41. *               
  42. *
  43. *    ARGUMENTS:    argc = Number of arguments
  44. *               argv = Array of ptrs to cmd line args
  45. *                
  46. *
  47. *     RETURNS:   Exit code for the program
  48. *                
  49. *\***************************************************************************/
  50. int main(int argc, char **argv)
  51. {
  52.     SOCKET s;
  53.  
  54.     /*
  55.     *   Initialize our default values before checking the command line 
  56.     */
  57.  
  58.     *Local_Socket_Number = 0x06;
  59.     *(Local_Socket_Number+1) = 0x00;
  60.  
  61.     /*
  62.     *   Parse the command line to set up any command line options 
  63.     */
  64.  
  65.     parse_cmd_line(argc, argv);
  66.  
  67.     /*
  68.     *   Initialize the network and set up our socket 
  69.     */
  70.  
  71.     if (net_init(&s))
  72.         return 1;
  73.  
  74.     /*
  75.     *   Do the receive thing 
  76.     */
  77.  
  78.     if (dg_recv(s))
  79.         return 1;
  80.  
  81.     /*
  82.     *   All Done - Close up the socket and exit 
  83.     */
  84.  
  85.     if (verbose)
  86.     printf("Calling closesocket()\n");
  87.  
  88.     closesocket(s);
  89.  
  90.     return 0;
  91. }
  92.  
  93. /****************************************************************************
  94. *
  95. *    FUNCTION:  net_init( SOCKET *skt )
  96. *
  97. *    PURPOSE:   Initializes the WinSock stuff and sets up our socket.
  98. *               
  99. *
  100. *    ARGUMENTS:    SOCKET * => struct to receive our socket info    
  101. *
  102. *     RETURNS:   0 if ok
  103. *                1 if error
  104. *
  105. *\***************************************************************************/
  106. int net_init(SOCKET *skt)
  107. {
  108.     SOCKET s;
  109.     int rc, addrlen = 16;
  110.     WSADATA wsdata;
  111.     WORD    wVersionRequested;
  112.  
  113.     if (verbose)
  114.         printf("Calling WSAStartup(), ");
  115.  
  116.     /*
  117.     *   Initialize with the WINSOCK library 
  118.     */
  119.  
  120.     wVersionRequested = MAKEWORD(1,1);
  121.     rc = WSAStartup(wVersionRequested, &wsdata);
  122.  
  123.     if (verbose)
  124.         printf("return = 0x%X (%d)\n", rc, rc);
  125.  
  126.     if (rc) {
  127.         printf("WSAStartup failed: error code = %d\n", rc);
  128.         return 1;
  129.     }
  130.  
  131.     if (verbose) {
  132.         printf("Contents of wsadata struct:\n");
  133.         print_wsa(&wsdata);
  134.     }
  135.  
  136.     /*
  137.     *   Open a DATAGRAM socket with IPX 
  138.     */
  139.  
  140.     if (verbose)
  141.         printf("Calling socket(address family = %d, socket type = %d, protocol = %d)\n", Local_Address_Family, Socket_Type, Protocol);
  142.  
  143.     s = socket(Local_Address_Family, Socket_Type, Protocol);
  144.  
  145.     if (verbose)
  146.     printf("socket() returned 0x%X (%d)\n", s, s);
  147.  
  148.     if (s == INVALID_SOCKET) {
  149.         dos_net_perror("Socket call failed");
  150.         exit(1);
  151.     }
  152.  
  153.     /*
  154.     *    Bind to a socket.  We dont care what socket we bind to,
  155.     *    so we will send down all 0's
  156.     */
  157.  
  158.     addr.sa_family = Local_Address_Family;
  159.  
  160.     memcpy(&addr.sa_netnum, Local_Network_Number, 4);
  161.     memcpy(&addr.sa_nodenum, Local_Node_Number, 6);
  162.     memcpy(&addr.sa_socket, Local_Socket_Number, 2);
  163.  
  164.     if (verbose) {
  165.         printf("calling bind(), local address =\n  ");
  166.         print_saddr(&addr);
  167.     }
  168.  
  169.     rc = bind(s, (struct sockaddr *) &addr, 16);
  170.  
  171.     if (verbose)
  172.         printf("\nbind() returned 0x%X (%d)\n", rc, rc);
  173.  
  174.     if (rc == SOCKET_ERROR) {
  175.         dos_net_perror("Error binding to socket");
  176.         closesocket(s);
  177.         return 1;
  178.     }
  179.  
  180.     /*
  181.     *   Get the address we bound to and print it out 
  182.     */
  183.  
  184.     if (verbose)
  185.         printf("calling getsockname(socket = %d), ", s);
  186.  
  187.     rc = getsockname(s, (struct sockaddr *) &baddr, &addrlen);
  188.  
  189.     if (verbose)
  190.         printf("return = 0x%lX (%d)\n", rc, rc);
  191.  
  192.     if (rc == SOCKET_ERROR) {
  193.         dos_net_perror("Error getting socket name");
  194.         closesocket(s);
  195.         return 1;
  196.     }
  197.  
  198.     /*
  199.     *   Set the packet type to send for this socket 
  200.     */
  201.  
  202.     if (verbose)
  203.         printf("Calling setsockopt for packet type %d\n", Local_Packet_Type);
  204.  
  205.     rc = setsockopt(s, NSPROTO_IPX, IPX_PTYPE, (const char *) &Local_Packet_Type, 4);
  206.  
  207.     if (rc == SOCKET_ERROR)
  208.         dos_net_perror("setsockopt() call failed");
  209.  
  210.  
  211.     if (Filter_Packet_Type) {
  212.  
  213.         /*
  214.         *   Set the packet type for this socket 
  215.         */
  216.  
  217.         if (verbose)
  218.             printf("Calling setsockopt to filter packet type %d\n", Filter_Packet_Type);
  219.  
  220.     rc = setsockopt(s, NSPROTO_IPX, IPX_FILTERPTYPE, (const char *) &Filter_Packet_Type, sizeof(int));
  221.     printf("RC from FILTER SOCKOPT = %d\n", rc);
  222.  
  223.         if (rc == SOCKET_ERROR)
  224.             dos_net_perror("setsockopt() call failed setting filter packet type");
  225.     }
  226.     /*
  227.     *   Print out the network address 
  228.     */
  229.  
  230.     if (verbose) {
  231.         printf("addrlen = %d\n", addrlen);
  232.         print_netaddr(baddr.sa_netnum, "  Bound to address ", "\n");
  233.     }
  234.  
  235.     *skt = s;
  236.  
  237.     return 0;
  238. }
  239.  
  240. /****************************************************************************
  241. *
  242. *    FUNCTION:  dg_recv( SOCKET s )
  243. *
  244. *    PURPOSE:   Receives datagrams.
  245. *
  246. *    ARGUMENTS:    SOCKET   socket to receive on
  247. *
  248. *     RETURNS:   0 if ok
  249. *               1 if error
  250. *
  251. *\***************************************************************************/
  252. int dg_recv(SOCKET s)
  253. {
  254.     int rc, errflag = 0;
  255.     int addrlen = 16;
  256.     UINT dgrams = 0;
  257.     LPSTR recvbuf;
  258.  
  259.  
  260.     if (verbose)
  261.         printf("allocating %d bytes for receive buffer\n", Receive_Length);
  262.  
  263.     /*
  264.     *   Set up the data buffer to send 
  265.     */
  266.  
  267.     recvbuf = (LPSTR)malloc(Send_Length);
  268.  
  269.     if (!recvbuf) {
  270.         printf("Error allocating %d bytes for receive buffer\n", Receive_Length);
  271.         return 1;
  272.     }
  273.  
  274.     if (verbose) {
  275.         printf("calling recvfrom(socket = %d, length = %d),\n", s, Receive_Length);
  276.     }
  277.     else
  278.         printf("Waiting for call...\n");
  279.  
  280.     while (1) {
  281.         /*
  282.         *   Receive a packet from anyone 
  283.         */
  284.  
  285.         rc = recvfrom(s, recvbuf, Receive_Length, 0, (struct sockaddr *)&raddr, &addrlen);
  286.  
  287.         if (rc == SOCKET_ERROR) {
  288.             dos_net_perror("recvfrom() failed");
  289.             errflag++;
  290.             break;
  291.         }
  292.  
  293.         if (verbose) {
  294.             printf("Received %d bytes from \n  ", rc);
  295.             print_saddr(&raddr);
  296.             printf("\n");
  297.         }
  298.         else
  299.             printf("\rReceived packet %d, length = %d bytes", ++dgrams, rc);
  300.  
  301.  
  302.         if (No_Loop)
  303.             break;
  304.     }
  305.  
  306.     if (verbose)
  307.         printf("Freeing receive buffer\n");
  308.  
  309.     free(recvbuf);
  310.  
  311.     return errflag;
  312. }
  313.