/*

 *

 * $ gcc -g createpersistenttcpportreservation_test1.c -liphlpapi -lws2_32

 * -o createpersistenttcpportreservation_test1

 */

#ifndef UNICODE

#define UNICODE

#endif

#ifndef WIN32_LEAN_AND_MEAN

#define WIN32_LEAN_AND_MEAN

#endif

#include <Windows.h>

#include <winsock2.h>

#include <mstcpip.h>

#include <ws2ipdef.h>

#include <iphlpapi.h>

#include <stdio.h>

#include <stdlib.h>

#include <wchar.h>

// Need to link with iphlpapi.lib

#pragma comment(lib, "iphlpapi.lib")

// Need to link with Ws2_32.lib for Winsock functions

#pragma comment(lib, "ws2_32.lib")

int main(int argc, char *argv[])

{

    // Declare and initialize variables

    int startPort = 0;         // host byte order

    int numPorts = 0;

    USHORT startPortns = 0;    // Network byte order

    ULONG64 resToken = { 0 };

    unsigned long status = 0;

    WSADATA wsaData = { 0 };

    int iResult = 0;

    SOCKET sock = INVALID_SOCKET;

    int iFamily = AF_INET;

    int iType = SOCK_STREAM;

    int iProtocol = IPPROTO_TCP;

    DWORD bytesReturned = 0;

    // Note that the sockaddr_in struct works only with AF_INET not AF_INET6

    // An application needs to use the sockaddr_in6 for AF_INET6

    struct sockaddr_in service;

    struct sockaddr_in sockName;

    int nameLen = sizeof(sockName);

    // Validate the parameters

    if (argc != 3) {

        printf("usage: %s <Starting Port> <Number of Ports>\n",

             argv[0]);

        printf("Creates a persistent TCP port reservation\n");

        printf("Example usage:\n");

        printf("   %s 5000 20\n", argv[0]);

        printf("   where StartPort=5000 NumPorts=20");

        return 1;

    }

    startPort = atoi(argv[1]);

    if (startPort < 0 || startPort > 65535) {

        printf("Starting point must be either 0 or between 1 and 65,535\n");

        return 1;

    }

    startPortns = htons((USHORT) startPort);

    numPorts = atoi(argv[2]);

    if (numPorts < 0) {

        printf("Number of ports must be a positive number\n");

        return 1;

    }

    status =

        CreatePersistentTcpPortReservation((USHORT) startPortns, (USHORT) numPorts,

                                           &resToken);

    if (status != NO_ERROR) {

        printf("CreatePersistentTcpPortReservation returned error: %ld\n", status);

        return 1;

    }

    printf("CreatePersistentTcpPortReservation call succeeded\n");

    // Comment out this block if you don't want to create a socket and associate it with the

    // persistent reservation

    // Initialize Winsock

    iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);

    if (iResult != 0) {

        printf("WSAStartup failed with error = %d\n", iResult);

        // return 1;

    }

    sock = socket(iFamily, iType, iProtocol);

    if (sock == INVALID_SOCKET)

        printf("socket function failed with error = %d\n", WSAGetLastError());

    else {

        printf("socket function succeeded\n");

        iResult =

            WSAIoctl(sock, SIO_ASSOCIATE_PORT_RESERVATION, (LPVOID) & resToken,

                     sizeof (ULONG64), NULL, 0, &bytesReturned, NULL, NULL);

        if (iResult != 0) {

            printf

                ("WSAIoctl(SIO_ASSOCIATE_PORT_RESERVATION) failed with error = %d\n",

                 WSAGetLastError());

        } else {

            printf("WSAIoctl(SIO_ASSOCIATE_PORT_RESERVATION) succeeded, bytesReturned = %u\n",

                bytesReturned);

            service.sin_family = AF_INET;

            service.sin_addr.s_addr = INADDR_ANY;

            service.sin_port = 0;

            iResult = bind(sock, (SOCKADDR*) &service, sizeof(service) );

            if (iResult == SOCKET_ERROR)

                printf("bind failed with error = %d\n", WSAGetLastError());

            else {

                printf("bind succeeded\n");

                iResult = getsockname(sock, (SOCKADDR*) &sockName, &nameLen);

                if (iResult == SOCKET_ERROR)

                    printf("getsockname failed with error = %d\n", WSAGetLastError() );

                else {

                    printf("getsockname succeeded\n");

                    printf("Port number allocated = %u\n", ntohs(sockName.sin_port) );

                }

            }

        }

        if (sock != INVALID_SOCKET) {

            iResult = closesocket(sock);

            if (iResult == SOCKET_ERROR) {

                printf("closesocket failed with error = %d\n", WSAGetLastError());

                WSACleanup();

            }

        }

    }

    // comment out this block of code if you don't want to delete the reservation just created

    status = DeletePersistentTcpPortReservation((USHORT) startPortns, (USHORT) numPorts);

    if (status != NO_ERROR) {

        printf("DeletePersistentTcpPortReservation returned error: %ld\n", status);

        return 1;

    }

    printf("DeletePersistentTcpPortReservation call succeeded\n");

    return 0;

}