createpersistenttcpportreservation_test1
Posted by Anonymous on Mon 25th Mar 2024 14:40
raw | new post
modification of post by Anonymous (view diff)


/*
 *
 * $ 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");
    printf("  Token = %llx\n", (long long)resToken);

    // 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;
}