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 / ping / ping.c < prev    next >
C/C++ Source or Header  |  1997-10-05  |  8KB  |  339 lines
  1. /****************************************************************************\
  2. *  ping.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, receiving a packet and sending
  9. *  that packet back to the original sender.
  10. * 
  11. ****************************************************************************/
  12. #include <windows.h>
  13. #include <stdio.h>
  14. #include <stdlib.h>
  15. #include <string.h>
  16. #include <malloc.h>
  17. #include <wsipx.h>
  18. #include <wsnwlink.h>
  19. #include "../testlib/testlib.h"
  20.  
  21. /*
  22. *   Sockaddr structures 
  23. */
  24.  
  25. SOCKADDR_IPX addr;
  26. SOCKADDR_IPX baddr;
  27. SOCKADDR_IPX raddr;
  28. int addrlen;
  29.  
  30. /*
  31. *   Function Prototypes 
  32. */
  33.  
  34. extern int main(int, char **);
  35. extern int net_init(SOCKET *);
  36. extern int do_ping(SOCKET);
  37.  
  38. /****************************************************************************
  39. *
  40. *    FUNCTION:  main( int argc, char **argv )
  41. *
  42. *    PURPOSE:   This is the main entry for the program
  43. *               
  44. *
  45. *    ARGUMENTS:    argc = Number of arguments
  46. *               argv = Array of ptrs to cmd line args
  47. *                
  48. *
  49. *     RETURNS:   Exit code for the program
  50. *                
  51. *\***************************************************************************/
  52. int main(int argc, char **argv)
  53. {
  54.     SOCKET s;
  55.  
  56.     /*
  57.     *   Set our default values before calling parse_cmd_line 
  58.     */
  59.  
  60.     *Local_Socket_Number = 0x30;
  61.     *(Local_Socket_Number+1) = 0x00;
  62.     Receive_Length = 2048;
  63.  
  64.     /*
  65.     *   Get any command line options 
  66.     */
  67.  
  68.     parse_cmd_line(argc, argv);
  69.  
  70.     /*
  71.     *   Initialize the network and set up our socket 
  72.     */
  73.  
  74.     if (net_init(&s))
  75.         return 1;
  76.  
  77.     do_ping(s);
  78.  
  79.     /*
  80.     *   All done (We only get here on error) 
  81.     */
  82.  
  83.     if (verbose)
  84.         printf("calling closesocket()");
  85.  
  86.     closesocket(s);
  87.     return 0;
  88. }
  89.  
  90. /****************************************************************************
  91. *
  92. *    FUNCTION:  net_init( SOCKET *skt )
  93. *
  94. *    PURPOSE:   Initializes the WinSock stuff and sets up our socket.
  95. *               
  96. *
  97. *    ARGUMENTS:    SOCKET * => struct to receive our socket info    
  98. *
  99. *     RETURNS:   0 if ok
  100. *                1 if error
  101. *
  102. *\***************************************************************************/
  103. int net_init(SOCKET *skt)
  104. {
  105.     int rc, addrlen;
  106.     WSADATA wsdata;
  107.     SOCKET s;
  108.     WORD    wVersionRequested;
  109.  
  110.     /*
  111.     *   Initialize with the WINSOCK library 
  112.     */
  113.  
  114.     if (verbose)
  115.         printf("calling WSAStartup(), ");
  116.  
  117.     wVersionRequested = MAKEWORD(1,1);
  118.     rc = WSAStartup(wVersionRequested, &wsdata);
  119.  
  120.     if (verbose)
  121.         printf("return = 0x%X (%d)\n", rc, rc);
  122.  
  123.     if (rc) {
  124.         printf("WSAStartup failed: error code = %d\n", rc);
  125.         return 1;
  126.     }
  127.  
  128.     if (verbose) {
  129.         printf("contents of wsdata struct: \n");
  130.         print_wsa(&wsdata);
  131.     }
  132.  
  133.     if (verbose)
  134.         printf("calling socket(address family = %d, socket type = %d, protocol = %d)\n", Local_Address_Family, Socket_Type, Protocol);
  135.  
  136.     /*
  137.     *   Open a DATAGRAM socket with IPX 
  138.     */
  139.  
  140.     s = socket(AF_NS, SOCK_DGRAM, NSPROTO_IPX);
  141.  
  142.     if (verbose)
  143.         printf("socket() returned 0x%lX\n", s);
  144.  
  145.     if (s == INVALID_SOCKET) {
  146.         dos_net_perror("Socket call failed");
  147.         exit(1);
  148.     }
  149.  
  150.     /*
  151.     *   Bind to a socket.  We want to bind to a well known
  152.     *   socket so that the app. that sends us a packet will
  153.     *   know where to send it.
  154.     */
  155.  
  156.     addr.sa_family = Local_Address_Family;
  157.  
  158.     memcpy(&addr.sa_netnum, Local_Network_Number, 4);
  159.     memcpy(&addr.sa_nodenum, Local_Node_Number, 6);
  160.     memcpy(&addr.sa_socket, Local_Socket_Number, 2);
  161.  
  162.     if (verbose) {
  163.         printf("calling bind():\n  ");
  164.         print_saddr(&addr);
  165.     }
  166.  
  167.     rc = bind(s, (const struct sockaddr *) &addr, 16);
  168.  
  169.     if (verbose)
  170.         printf("bind() returned 0x%X\n", rc);
  171.  
  172.     if (rc == SOCKET_ERROR) {
  173.         dos_net_perror("Error binding to socket");
  174.         closesocket(s);
  175.         return 1;
  176.     }
  177.     /*
  178.     *   Set the packet type for this socket 
  179.     */
  180.  
  181.     if (verbose)
  182.         printf("Calling setsockopt for packet type %d\n", Local_Packet_Type);
  183.  
  184.     rc = setsockopt(s, SOL_SOCKET, IPX_PTYPE, (const char *) &Local_Packet_Type, 4);
  185.  
  186.     if (rc == SOCKET_ERROR)
  187.         dos_net_perror("setsockopt() call failed");
  188.  
  189.  
  190.     /*
  191.     *   Get the address we bound to and print it out 
  192.     */
  193.  
  194.     if (verbose)
  195.         printf("Calling getsockname(socket = %d), ");
  196.  
  197.     addrlen = 16;
  198.     rc = getsockname(s, (struct sockaddr *) &baddr, &addrlen);
  199.  
  200.     if (verbose)
  201.         printf("return = 0x%X (%d)\n", rc, rc);
  202.  
  203.     if (rc == SOCKET_ERROR) {
  204.         dos_net_perror("Error getting socket name");
  205.         closesocket(s);
  206.         return 1;
  207.     }
  208.  
  209.     /*
  210.     *   Print out the network address 
  211.     */
  212.  
  213.     if (verbose) {
  214.         printf("addrlen = %d\n", addrlen);
  215.         print_netaddr(baddr.sa_netnum, "Bound address = ", "\n");
  216.     }
  217.  
  218.     if (verbose)
  219.         printf("Allocating %d byte receive buffer\n", Receive_Length);
  220.  
  221.     /*
  222.     *   Set up socket to send back 
  223.     */
  224.  
  225.     *skt = s;
  226.  
  227.     return 0;
  228. }
  229.  
  230. /****************************************************************************
  231. *
  232. *    FUNCTION:  do_ping( SOCKET s )
  233. *
  234. *    PURPOSE:   This will receive a packet then send it back.
  235. *               
  236. *    ARGUMENTS:    SOCKET socket we are transmitting on.
  237. *
  238. *     RETURNS:   0 if ok
  239. *                1 if error
  240. *
  241. *\***************************************************************************/
  242. int do_ping(SOCKET s)
  243. {
  244.     LPSTR recvbuf;
  245.     int nbytes, rc, errflag = 0;
  246.     int rcvpkts = 0, sndpkts = 0;
  247.  
  248.     /*
  249.     *   Allocate a buffer for receives 
  250.     */
  251.  
  252.     recvbuf = malloc(Receive_Length);
  253.  
  254.     if (!recvbuf) {
  255.         printf("Error allocating %d bytes for receive buffer\n", Receive_Length);
  256.         return 1;
  257.     }
  258.  
  259.     /*
  260.     *   This loop will receive a packet and then send
  261.     *   it back to whoever sent it to us.
  262.     *
  263.     *      To exit the loop - hit CTRL-C.
  264.     */
  265.  
  266.     while (1) {
  267.  
  268.         /*
  269.         *   Recv a packet 
  270.         */
  271.  
  272.         /*
  273.         *    NOTE:  If you want to know the packet type field of the
  274.         *           packet you just received, look at the byte at
  275.         *           raddr.sa_ptype.  The addrlen returned will be
  276.         *           15.
  277.         *
  278.         *           By using the addrlen field unchanged when sending
  279.         *           the packet back, you will send the packet back with
  280.         *           the same packet type that is was sent with originally
  281.         */
  282.  
  283.         addrlen = 16;
  284.  
  285.         if (verbose)
  286.             printf("calling recvfrom(socket = %d, len = %d)\n", s, Receive_Length);
  287.  
  288.         nbytes = recvfrom(s, recvbuf, Receive_Length, 0, (struct sockaddr *) &raddr, &addrlen);
  289.  
  290.         /*
  291.         *   If error - print it and exit 
  292.         */
  293.  
  294.         if (nbytes == SOCKET_ERROR) {
  295.             dos_net_perror("recvfrom() failed");
  296.             errflag++;
  297.             break;
  298.         }
  299.  
  300.         if (verbose) {
  301.             printf("received %d bytes, raddr = \n  ", nbytes);
  302.             print_saddr(&raddr);
  303.         }
  304.  
  305.         if (!verbose)
  306.             printf("\rRecv packet number %d", ++rcvpkts);
  307.  
  308.         /*
  309.         *   Send the data back 
  310.         */
  311.  
  312.         if (verbose)
  313.             printf("calling sendto(socket = %d, len = %d\n", s, nbytes);
  314.  
  315.     addrlen = 16;
  316.         rc = sendto(s, recvbuf, nbytes, 0, (const struct sockaddr *) &raddr, addrlen);
  317.  
  318.         if (verbose)
  319.             printf("sendto() returned 0x%X\n", rc);
  320.  
  321.         if (rc == SOCKET_ERROR) {
  322.             dos_net_perror("sendto() failed");
  323.             errflag++;
  324.             break;
  325.         }
  326.  
  327.         if (!verbose)
  328.             printf(" : Send packet number %d", ++sndpkts);
  329.  
  330.     }
  331.  
  332.     if (verbose)
  333.         printf("Freeing receive buffer\n");
  334.  
  335.     free(recvbuf);
  336.  
  337.     return errflag;
  338. }
  339.