windows qos api使用示例

简介

本文给出C++版以及Rust版调用windows API实现QOS的示例,并提出一些注意事项。QOS可以用来区分socket 优先级、实现带宽控制等诸多功能。

UDP版本

该示例的重要部分是客户端应用程序。客户端代码的工作方式如下:

  1. 通过套接字连接到目标 IP 地址。

  2. 通过调用 QOSCreateHandle 初始化 QOS 子系统。

  3. 通过调用 QOSAddSocketToFlow 为套接字创建一个流。

  4. 通过调用 QOSQueryFlow 查询源主机和目标主机之间路径的特性。使用的两个估计值是:

  • 瓶颈带宽:最大理论吞吐量;

  • 可用带宽:新流量的当前可用吞吐量。

根据网络配置的不同:

A. 可能进行带宽估算。例如,目标可能在另一个子网上。如果是这样,客户端将仅进行流量整形并根据指定的流量类型标记优先级,但永远无法接收可用带宽的通知或估计,而出站流量将根据 DLNA 规范进行标记。

B. QOS 子系统可能没有足够的时间来估计到目的地主机的条件。如果发生这种情况,客户端将在重试之前休眠 1 秒。在生产应用程序中,您可以通过在 A/V 流量开始之前调用 QOSStartTrackingClient API 来最小化发生这种情况的可能性。建议在目标与客户端位于同一子网上时始终使用此 API。

  1. 如果可用带宽的估计值小于请求的带宽,则客户端将修订请求的带宽。

  2. 客户端为我们的流设置了一个固定速率,该速率将限制出站流量到请求的带宽。为此,它考虑了地址族和使用的协议(UDP)的线路开销。

  3. 客户端注册了拥塞通知。

  4. 使用形状机制作为控制循环以期望速率发送流量。应用程序使用 TransmitPackets API 调用发送数据包。客户端利用形状器仅在所有数据包离开主机时才完成此 Winsock 传输操作的事实,以确定一次发送多少个数据包。

  5. 当外部条件将网络推入拥塞状态时,将接收到拥塞通知。作为反应,客户端将将目标比特率降低到初始值的十分之一。此外,客户端将在可用带宽足以返回到先前的比特率时请求通知。

代码如下:

cpp 复制代码
/*++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

   THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
   ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
   THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
   PARTICULAR PURPOSE.

   Copyright (c) Microsoft Corporation. All rights reserved.

Module Name:

    qossample.c

Abstract:

    This program implements a simple traffic generator which uses the QOS2
    api to not overload the network path between the source and its destination.

    Although the application can be run as a receiver to process the traffic
    generated, the client side of the application is the important part of this
    sample. This code is in the Client function.

    The code for the client works as follows:

    1) A socket is connected to the destination IP address
    2) The QOS subsystem is initialized by a call to QOSCreateHandle
    3) A flow is created for the socket by a call to QOSAddSocketToFlow
    4) The characteristics of the path between the source host and destination
       host are queried through QOSQueryFlow. The two estimates used are:

        - the bottleneck bandwidth: the maximum theoretical throughput;
        - and the available bandwidth: the currently available throughput for
          new traffic

       Depending on your network configuration:

       A) It may be impossible to run network experiments to the destination.
          For example, the destination might be on another subnet. If so, the
          client will only shape traffic and mark priority with DSCP based on
          the specified traffic type, but will never be able to receive
          notifications or estimates of available bandwidth and outgoing traffic
          would be marked per the DLNA specification.
       B) The QOS subsystem may not have had enough time to estimate conditions
          to the destination host. If this happens, the client will sleep for 1s
          before retrying. In a production application you could minize the
          likelihood of this happening by calling, before the start of your
          A/V traffic, the QOSStartTrackingClient API. It is suggested that this
          API always be used when the destination is on the same subnet as the
          client.

    5) If the estimate of available bandwidth is less than the requested
       bandwidth, the client revises the requested bandwidth.
    6) The client sets a shaping rate for our flow that will limit the outgoing
       traffic to the requested bandwidth. To do so, it accounts for the wire
       overhead given both the address family as well as the protocol used, UDP.
    7) The client registers for notifications of congestion.
    8) Traffic is sent at the desired rate using the shaping mechanism as the
       control loop. The application uses the API call TransmitPackets to
       send the packet bunches. The client uses the fact that the shaper will
       only complete this Winsock transmit operation when all packets have left
       the host to determine how many packets to send at a time.
    9) A congestion notification will be received whenever external conditions
       push the network into a congested state. The client, as a reaction, will
       lower the targeted bit rate to one-tenth of the initial value. Moreover,
       the client will request notification when the bandwidth available is
       enough to return to the previous bit rate.

Environment:

    User-Mode only

Revision History:

------------------------------------------------------------------------------*/

#include <winsock2.h>
#include <mswsock.h>
#include <stdlib.h>
#include <stdio.h>
#include <qos2.h>
#pragma comment(lib, "ws2_32.lib")
#pragma comment(lib, "QWave.lib")
#pragma warning(disable:4127)   // condition expression is constant

//******************************************************************************
// Forward declarations
//******************************************************************************
VOID
client(
    __in                int     argc,
    __in_ecount(argc)   char* argv[]
);

VOID
server();

VOID
help();

//******************************************************************************
// Global defines
//******************************************************************************

// Size of the data part of each datagram that is exchanged between the
// client and the server. This does not count the IP and UDP header.
#define DATAGRAM_SIZE       1400

// Number of bursts we are aiming for per second
#define BURSTS_PER_SECOND   40

// Port to be used for communication, 
// 40007 in hex in network byte order 
#define PORT                0x479c

//******************************************************************************
// Global variables
//******************************************************************************

// Array used by both the client routine and the server routine to
// send and receive data
CHAR                dataBuffer[DATAGRAM_SIZE];

// Version of the QOS subsystem we wish to use
QOS_VERSION QosVersion = { 1 , 0 };

//******************************************************************************
// Routine: 
//      main
//
// Description:
//      Entry point. Verifies that the number of parameters is enough to 
//      ascertain whether this instance is to be a client or server.
// 
//******************************************************************************
int
_cdecl
main(
    __in                int argc,
    __in_ecount(argc)   char* argv[]
)
{
    // Verify the number of command line arguments
    if (argc < 2) {
        help();
        exit(0);
    }

    if (strcmp(argv[1], "-s") == 0) {
        // If the call was "qossample -s" run as a server
        server();
    }
    else if (strcmp(argv[1], "-c") == 0) {
        // If the call was "qossample -c ..." run as a client
        client(argc, argv);
    }

    // Else, run the help routine
    help();
    return 1;
}

//******************************************************************************
// Routine: 
//      socketCreate
//
// Description:
//      This routine prepares the socket for the client instance. To do so,
//      it converts the address parameter from string to IP address. Then
//      it creates a UDP socket which it connects to this destination.
//
//      Since we will use TransmitPackets for our send operations, the function
//      pointer is obtained from Winsock.
// 
//******************************************************************************
VOID
socketCreate(
    __in    PCHAR                       destination,
    __out   SOCKET* socket,
    __out   ADDRESS_FAMILY* addressFamily,
    __out   LPFN_TRANSMITPACKETS* transmitPackets
) {
    // Return value of WSAStartup
    WSADATA             wsaData;

    // Temporarily used to represent the destination IP address
    SOCKADDR_STORAGE    destAddr;

    // Result of the various Winsock API calls
    INT                 returnValue;

    // Used by WSAStringToAddressA
    INT                 sockaddrLen;

    // Used by WSAIoctl
    DWORD               bytesReturned;

    // GUID of the TransmitPacket Winsock2 function which we will
    // use to send the traffic at the client side.
    GUID TransmitPacketsGuid = WSAID_TRANSMITPACKETS;

    // Start Winsock
    returnValue = WSAStartup(MAKEWORD(2, 2), &wsaData);

    if (returnValue != 0) {
        printf("%s:%d - WSAStartup failed (%d)\n",
            __FILE__, __LINE__, returnValue);
        exit(1);
    }

    // First attempt to convert the string to an IPv4 address
    sockaddrLen = sizeof(destAddr);
    destAddr.ss_family = AF_INET;
    returnValue = WSAStringToAddressA(destination,
        AF_INET,
        NULL,
        (LPSOCKADDR)&destAddr,
        &sockaddrLen);

    if (returnValue != ERROR_SUCCESS) {
        // If this fails,
        // Attempt to convert the string to an IPv6 address
        sockaddrLen = sizeof(destAddr);
        destAddr.ss_family = AF_INET6;
        returnValue = WSAStringToAddressA(destination,
            AF_INET6,
            NULL,
            (LPSOCKADDR)&destAddr,
            &sockaddrLen);

        if (returnValue != ERROR_SUCCESS) {
            // If this again fails, exit the application
            // But print out the help information first.
            printf("%s:%d - WSAStringToAddressA failed (%d)\n",
                __FILE__, __LINE__, WSAGetLastError());
            help();
            exit(1);
        }
    }

    // Set the destination port.
    SS_PORT((PSOCKADDR)&destAddr) = PORT;

    // Copy the address family back to caller
    *addressFamily = destAddr.ss_family;

    // Create a UDP socket
    *socket = WSASocket(destAddr.ss_family,
        SOCK_DGRAM,
        0,
        NULL,
        0,
        WSA_FLAG_OVERLAPPED);

    if (*socket == INVALID_SOCKET) {
        printf("%s:%d - WSASocket failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Connect the new socket to the destination
    returnValue = WSAConnect(*socket,
        (PSOCKADDR)&destAddr,
        sizeof(destAddr),
        NULL,
        NULL,
        NULL,
        NULL);

    if (returnValue != NO_ERROR) {
        printf("%s:%d - WSAConnect failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Query the function pointer for the TransmitPacket function
    returnValue = WSAIoctl(*socket,
        SIO_GET_EXTENSION_FUNCTION_POINTER,
        &TransmitPacketsGuid,
        sizeof(GUID),
        transmitPackets,
        sizeof(PVOID),
        &bytesReturned,
        NULL,
        NULL);

    if (returnValue == SOCKET_ERROR) {
        printf("%s:%d - WSAIoctl failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }
}

//******************************************************************************
// Routine: 
//      client
//
// Description:
//      This routine creates a datagram socket which it connects to the remote
//      IP address. This socket is then added to a QOS flow. The application
//      uses the flow to rate control its outgoing traffic. Packets on this flow
//      will be prioritized if the network path between the client and receiver
//      support prioritization. Specifically, if the receiver is:
//
//      A) On-link (same subnet), both 802.1Q and DSCP are applied subject to
//         available bandwidth and network support.
//      B) Off-link (different subnet), only DSCP is applied regardless of
//         available bandwidth.
//
//      Moreover, the application queries the characteristics of the network 
//      path for the socket. If estimates are available, the application:
//
//      1) may readjust its throughput so as to not cause congestion on the 
//         network.
//      2) will be notified when the network enters congestion. As a result,
//         it will lower it's throughput to 1/10 the targeted rate.
//      3) will be notified when the network is no longer congested and enough
//         bandwidth is available for the application to return to its targeted
//         rate.
//
//******************************************************************************
VOID
client(
    __in                int     argc,
    __in_ecount(argc)   char* argv[]
) {
    // Address family of the datagram socket
    ADDRESS_FAMILY              addressFamily;

    // Socket for our traffic experiment
    SOCKET                      socket;

    // Function pointer to the TransmitPacket Winsock2 function
    LPFN_TRANSMITPACKETS        transmitPacketsFn;
    // Array of transmit elements
    LPTRANSMIT_PACKETS_ELEMENT  transmitEl;

    // Target packet rate and bit rate this application will aim to send
    // If there is no congestion
    ULONG                       targetPacketRate, targetBitRate;
    // Current rate at which the application is sending. If the network is
    // congested, this will be less than the target rate
    ULONG                       currentPacketRate, currentBitRate;
    // Counters for the achieved packet rate and achieved bit rate
    ULONG                       achievedPacketRate, achievedBitRate;

    // Timer; used to periodically output to the screen our counters
    HANDLE                      timerEvent;
    LARGE_INTEGER               timerAwakenTime;

    // Handle to the QOS subsystem
    // Returned by QOSCreateHandle
    HANDLE                      qosHandle;
    // ID of the QOS flow we will create for the socket. 
    // Returned by QOSAddSocketToFlow
    QOS_FLOWID                  flowID;
    // Result of QOSQueryFlow
    QOS_FLOW_FUNDAMENTALS       flowFund;
    // Parameter to QOSSetFlow
    QOS_FLOWRATE_OUTGOING       flowRate;
    // If TRUE, the QOS subsystem is running network experiments on the
    // network path to the destination. If false, estimates are not available.
    // The flow is still marked and shaped.
    BOOL                        estimatesAvailable;
    // Result of the QOS API calls.
    BOOL                        result;

    // Overlapped operation used for TransmitPackets
    WSAOVERLAPPED               sendOverlapped;
    // Overlapped operation used for QOSNotifyFlow to be notified
    // of network congestions.
    OVERLAPPED                  congestionOverlapped;
    // Overlapped operation used for QOSNotifyFlow to be notified when, 
    // after congestion, there is enough bandwidth for the target rate
    OVERLAPPED                  availableOverlapped;
    // TRUE if the network is currently congested
    BOOL                        congestion;

    ULONG                       temp;

    // Verify the number of command line arguments
    if (argc != 4) {
        help();
        exit(1);
    }

    // Identify what destination IP address we're trying to talk to and
    // connect a UDP socket to it
    socketCreate(argv[2],
        &socket,
        &addressFamily,
        &transmitPacketsFn);

    printf("socket resut %d\n", socket);

    if (FALSE == QOSCreateHandle(&QosVersion, &qosHandle)) {
        printf("%s:%d - QOSCreateHandle failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // The Flow ID parameter MUST be set to 0 as input to QOSAddSocketToFlow
    flowID = 0;

    // Create a flow for our socket
    result = QOSAddSocketToFlow(qosHandle,
        socket,
        NULL,
        QOSTrafficTypeVoice,
        QOS_NON_ADAPTIVE_FLOW,
        &flowID);

    if (result == FALSE) {
        printf("%s:%d - QOSAddSocketToFlow failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Read the data rate in bits/s passed on the command line
    targetBitRate = atol(argv[3]);

    if (targetBitRate == 0) {
        help();
        exit(1);
    }

    // Convert from bits to bytes
    targetBitRate /= 8;

    // Calculate how many packets we need; round up.
    targetPacketRate = targetBitRate / ARRAYSIZE(dataBuffer);
    if (targetBitRate % ARRAYSIZE(dataBuffer) != 0)
        targetPacketRate++;

    // Calculate the number of packets per bursts; round up
    if (targetPacketRate % BURSTS_PER_SECOND != 0) {
        targetPacketRate /= BURSTS_PER_SECOND;
        targetPacketRate++;
    }
    else {
        targetPacketRate /= BURSTS_PER_SECOND;
    }

    // Calculate the final bit rate, targetBitRate, in bps that the application
    // will send.
    targetBitRate = BURSTS_PER_SECOND
        * targetPacketRate
        * ARRAYSIZE(dataBuffer)
        * 8;

    //
    // Allocate an array of transmit elements big enough to send 
    // targetPacketRate packets every burst.
    transmitEl = static_cast<LPTRANSMIT_PACKETS_ELEMENT>( HeapAlloc(GetProcessHeap(),
        HEAP_ZERO_MEMORY,
        sizeof(TRANSMIT_PACKETS_ELEMENT)
        * targetPacketRate));

    if (transmitEl == NULL) {
        printf("%s:%d - HeapAlloc failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }


    // For the purpose of this application, we only use one data buffer for all 
    // of our packets.
    ZeroMemory(&dataBuffer, sizeof(dataBuffer));

    //
    // Initialize each of these transmit element to point to the same zeroed out
    // data buffer
    for (temp = 0; temp < targetPacketRate; temp++) {
        transmitEl[temp].dwElFlags = TP_ELEMENT_MEMORY | TP_ELEMENT_EOP;
        transmitEl[temp].pBuffer = dataBuffer;
        transmitEl[temp].cLength = sizeof(dataBuffer);
    }

    // Print out what we'll be doing
    printf("----------------------------------"
        "----------------------------------\n"
        "This instance of the QOS sample program is aiming to:\n"
        "\t - Send %d bits of UDP traffic per second\n"
        "\t - In packets of %d bytes\n"
        "\t - In bursts of %d packets every %d milliseconds\n\n"
        "----------------------------------"
        "----------------------------------\n",
        targetBitRate,
        ARRAYSIZE(dataBuffer),
        targetPacketRate,
        1000 / BURSTS_PER_SECOND);

    // Assume, by default, that estimates are not available
    estimatesAvailable = FALSE;

    printf("Querying fundamentals about the network path: ");
    do {
        temp = sizeof(flowFund);

        // Query the flow fundamentals for the path to the destination.
        // This will return estimates of the bottleneck bandwidth, available
        // bandwidth and average RTT.
        result = QOSQueryFlow(qosHandle,
            flowID,
            QOSQueryFlowFundamentals,
            &temp,
            &flowFund,
            0,
            NULL);

        if (result == FALSE) {
            DWORD lastError;

            lastError = GetLastError();

            if (lastError == ERROR_NO_DATA) {
                // If the call failed, this could be because the QOS subsystem 
                // has not yet gathered enough data for estimates. If so, the
                // QOS2 api returns ERROR_NO_DATA.
                printf(".");
                Sleep(1000);
            }
            else if (lastError == ERROR_NOT_SUPPORTED)
            {
                // If the call failed, this could be because the QOS subsystem 
                // cannot run network experiments on the network path to the
                // destination. If so, the API returns ERROR_NOT_SUPPORTED.
                printf("NOT SUPPORTED\n"
                    "\t   Network conditions to this destination could not\n"
                    "\t   be detected as your network configuration is not\n"
                    "\t   supported for network experiments\n");
                break;
            }
            else {
                printf("%s:%d - QOSQueryFlow failed (%d)\n",
                    __FILE__, __LINE__, lastError);
                exit(1);
            }
        }
        else if ((flowFund.BottleneckBandwidthSet == FALSE)
            || (flowFund.AvailableBandwidthSet == FALSE)) {
            // If the call succeeded but bottleneck bandwidth or 
            // available bandwidth are not known then estimates are still
            // processing; query again in 1 second.
            printf(".");
            Sleep(1000);
        }
        else {
            // Estimate where available.
            double bottleneck;
            double available;

            estimatesAvailable = TRUE;

            // Convert the bottleneck bandwidth from bps to mbps
            bottleneck = (double)flowFund.BottleneckBandwidth;
            bottleneck /= 1000000.0;

            // Convert available bandwidth from bps to mbps
            available = (double)flowFund.AvailableBandwidth;
            available /= 1000000.0;

            printf("DONE\n"
                "\t - Bottleneck bandwidth is approximately %4.2f Mbps\n"
                "\t - Available bandwidth is approximately  %4.2f Mbps\n",
                bottleneck,
                available);

            break;
        }
    } while (TRUE);

    if (estimatesAvailable == TRUE) {
        UINT64 targetBitRateWithHeaders;

        printf("\nNOTE: the accuracy of the QOS estimates can be impacted by\n"
            "any of the following,\n\n"
            "\t - Both the network interface at the client\n"
            "\t   and at the server must be using NDIS 6 drivers.\n"
            "\t - If the server is not a Windows Vista host, verify that \n"
            "\t   it implements the LLD2 networking protocol. You can\n"
            "\t   find more information at http://www.microsoft.com.\n"
            "\t - IPSec, VPNs and enterprise class networking equipment\n"
            "\t   may interfere with network experiments.\n\n");

        // Calculate what our target bit rate, with protocol headers for 
        // IP(v4/v6) and UDP will be.
        targetBitRateWithHeaders = QOS_ADD_OVERHEAD(addressFamily,
            IPPROTO_UDP,
            ARRAYSIZE(dataBuffer),
            targetBitRate);

        if (flowFund.AvailableBandwidth < targetBitRateWithHeaders) {
            // If the estimate of available bandwidth is not sufficient for the
            // target bit rate (with headers), drop to a lesser throughput
            UINT64 availableBandwidth;

            // The estimate returned does not account for headers
            // Remove the estimated overhead for our address family and UDP.
            availableBandwidth = QOS_SUBTRACT_OVERHEAD(
                addressFamily,
                IPPROTO_UDP,
                ARRAYSIZE(dataBuffer),
                flowFund.AvailableBandwidth);

            // Calculate the rate of packets we can realistically send
            targetPacketRate = (LONG)availableBandwidth / 8;
            targetPacketRate /= ARRAYSIZE(dataBuffer);
            targetPacketRate /= BURSTS_PER_SECOND;

            // Calculate the real bit rate we'll be using
            targetBitRate = BURSTS_PER_SECOND
                * targetPacketRate
                * ARRAYSIZE(dataBuffer)
                * 8;

            printf("Not enough available bandwidth for the requested bitrate.\n"
                "Downgrading to lesser bitrate - %d.\n", targetBitRate);
        }
    }

    // Our starting rate is this target bit rate
    currentBitRate = targetBitRate;
    currentPacketRate = targetPacketRate;

    // Ask the QOS subsystem to shape our traffic to this bit rate. Note that
    // the application needs to account for the size of the IP(v4/v6) 
    // and UDP headers.

    // Calculate the real bandwidth we will need to be shaped to.
    flowRate.Bandwidth = QOS_ADD_OVERHEAD(addressFamily,
        IPPROTO_UDP,
        ARRAYSIZE(dataBuffer),
        currentBitRate);

    // Set shaping characteristics on our QOS flow to smooth out our bursty
    // traffic.
    flowRate.ShapingBehavior = QOSShapeAndMark;

    // The reason field is not applicable for the initial call.
    flowRate.Reason = QOSFlowRateNotApplicable;
    result = QOSSetFlow(qosHandle,
        flowID,
        QOSSetOutgoingRate,
        sizeof(flowRate),
        &flowRate,
        0,
        NULL);

    if (result == FALSE) {
        printf("%s:%d - QOSSetFlow failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Allocate a waitable timer. We will use this timer to periodically
    // awaken and output statistics.
    timerEvent = CreateWaitableTimer(NULL, FALSE, NULL);

    if (timerEvent == NULL) {
        printf("%s:%d - CreateWaitableTimer failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    //
    // Set the sampling time to 1 second
    timerAwakenTime.QuadPart = -10000000; // 1 second

    if (FALSE == SetWaitableTimer(timerEvent,
        &timerAwakenTime,
        1000,
        NULL,
        NULL,
        FALSE)) {

        printf("%s:%d - SetWaitableTimer failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Prepare the support overlapped structures to detect congestion,
    // notifications of bandwidth change and the completion of our send 
    // routines.
    ZeroMemory(&congestionOverlapped, sizeof(congestionOverlapped));
    congestionOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (congestionOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    ZeroMemory(&availableOverlapped, sizeof(availableOverlapped));
    availableOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (availableOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    ZeroMemory(&sendOverlapped, sizeof(sendOverlapped));
    sendOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (sendOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    if (estimatesAvailable == TRUE) {
        // If estimates are available, we request a notification 
        // for congestion. This notification will complete whenever network
        // congestion is detected.
        result = QOSNotifyFlow(qosHandle,
            flowID,
            QOSNotifyCongested,
            NULL,
            NULL,
            0,
            &congestionOverlapped);

        if (result == FALSE) {
            DWORD lastError;

            lastError = GetLastError();
            if (lastError != ERROR_IO_PENDING) {
                printf("%s:%d - QOSNotifyFlow failed (%d)\n",
                    __FILE__, __LINE__, lastError);
                exit(1);
            }
        }
    }

    printf("----------------------------------"
        "----------------------------------\n"
        " # packets | # of bits  |  Bottleneck "
        "|  Available |  Congestion   \n"
        "----------------------------------"
        "----------------------------------\n");

    // Initialize our counters to 0
    achievedBitRate = achievedPacketRate = 0;
    congestion = FALSE;

    do {
        BOOL sendNextGroup;

        // Send the first burst of packets
        result = (*transmitPacketsFn)(socket,
            transmitEl,
            currentPacketRate,
            0xFFFFFFFF,
            &sendOverlapped,
            TF_USE_KERNEL_APC);

        if (result == FALSE) {
            DWORD lastError;

            lastError = WSAGetLastError();
            if (lastError != ERROR_IO_PENDING) {
                printf("%s:%d - TransmitPackets failed (%d)\n",
                    __FILE__, __LINE__, GetLastError());
                exit(1);
            }
        }

        // Increase the counter of sent data
        achievedPacketRate += currentPacketRate;
        achievedBitRate += currentPacketRate * ARRAYSIZE(dataBuffer) * 8;
        sendNextGroup = FALSE;

        do {
            HANDLE          waitEvents[3];
            DWORD           waitResult;

            // Wait for any of the 3 events to complete

            // The 1 second periodic timer
            waitEvents[0] = timerEvent;

            if (congestion)
                // Notification of available bandwidth
                waitEvents[1] = availableOverlapped.hEvent;
            else
                // Notification of congestion
                waitEvents[1] = congestionOverlapped.hEvent;

            // Notification that the send operation has completed
            waitEvents[2] = sendOverlapped.hEvent;

            waitResult = WaitForMultipleObjects(ARRAYSIZE(waitEvents),
                (PHANDLE)waitEvents,
                FALSE,
                INFINITE);

            switch (waitResult) {
            case WAIT_OBJECT_0: {
                // The event for the periodic timer is set

                printf("%10d | ", achievedPacketRate);
                printf("%10d | ", achievedBitRate);

                if (estimatesAvailable) {
                    // If estimates are available for the network path
                    // query for estimates and output to the console along
                    // with our counters.

                    // Ascertained through QOSQueryFlow
                    BOOL estimateIndicatesCongestion;

                    // Default is unknown
                    estimateIndicatesCongestion = FALSE;

                    ZeroMemory(&flowFund, sizeof(flowFund));

                    temp = sizeof(flowFund);
                    QOSQueryFlow(qosHandle,
                        flowID,
                        QOSQueryFlowFundamentals,
                        &temp,
                        &flowFund,
                        0,
                        NULL);

                    if (flowFund.BottleneckBandwidthSet)
                        printf("%10d  | ", flowFund.BottleneckBandwidth);
                    else
                        printf("   NO DATA  | ");

                    if (flowFund.AvailableBandwidthSet) {
                        if (flowFund.AvailableBandwidth == 0)
                            estimateIndicatesCongestion = TRUE;

                        printf("%10d | ", flowFund.AvailableBandwidth);
                    }
                    else
                        printf("   NO DATA | ");

                    if (estimateIndicatesCongestion)
                        printf(" CONGESTION\n");
                    else
                        printf("\n");
                }
                else {
                    // Bandwidth estimates are not available
                    printf("    N/A     |     N/A    |\n");
                }

                //
                // Reset the counters
                achievedPacketRate = achievedBitRate = 0;
                break;
            }
            case WAIT_OBJECT_0 + 1: {
                // This is either a notification for congestion or 
                // for bandwidth available 

                if (congestion == FALSE) {
                    UINT64 targetBandwidthWithOverhead;
                    ULONG  bufferSize;
                    //
                    // Congestion
                    //
                    printf("----------------------------------"
                        "----------------------------------\n"
                        "CONGESTION\n"
                        "----------------------------------"
                        "----------------------------------\n");

                    //
                    // Reduce the current rate to one-tenth of the initial rate
                    // Insure you're always sending at least 1 packet per
                    // burst.
                    if (currentPacketRate < 10)
                        currentPacketRate = 1;
                    else
                        currentPacketRate /= 10;

                    // Calculate the new bit rate we'll be using
                    currentBitRate = BURSTS_PER_SECOND
                        * currentPacketRate
                        * ARRAYSIZE(dataBuffer)
                        * 8;


                    // Update the shaping characteristics on our QOS flow 
                    // to smooth out our bursty traffic.
                    flowRate.Bandwidth = QOS_ADD_OVERHEAD(
                        addressFamily,
                        IPPROTO_UDP,
                        ARRAYSIZE(dataBuffer),
                        currentBitRate);

                    flowRate.ShapingBehavior = QOSShapeAndMark;
                    flowRate.Reason = QOSFlowRateCongestion;

                    result = QOSSetFlow(qosHandle,
                        flowID,
                        QOSSetOutgoingRate,
                        sizeof(flowRate),
                        &flowRate,
                        0,
                        NULL);

                    if (result == FALSE) {
                        printf("%s:%d - QOSSetFlow failed (%d)\n",
                            __FILE__, __LINE__, GetLastError());
                        exit(1);
                    }

                    // Request a notification for when there is enough bandwidth
                    // to return to our previous targeted bit rate.
                    // This will complete only when the network is no longer
                    // congested and bandwidth is available.
                    targetBandwidthWithOverhead = QOS_ADD_OVERHEAD(
                        addressFamily,
                        IPPROTO_UDP,
                        ARRAYSIZE(dataBuffer),
                        targetBitRate);

                    bufferSize = sizeof(targetBandwidthWithOverhead);

                    result = QOSNotifyFlow(qosHandle,
                        flowID,
                        QOSNotifyAvailable,
                        &bufferSize,
                        (PVOID)&targetBandwidthWithOverhead,
                        0,
                        &availableOverlapped);

                    if (result == FALSE) {
                        DWORD lastError;

                        lastError = GetLastError();
                        if (lastError != ERROR_IO_PENDING) {
                            printf("%s:%d - QOSNotifyFlow failed (%d)\n",
                                __FILE__, __LINE__, lastError);
                            exit(1);
                        }
                    }

                    congestion = TRUE;
                }
                else {
                    //
                    // End of congestion
                    //
                    printf("----------------------------------"
                        "----------------------------------\n"
                        "END OF CONGESTION\n"
                        "----------------------------------"
                        "----------------------------------\n");

                    //
                    // Reset the current packet rate to the initial target rate
                    currentPacketRate = targetPacketRate;

                    // Reset the current bit rate to the initial target rate
                    currentBitRate = targetBitRate;


                    // Update the shaping characteristics on our QOS flow 
                    // to smooth out our bursty traffic.
                    flowRate.Bandwidth = QOS_ADD_OVERHEAD(addressFamily,
                        IPPROTO_UDP,
                        ARRAYSIZE(dataBuffer),
                        targetBitRate);

                    flowRate.ShapingBehavior = QOSShapeAndMark;
                    flowRate.Reason = QOSFlowRateCongestion;
                    result = QOSSetFlow(qosHandle,
                        flowID,
                        QOSSetOutgoingRate,
                        sizeof(flowRate),
                        &flowRate,
                        0,
                        NULL);

                    if (result == FALSE) {
                        printf("%s:%d - QOSSetFlow failed (%d)\n",
                            __FILE__, __LINE__, GetLastError());
                        exit(1);
                    }

                    // Request a notification for the next network congestion
                    result = QOSNotifyFlow(qosHandle,
                        flowID,
                        QOSNotifyCongested,
                        NULL,
                        NULL,
                        0,
                        &congestionOverlapped);

                    if (result == FALSE) {
                        DWORD lastError;

                        lastError = GetLastError();
                        if (lastError != ERROR_IO_PENDING) {
                            printf("%s:%d - QOSNotifyFlow failed (%d)\n",
                                __FILE__, __LINE__, lastError);
                            exit(1);
                        }
                    }

                    congestion = FALSE;
                }
                break;
            }
            case WAIT_OBJECT_0 + 2: {
                // The transmit packet has completed its send, 
                // If it did so successfully, it's time to send the next 
                // burst of packets.
                BOOL    overlappedResult;
                DWORD   ignoredNumOfBytesSent;
                DWORD   ignoredFlags;

                overlappedResult = WSAGetOverlappedResult(
                    socket,
                    &sendOverlapped,
                    &ignoredNumOfBytesSent,
                    FALSE,
                    &ignoredFlags);

                if (overlappedResult == FALSE) {
                    printf("%s:%d - TransmitPackets failed (%d)\n",
                        __FILE__, __LINE__, WSAGetLastError());
                    exit(1);
                }

                // Time to send out the next bunch of packets
                sendNextGroup = TRUE;
                break;
            }
            default:
                // The wait call failed.
                printf("%s:%d - WaitForMultipleObjects failed (%d)\n",
                    __FILE__, __LINE__, GetLastError());
                exit(1);
            }

        } while (sendNextGroup == FALSE);
    } while (TRUE);
}

//******************************************************************************
// Routine: 
//      server
//
// Description:
//      This routine creates a socket through which it will receive
//      any datagram sent by the client. It counts the number of packet 
//      and the number of bytes received. Periodically, it outputs this 
//      information to the console.
//
//******************************************************************************
VOID
server()
{
    HANDLE              timerEvent;
    LARGE_INTEGER       timerAwakenTime;

    // Return value of WSAStartup
    WSADATA             wsaData;

    // The socket used to receive data
    SOCKET              socket;

    // IPv6 wildcard address and port number 40007 to listen on at the server
    SOCKADDR_IN6 IPv6ListeningAddress = { AF_INET6,
                                            PORT,
                                            0,
                                            IN6ADDR_ANY_INIT,
                                            0 };

    // Result value from the various API calls
    DWORD               result;

    // Used to specify an option to setsocktopt
    ULONG               optionValue;

    // Overlapped structure used to post asynchronous receive calls
    WSAOVERLAPPED       recvOverlapped;

    // Counters of the number of bytes and packets received over 
    // a period of time.
    DWORD               numPackets, numBytes;



    // Initialize Winsock
    result = WSAStartup(MAKEWORD(2, 2), &wsaData);

    if (result != 0) {
        printf("%s:%d - WSAStartup failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Create an IPv6 datagram socket
    socket = WSASocket(AF_INET6,
        SOCK_DGRAM,
        0,
        NULL,
        0,
        WSA_FLAG_OVERLAPPED);

    if (socket == INVALID_SOCKET) {
        printf("%s:%d - WSASocket failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Set IPV6_V6ONLY to FALSE before the bind operation
    // This will permit us to receive both IPv6 and IPv4 traffic on the socket.
    optionValue = FALSE;
    result = setsockopt(socket,
        IPPROTO_IPV6,
        IPV6_V6ONLY,
        (PCHAR)&optionValue,
        sizeof(optionValue));

    if (SOCKET_ERROR == result) {
        printf("%s:%d - setsockopt failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Bind the socket
    result = bind(socket,
        (PSOCKADDR)&IPv6ListeningAddress,
        sizeof(IPv6ListeningAddress));

    if (result != NO_ERROR) {
        printf("%s:%d - bind failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Create an event to be used for the overlapped of our
    // receive operations. The event is initialized to FALSE and set
    // to auto-reset.
    recvOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (recvOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Create a timer event on which we will be able to wait.
    // We wish to be awaken every second to print out the count of packets
    // and number of bytes received.
    timerEvent = CreateWaitableTimer(NULL, FALSE, NULL);
    if (timerEvent == NULL) {
        printf("%s:%d - CreateWaitableTimer failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Awaken first in 1 second
    timerAwakenTime.QuadPart = -10000000; // 1 second

    if (FALSE == SetWaitableTimer(timerEvent,
        &timerAwakenTime,
        1000, // Awaken every second
        NULL,
        NULL,
        FALSE)) {

        printf("%s:%d - SetWaitableTimer failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Initialize the counters to 0
    numPackets = 0;
    numBytes = 0;

    printf("-------------------------\n"
        " # packets |  # of bits |\n"
        "-------------------------\n");
    do {
        // Array of events for WaitForMultipleObjects
        HANDLE  waitEvents[2];

        // Used for WSARecv
        DWORD   numberOfBytesReceived;

        // Used for WSARecv
        DWORD   dwFlags;

        // Used for WSARecv
        WSABUF  buf;

        // The buffer is always the same global array
        buf.len = sizeof(dataBuffer);
        buf.buf = (PCHAR)dataBuffer;

        // No flags.
        dwFlags = 0;

        // Post a receive operation
        // We only have one receive outstanding at a time. 
        result = WSARecv(socket,
            &buf,
            1,
            &numberOfBytesReceived,
            &dwFlags,
            &recvOverlapped,
            NULL);

        if (result != 0) {
            // The receive call failed. This could be because the
            // call will be completed asynchronously (WSA_IO_PENDING) or
            // it could be a legitimate error
            DWORD errorCode;

            errorCode = WSAGetLastError();

            if (errorCode != WSA_IO_PENDING) {
                printf("%s:%d - WSARecv failed (%d)\n",
                    __FILE__, __LINE__, errorCode);
                exit(1);
            }

            // If the error was WSA_IO_PENDING the call will be completed
            // asynchronously.
        }

        // Prepare our array of events to wait on. We will wait on:
        // 1) The event from the receive operation
        // 2) The event for the timer
        waitEvents[0] = recvOverlapped.hEvent;
        waitEvents[1] = timerEvent;

        // We wait for either event to complete
        result = WaitForMultipleObjects(ARRAYSIZE(waitEvents),
            (PHANDLE)waitEvents,
            FALSE,
            INFINITE);

        switch (result) {
        case WAIT_OBJECT_0: {
            // The receive operation completed.
            // Determine the true result of the receive call.
            BOOL overlappedResult;

            overlappedResult = WSAGetOverlappedResult(socket,
                &recvOverlapped,
                &numberOfBytesReceived,
                FALSE,
                &dwFlags);

            if (overlappedResult == FALSE) {

                // The receive call failed.

                printf("%s:%d - WSARecv failed (%d)\n",
                    __FILE__, __LINE__, WSAGetLastError());
                exit(1);
            }

            // The receive call succeeded
            // Increase our counters and post a new receive.
            numPackets++;
            numBytes += numberOfBytesReceived;
            break;
        }
        case WAIT_OBJECT_0 + 1: {
            // The timer event fired: our 1 second period has gone by.
            // Print the average to the console

            printf("%10d | %10d |\n", numPackets, numBytes * 8);

            // Reset the counters
            numPackets = numBytes = 0;
            break;
        }
        default:
            // The wait call failed.
            printf("%s:%d - WaitForMultipleObjects failed (%d)\n",
                __FILE__, __LINE__, GetLastError());
            exit(1);
        }

        // We continue this loop until the process is forceably stopped
        // through Ctrl-C.
    } while (TRUE);

}

//******************************************************************************
// Routine: 
//      help
//
// Description:
//      This routine prints out the usage information for the program
//
//******************************************************************************
VOID
help()
{
    printf("USAGE:\n"
        "\tSERVER: qossample -s\n"
        "\tCLIENT: qossample -c IP-ADDRESS BIT-RATE\n\n"
        "\tIn the following example, the application would try to send\n"
        "\t20 Mb of traffic to the host at 10.0.0.1:\n"
        "\tqossample -c 10.0.0.1 20000000\n");
    return;
}

我测试时客户端和服务端是在不同的子网下,所以不支持带宽估计。运行结果如下:

客户端:

注意事项:

TCP版本

TCP代码:

流程和UDP一样,代码中的注释可以忽略。

cpp 复制代码
/*++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

   THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
   ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
   THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
   PARTICULAR PURPOSE.

   Copyright (c) Microsoft Corporation. All rights reserved.

Module Name:

    qossample.c

Abstract:

    This program implements a simple traffic generator which uses the QOS2
    api to not overload the network path between the source and its destination.

    Although the application can be run as a receiver to process the traffic
    generated, the client side of the application is the important part of this
    sample. This code is in the Client function.

    The code for the client works as follows:

    1) A socket is connected to the destination IP address
    2) The QOS subsystem is initialized by a call to QOSCreateHandle
    3) A flow is created for the socket by a call to QOSAddSocketToFlow
    4) The characteristics of the path between the source host and destination
       host are queried through QOSQueryFlow. The two estimates used are:

        - the bottleneck bandwidth: the maximum theoretical throughput;
        - and the available bandwidth: the currently available throughput for
          new traffic

       Depending on your network configuration:

       A) It may be impossible to run network experiments to the destination.
          For example, the destination might be on another subnet. If so, the
          client will only shape traffic and mark priority with DSCP based on
          the specified traffic type, but will never be able to receive
          notifications or estimates of available bandwidth and outgoing traffic
          would be marked per the DLNA specification.
       B) The QOS subsystem may not have had enough time to estimate conditions
          to the destination host. If this happens, the client will sleep for 1s
          before retrying. In a production application you could minize the
          likelihood of this happening by calling, before the start of your
          A/V traffic, the QOSStartTrackingClient API. It is suggested that this
          API always be used when the destination is on the same subnet as the
          client.

    5) If the estimate of available bandwidth is less than the requested
       bandwidth, the client revises the requested bandwidth.
    6) The client sets a shaping rate for our flow that will limit the outgoing
       traffic to the requested bandwidth. To do so, it accounts for the wire
       overhead given both the address family as well as the protocol used, UDP.
    7) The client registers for notifications of congestion.
    8) Traffic is sent at the desired rate using the shaping mechanism as the
       control loop. The application uses the API call TransmitPackets to
       send the packet bunches. The client uses the fact that the shaper will
       only complete this Winsock transmit operation when all packets have left
       the host to determine how many packets to send at a time.
    9) A congestion notification will be received whenever external conditions
       push the network into a congested state. The client, as a reaction, will
       lower the targeted bit rate to one-tenth of the initial value. Moreover,
       the client will request notification when the bandwidth available is
       enough to return to the previous bit rate.

Environment:

    User-Mode only

Revision History:

------------------------------------------------------------------------------*/

#include <winsock2.h>
#include <mswsock.h>
#include <stdlib.h>
#include <stdio.h>
#include <qos2.h>
#pragma comment(lib, "ws2_32.lib")
#pragma comment(lib, "QWave.lib")
#pragma warning(disable:4127)   // condition expression is constant

//******************************************************************************
// Forward declarations
//******************************************************************************
VOID
client(
    __in                int     argc,
    __in_ecount(argc)   char* argv[]
);

VOID
server();

VOID
help();

//******************************************************************************
// Global defines
//******************************************************************************

// Size of the data part of each datagram that is exchanged between the
// client and the server. This does not count the IP and UDP header.
#define DATAGRAM_SIZE       1400

// Number of bursts we are aiming for per second
#define BURSTS_PER_SECOND   40

// Port to be used for communication, 
// 40007 in hex in network byte order 
#define PORT                0x479c

//******************************************************************************
// Global variables
//******************************************************************************

// Array used by both the client routine and the server routine to
// send and receive data
CHAR                dataBuffer[DATAGRAM_SIZE];

// Version of the QOS subsystem we wish to use
QOS_VERSION QosVersion = { 1 , 0 };

//******************************************************************************
// Routine: 
//      main
//
// Description:
//      Entry point. Verifies that the number of parameters is enough to 
//      ascertain whether this instance is to be a client or server.
// 
//******************************************************************************
int
_cdecl
main(
    __in                int argc,
    __in_ecount(argc)   char* argv[]
)
{
    // Verify the number of command line arguments
    if (argc < 2) {
        help();
        exit(0);
    }

    if (strcmp(argv[1], "-s") == 0) {
        // If the call was "qossample -s" run as a server
        server();
    }
    else if (strcmp(argv[1], "-c") == 0) {
        // If the call was "qossample -c ..." run as a client
        client(argc, argv);
    }

    // Else, run the help routine
    help();
    return 1;
}

//******************************************************************************
// Routine: 
//      socketCreate
//
// Description:
//      This routine prepares the socket for the client instance. To do so,
//      it converts the address parameter from string to IP address. Then
//      it creates a UDP socket which it connects to this destination.
//
//      Since we will use TransmitPackets for our send operations, the function
//      pointer is obtained from Winsock.
// 
//******************************************************************************
VOID
socketCreate(
    __in    PCHAR                       destination,
    __out   SOCKET* socket,
    __out   ADDRESS_FAMILY* addressFamily,
    __out   LPFN_TRANSMITPACKETS* transmitPackets
) {
    // Return value of WSAStartup
    WSADATA             wsaData;

    // Temporarily used to represent the destination IP address
    SOCKADDR_STORAGE    destAddr;

    // Result of the various Winsock API calls
    INT                 returnValue;

    // Used by WSAStringToAddressA
    INT                 sockaddrLen;

    // Used by WSAIoctl
    DWORD               bytesReturned;

    // GUID of the TransmitPacket Winsock2 function which we will
    // use to send the traffic at the client side.
    GUID TransmitPacketsGuid = WSAID_TRANSMITPACKETS;

    // Start Winsock
    returnValue = WSAStartup(MAKEWORD(2, 2), &wsaData);

    if (returnValue != 0) {
        printf("%s:%d - WSAStartup failed (%d)\n",
            __FILE__, __LINE__, returnValue);
        exit(1);
    }

    // First attempt to convert the string to an IPv4 address
    sockaddrLen = sizeof(destAddr);
    destAddr.ss_family = AF_INET;
    returnValue = WSAStringToAddressA(destination,
        AF_INET,
        NULL,
        (LPSOCKADDR)&destAddr,
        &sockaddrLen);

    if (returnValue != ERROR_SUCCESS) {
        // If this fails,
        // Attempt to convert the string to an IPv6 address
        sockaddrLen = sizeof(destAddr);
        destAddr.ss_family = AF_INET6;
        returnValue = WSAStringToAddressA(destination,
            AF_INET6,
            NULL,
            (LPSOCKADDR)&destAddr,
            &sockaddrLen);

        if (returnValue != ERROR_SUCCESS) {
            // If this again fails, exit the application
            // But print out the help information first.
            printf("%s:%d - WSAStringToAddressA failed (%d)\n",
                __FILE__, __LINE__, WSAGetLastError());
            help();
            exit(1);
        }
    }

    // Set the destination port.
    SS_PORT((PSOCKADDR)&destAddr) = PORT;

    // Copy the address family back to caller
    *addressFamily = destAddr.ss_family;

    // Create a UDP socket
    *socket = WSASocket(destAddr.ss_family,
        SOCK_STREAM,
        0,
        NULL,
        0,
        WSA_FLAG_OVERLAPPED);

    if (*socket == INVALID_SOCKET) {
        printf("%s:%d - WSASocket failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Connect the new socket to the destination
    returnValue = WSAConnect(*socket,
        (PSOCKADDR)&destAddr,
        sizeof(destAddr),
        NULL,
        NULL,
        NULL,
        NULL);

    if (returnValue != NO_ERROR) {
        printf("%s:%d - WSAConnect failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }

    // Query the function pointer for the TransmitPacket function
    returnValue = WSAIoctl(*socket,
        SIO_GET_EXTENSION_FUNCTION_POINTER,
        &TransmitPacketsGuid,
        sizeof(GUID),
        transmitPackets,
        sizeof(PVOID),
        &bytesReturned,
        NULL,
        NULL);

    if (returnValue == SOCKET_ERROR) {
        printf("%s:%d - WSAIoctl failed (%d)\n",
            __FILE__, __LINE__, WSAGetLastError());
        exit(1);
    }
}

//******************************************************************************
// Routine: 
//      client
//
// Description:
//      This routine creates a datagram socket which it connects to the remote
//      IP address. This socket is then added to a QOS flow. The application
//      uses the flow to rate control its outgoing traffic. Packets on this flow
//      will be prioritized if the network path between the client and receiver
//      support prioritization. Specifically, if the receiver is:
//
//      A) On-link (same subnet), both 802.1Q and DSCP are applied subject to
//         available bandwidth and network support.
//      B) Off-link (different subnet), only DSCP is applied regardless of
//         available bandwidth.
//
//      Moreover, the application queries the characteristics of the network 
//      path for the socket. If estimates are available, the application:
//
//      1) may readjust its throughput so as to not cause congestion on the 
//         network.
//      2) will be notified when the network enters congestion. As a result,
//         it will lower it's throughput to 1/10 the targeted rate.
//      3) will be notified when the network is no longer congested and enough
//         bandwidth is available for the application to return to its targeted
//         rate.
//
//******************************************************************************
VOID
client(
    __in                int     argc,
    __in_ecount(argc)   char* argv[]
) {
    // Address family of the datagram socket
    ADDRESS_FAMILY              addressFamily;

    // Socket for our traffic experiment
    SOCKET                      socket;

    // Function pointer to the TransmitPacket Winsock2 function
    LPFN_TRANSMITPACKETS        transmitPacketsFn;
    // Array of transmit elements
    LPTRANSMIT_PACKETS_ELEMENT  transmitEl;

    // Target packet rate and bit rate this application will aim to send
    // If there is no congestion
    ULONG                       targetPacketRate, targetBitRate;
    // Current rate at which the application is sending. If the network is
    // congested, this will be less than the target rate
    ULONG                       currentPacketRate, currentBitRate;
    // Counters for the achieved packet rate and achieved bit rate
    ULONG                       achievedPacketRate, achievedBitRate;

    // Timer; used to periodically output to the screen our counters
    HANDLE                      timerEvent;
    LARGE_INTEGER               timerAwakenTime;

    // Handle to the QOS subsystem
    // Returned by QOSCreateHandle
    HANDLE                      qosHandle;
    // ID of the QOS flow we will create for the socket. 
    // Returned by QOSAddSocketToFlow
    QOS_FLOWID                  flowID;
    // Result of QOSQueryFlow
    QOS_FLOW_FUNDAMENTALS       flowFund;
    // Parameter to QOSSetFlow
    QOS_FLOWRATE_OUTGOING       flowRate;
    // If TRUE, the QOS subsystem is running network experiments on the
    // network path to the destination. If false, estimates are not available.
    // The flow is still marked and shaped.
    BOOL                        estimatesAvailable;
    // Result of the QOS API calls.
    BOOL                        result;

    // Overlapped operation used for TransmitPackets
    WSAOVERLAPPED               sendOverlapped;
    // Overlapped operation used for QOSNotifyFlow to be notified
    // of network congestions.
    OVERLAPPED                  congestionOverlapped;
    // Overlapped operation used for QOSNotifyFlow to be notified when, 
    // after congestion, there is enough bandwidth for the target rate
    OVERLAPPED                  availableOverlapped;
    // TRUE if the network is currently congested
    BOOL                        congestion;

    ULONG                       temp;

    // Verify the number of command line arguments
    if (argc != 4) {
        help();
        exit(1);
    }

    // Identify what destination IP address we're trying to talk to and
    // connect a UDP socket to it
    socketCreate(argv[2],
        &socket,
        &addressFamily,
        &transmitPacketsFn);

    printf("socket resut %d\n", socket);

    if (FALSE == QOSCreateHandle(&QosVersion, &qosHandle)) {
        printf("%s:%d - QOSCreateHandle failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // The Flow ID parameter MUST be set to 0 as input to QOSAddSocketToFlow
    flowID = 0;

    // Create a flow for our socket
    result = QOSAddSocketToFlow(qosHandle,
        socket,
        NULL,
        QOSTrafficTypeExcellentEffort,
        QOS_NON_ADAPTIVE_FLOW,
        &flowID);

    if (result == FALSE) {
        printf("%s:%d - QOSAddSocketToFlow failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Read the data rate in bits/s passed on the command line
    targetBitRate = atol(argv[3]);

    if (targetBitRate == 0) {
        help();
        exit(1);
    }

    // Convert from bits to bytes
    targetBitRate /= 8;

    // Calculate how many packets we need; round up.
    targetPacketRate = targetBitRate / ARRAYSIZE(dataBuffer);
    if (targetBitRate % ARRAYSIZE(dataBuffer) != 0)
        targetPacketRate++;

    // Calculate the number of packets per bursts; round up
    if (targetPacketRate % BURSTS_PER_SECOND != 0) {
        targetPacketRate /= BURSTS_PER_SECOND;
        targetPacketRate++;
    }
    else {
        targetPacketRate /= BURSTS_PER_SECOND;
    }

    // Calculate the final bit rate, targetBitRate, in bps that the application
    // will send.
    targetBitRate = BURSTS_PER_SECOND
        * targetPacketRate
        * ARRAYSIZE(dataBuffer)
        * 8;

    //
    // Allocate an array of transmit elements big enough to send 
    // targetPacketRate packets every burst.
    transmitEl = static_cast<LPTRANSMIT_PACKETS_ELEMENT>(HeapAlloc(GetProcessHeap(),
        HEAP_ZERO_MEMORY,
        sizeof(TRANSMIT_PACKETS_ELEMENT)
        * targetPacketRate));

    if (transmitEl == NULL) {
        printf("%s:%d - HeapAlloc failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }


    // For the purpose of this application, we only use one data buffer for all 
    // of our packets.
    ZeroMemory(&dataBuffer, sizeof(dataBuffer));

    //
    // Initialize each of these transmit element to point to the same zeroed out
    // data buffer
    for (temp = 0; temp < targetPacketRate; temp++) {
        transmitEl[temp].dwElFlags = TP_ELEMENT_MEMORY | TP_ELEMENT_EOP;
        transmitEl[temp].pBuffer = dataBuffer;
        transmitEl[temp].cLength = sizeof(dataBuffer);
    }

    // Print out what we'll be doing
    printf("----------------------------------"
        "----------------------------------\n"
        "This instance of the QOS sample program is aiming to:\n"
        "\t - Send %d bits of UDP traffic per second\n"
        "\t - In packets of %d bytes\n"
        "\t - In bursts of %d packets every %d milliseconds\n\n"
        "----------------------------------"
        "----------------------------------\n",
        targetBitRate,
        ARRAYSIZE(dataBuffer),
        targetPacketRate,
        1000 / BURSTS_PER_SECOND);

    // Assume, by default, that estimates are not available
    estimatesAvailable = FALSE;

    printf("Querying fundamentals about the network path: ");
    do {
        temp = sizeof(flowFund);

        // Query the flow fundamentals for the path to the destination.
        // This will return estimates of the bottleneck bandwidth, available
        // bandwidth and average RTT.
        result = QOSQueryFlow(qosHandle,
            flowID,
            QOSQueryFlowFundamentals,
            &temp,
            &flowFund,
            0,
            NULL);

        if (result == FALSE) {
            DWORD lastError;

            lastError = GetLastError();

            if (lastError == ERROR_NO_DATA) {
                // If the call failed, this could be because the QOS subsystem 
                // has not yet gathered enough data for estimates. If so, the
                // QOS2 api returns ERROR_NO_DATA.
                printf(".");
                Sleep(1000);
            }
            else if (lastError == ERROR_NOT_SUPPORTED)
            {
                // If the call failed, this could be because the QOS subsystem 
                // cannot run network experiments on the network path to the
                // destination. If so, the API returns ERROR_NOT_SUPPORTED.
                printf("NOT SUPPORTED\n"
                    "\t   Network conditions to this destination could not\n"
                    "\t   be detected as your network configuration is not\n"
                    "\t   supported for network experiments\n");
                break;
            }
            else {
                printf("%s:%d - QOSQueryFlow failed (%d)\n",
                    __FILE__, __LINE__, lastError);
                exit(1);
            }
        }
        else if ((flowFund.BottleneckBandwidthSet == FALSE)
            || (flowFund.AvailableBandwidthSet == FALSE)) {
            // If the call succeeded but bottleneck bandwidth or 
            // available bandwidth are not known then estimates are still
            // processing; query again in 1 second.
            printf(".");
            Sleep(1000);
        }
        else {
            // Estimate where available.
            double bottleneck;
            double available;

            estimatesAvailable = TRUE;

            // Convert the bottleneck bandwidth from bps to mbps
            bottleneck = (double)flowFund.BottleneckBandwidth;
            bottleneck /= 1000000.0;

            // Convert available bandwidth from bps to mbps
            available = (double)flowFund.AvailableBandwidth;
            available /= 1000000.0;

            printf("DONE\n"
                "\t - Bottleneck bandwidth is approximately %4.2f Mbps\n"
                "\t - Available bandwidth is approximately  %4.2f Mbps\n",
                bottleneck,
                available);

            break;
        }
    } while (TRUE);

    if (estimatesAvailable == TRUE) {
        UINT64 targetBitRateWithHeaders;

        printf("\nNOTE: the accuracy of the QOS estimates can be impacted by\n"
            "any of the following,\n\n"
            "\t - Both the network interface at the client\n"
            "\t   and at the server must be using NDIS 6 drivers.\n"
            "\t - If the server is not a Windows Vista host, verify that \n"
            "\t   it implements the LLD2 networking protocol. You can\n"
            "\t   find more information at http://www.microsoft.com.\n"
            "\t - IPSec, VPNs and enterprise class networking equipment\n"
            "\t   may interfere with network experiments.\n\n");

        // Calculate what our target bit rate, with protocol headers for 
        // IP(v4/v6) and UDP will be.
        targetBitRateWithHeaders = QOS_ADD_OVERHEAD(addressFamily,
            IPPROTO_UDP,
            ARRAYSIZE(dataBuffer),
            targetBitRate);

        if (flowFund.AvailableBandwidth < targetBitRateWithHeaders) {
            // If the estimate of available bandwidth is not sufficient for the
            // target bit rate (with headers), drop to a lesser throughput
            UINT64 availableBandwidth;

            // The estimate returned does not account for headers
            // Remove the estimated overhead for our address family and UDP.
            availableBandwidth = QOS_SUBTRACT_OVERHEAD(
                addressFamily,
                IPPROTO_UDP,
                ARRAYSIZE(dataBuffer),
                flowFund.AvailableBandwidth);

            // Calculate the rate of packets we can realistically send
            targetPacketRate = (LONG)availableBandwidth / 8;
            targetPacketRate /= ARRAYSIZE(dataBuffer);
            targetPacketRate /= BURSTS_PER_SECOND;

            // Calculate the real bit rate we'll be using
            targetBitRate = BURSTS_PER_SECOND
                * targetPacketRate
                * ARRAYSIZE(dataBuffer)
                * 8;

            printf("Not enough available bandwidth for the requested bitrate.\n"
                "Downgrading to lesser bitrate - %d.\n", targetBitRate);
        }
    }

    // Our starting rate is this target bit rate
    currentBitRate = targetBitRate;
    currentPacketRate = targetPacketRate;

    // Ask the QOS subsystem to shape our traffic to this bit rate. Note that
    // the application needs to account for the size of the IP(v4/v6) 
    // and UDP headers.

    // Calculate the real bandwidth we will need to be shaped to.
    flowRate.Bandwidth = QOS_ADD_OVERHEAD(addressFamily,
        IPPROTO_UDP,
        ARRAYSIZE(dataBuffer),
        currentBitRate);

    // Set shaping characteristics on our QOS flow to smooth out our bursty
    // traffic.
    flowRate.ShapingBehavior = QOSShapeAndMark;

    // The reason field is not applicable for the initial call.
    flowRate.Reason = QOSFlowRateNotApplicable;
    result = QOSSetFlow(qosHandle,
        flowID,
        QOSSetOutgoingRate,
        sizeof(flowRate),
        &flowRate,
        0,
        NULL);

    if (result == FALSE) {
        printf("%s:%d - QOSSetFlow failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Allocate a waitable timer. We will use this timer to periodically
    // awaken and output statistics.
    timerEvent = CreateWaitableTimer(NULL, FALSE, NULL);

    if (timerEvent == NULL) {
        printf("%s:%d - CreateWaitableTimer failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    //
    // Set the sampling time to 1 second
    timerAwakenTime.QuadPart = -10000000; // 1 second

    if (FALSE == SetWaitableTimer(timerEvent,
        &timerAwakenTime,
        1000,
        NULL,
        NULL,
        FALSE)) {

        printf("%s:%d - SetWaitableTimer failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // Prepare the support overlapped structures to detect congestion,
    // notifications of bandwidth change and the completion of our send 
    // routines.
    ZeroMemory(&congestionOverlapped, sizeof(congestionOverlapped));
    congestionOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (congestionOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    ZeroMemory(&availableOverlapped, sizeof(availableOverlapped));
    availableOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (availableOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    ZeroMemory(&sendOverlapped, sizeof(sendOverlapped));
    sendOverlapped.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

    if (sendOverlapped.hEvent == NULL) {
        printf("%s:%d - CreateEvent failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    if (estimatesAvailable == TRUE) {
        // If estimates are available, we request a notification 
        // for congestion. This notification will complete whenever network
        // congestion is detected.
        result = QOSNotifyFlow(qosHandle,
            flowID,
            QOSNotifyCongested,
            NULL,
            NULL,
            0,
            &congestionOverlapped);

        if (result == FALSE) {
            DWORD lastError;

            lastError = GetLastError();
            if (lastError != ERROR_IO_PENDING) {
                printf("%s:%d - QOSNotifyFlow failed (%d)\n",
                    __FILE__, __LINE__, lastError);
                exit(1);
            }
        }
    }

    printf("----------------------------------"
        "----------------------------------\n"
        " # packets | # of bits  |  Bottleneck "
        "|  Available |  Congestion   \n"
        "----------------------------------"
        "----------------------------------\n");

    // Initialize our counters to 0
    achievedBitRate = achievedPacketRate = 0;
    congestion = FALSE;

    do {
        BOOL sendNextGroup;

        // Send the first burst of packets
        result = (*transmitPacketsFn)(socket,
            transmitEl,
            currentPacketRate,
            0xFFFFFFFF,
            &sendOverlapped,
            TF_USE_KERNEL_APC);

        if (result == FALSE) {
            DWORD lastError;

            lastError = WSAGetLastError();
            if (lastError != ERROR_IO_PENDING) {
                printf("%s:%d - TransmitPackets failed (%d)\n",
                    __FILE__, __LINE__, GetLastError());
                exit(1);
            }
        }

        // Increase the counter of sent data
        achievedPacketRate += currentPacketRate;
        achievedBitRate += currentPacketRate * ARRAYSIZE(dataBuffer) * 8;
        sendNextGroup = FALSE;

        do {
            HANDLE          waitEvents[3];
            DWORD           waitResult;

            // Wait for any of the 3 events to complete

            // The 1 second periodic timer
            waitEvents[0] = timerEvent;

            if (congestion)
                // Notification of available bandwidth
                waitEvents[1] = availableOverlapped.hEvent;
            else
                // Notification of congestion
                waitEvents[1] = congestionOverlapped.hEvent;

            // Notification that the send operation has completed
            waitEvents[2] = sendOverlapped.hEvent;

            waitResult = WaitForMultipleObjects(ARRAYSIZE(waitEvents),
                (PHANDLE)waitEvents,
                FALSE,
                INFINITE);

            switch (waitResult) {
            case WAIT_OBJECT_0: {
                // The event for the periodic timer is set

                printf("%10d | ", achievedPacketRate);
                printf("%10d | ", achievedBitRate);

                if (estimatesAvailable) {
                    // If estimates are available for the network path
                    // query for estimates and output to the console along
                    // with our counters.

                    // Ascertained through QOSQueryFlow
                    BOOL estimateIndicatesCongestion;

                    // Default is unknown
                    estimateIndicatesCongestion = FALSE;

                    ZeroMemory(&flowFund, sizeof(flowFund));

                    temp = sizeof(flowFund);
                    QOSQueryFlow(qosHandle,
                        flowID,
                        QOSQueryFlowFundamentals,
                        &temp,
                        &flowFund,
                        0,
                        NULL);

                    if (flowFund.BottleneckBandwidthSet)
                        printf("%10d  | ", flowFund.BottleneckBandwidth);
                    else
                        printf("   NO DATA  | ");

                    if (flowFund.AvailableBandwidthSet) {
                        if (flowFund.AvailableBandwidth == 0)
                            estimateIndicatesCongestion = TRUE;

                        printf("%10d | ", flowFund.AvailableBandwidth);
                    }
                    else
                        printf("   NO DATA | ");

                    if (estimateIndicatesCongestion)
                        printf(" CONGESTION\n");
                    else
                        printf("\n");
                }
                else {
                    // Bandwidth estimates are not available
                    printf("    N/A     |     N/A    |\n");
                }

                //
                // Reset the counters
                achievedPacketRate = achievedBitRate = 0;
                break;
            }
            case WAIT_OBJECT_0 + 1: {
                // This is either a notification for congestion or 
                // for bandwidth available 

                if (congestion == FALSE) {
                    UINT64 targetBandwidthWithOverhead;
                    ULONG  bufferSize;
                    //
                    // Congestion
                    //
                    printf("----------------------------------"
                        "----------------------------------\n"
                        "CONGESTION\n"
                        "----------------------------------"
                        "----------------------------------\n");

                    //
                    // Reduce the current rate to one-tenth of the initial rate
                    // Insure you're always sending at least 1 packet per
                    // burst.
                    if (currentPacketRate < 10)
                        currentPacketRate = 1;
                    else
                        currentPacketRate /= 10;

                    // Calculate the new bit rate we'll be using
                    currentBitRate = BURSTS_PER_SECOND
                        * currentPacketRate
                        * ARRAYSIZE(dataBuffer)
                        * 8;


                    // Update the shaping characteristics on our QOS flow 
                    // to smooth out our bursty traffic.
                    flowRate.Bandwidth = QOS_ADD_OVERHEAD(
                        addressFamily,
                        IPPROTO_UDP,
                        ARRAYSIZE(dataBuffer),
                        currentBitRate);

                    flowRate.ShapingBehavior = QOSShapeAndMark;
                    flowRate.Reason = QOSFlowRateCongestion;

                    result = QOSSetFlow(qosHandle,
                        flowID,
                        QOSSetOutgoingRate,
                        sizeof(flowRate),
                        &flowRate,
                        0,
                        NULL);

                    if (result == FALSE) {
                        printf("%s:%d - QOSSetFlow failed (%d)\n",
                            __FILE__, __LINE__, GetLastError());
                        exit(1);
                    }

                    // Request a notification for when there is enough bandwidth
                    // to return to our previous targeted bit rate.
                    // This will complete only when the network is no longer
                    // congested and bandwidth is available.
                    targetBandwidthWithOverhead = QOS_ADD_OVERHEAD(
                        addressFamily,
                        IPPROTO_UDP,
                        ARRAYSIZE(dataBuffer),
                        targetBitRate);

                    bufferSize = sizeof(targetBandwidthWithOverhead);

                    result = QOSNotifyFlow(qosHandle,
                        flowID,
                        QOSNotifyAvailable,
                        &bufferSize,
                        (PVOID)&targetBandwidthWithOverhead,
                        0,
                        &availableOverlapped);

                    if (result == FALSE) {
                        DWORD lastError;

                        lastError = GetLastError();
                        if (lastError != ERROR_IO_PENDING) {
                            printf("%s:%d - QOSNotifyFlow failed (%d)\n",
                                __FILE__, __LINE__, lastError);
                            exit(1);
                        }
                    }

                    congestion = TRUE;
                }
                else {
                    //
                    // End of congestion
                    //
                    printf("----------------------------------"
                        "----------------------------------\n"
                        "END OF CONGESTION\n"
                        "----------------------------------"
                        "----------------------------------\n");

                    //
                    // Reset the current packet rate to the initial target rate
                    currentPacketRate = targetPacketRate;

                    // Reset the current bit rate to the initial target rate
                    currentBitRate = targetBitRate;


                    // Update the shaping characteristics on our QOS flow 
                    // to smooth out our bursty traffic.
                    flowRate.Bandwidth = QOS_ADD_OVERHEAD(addressFamily,
                        IPPROTO_UDP,
                        ARRAYSIZE(dataBuffer),
                        targetBitRate);

                    flowRate.ShapingBehavior = QOSShapeAndMark;
                    flowRate.Reason = QOSFlowRateCongestion;
                    result = QOSSetFlow(qosHandle,
                        flowID,
                        QOSSetOutgoingRate,
                        sizeof(flowRate),
                        &flowRate,
                        0,
                        NULL);

                    if (result == FALSE) {
                        printf("%s:%d - QOSSetFlow failed (%d)\n",
                            __FILE__, __LINE__, GetLastError());
                        exit(1);
                    }

                    // Request a notification for the next network congestion
                    result = QOSNotifyFlow(qosHandle,
                        flowID,
                        QOSNotifyCongested,
                        NULL,
                        NULL,
                        0,
                        &congestionOverlapped);

                    if (result == FALSE) {
                        DWORD lastError;

                        lastError = GetLastError();
                        if (lastError != ERROR_IO_PENDING) {
                            printf("%s:%d - QOSNotifyFlow failed (%d)\n",
                                __FILE__, __LINE__, lastError);
                            exit(1);
                        }
                    }

                    congestion = FALSE;
                }
                break;
            }
            case WAIT_OBJECT_0 + 2: {
                // The transmit packet has completed its send, 
                // If it did so successfully, it's time to send the next 
                // burst of packets.
                BOOL    overlappedResult;
                DWORD   ignoredNumOfBytesSent;
                DWORD   ignoredFlags;

                overlappedResult = WSAGetOverlappedResult(
                    socket,
                    &sendOverlapped,
                    &ignoredNumOfBytesSent,
                    FALSE,
                    &ignoredFlags);

                if (overlappedResult == FALSE) {
                    printf("%s:%d - TransmitPackets failed (%d)\n",
                        __FILE__, __LINE__, WSAGetLastError());
                    exit(1);
                }

                // Time to send out the next bunch of packets
                sendNextGroup = TRUE;
                break;
            }
            default:
                // The wait call failed.
                printf("%s:%d - WaitForMultipleObjects failed (%d)\n",
                    __FILE__, __LINE__, GetLastError());
                exit(1);
            }

        } while (sendNextGroup == FALSE);
    } while (TRUE);
}

//******************************************************************************
// Routine: 
//      server
//
// Description:
//      This routine creates a socket through which it will receive
//      any datagram sent by the client. It counts the number of packet 
//      and the number of bytes received. Periodically, it outputs this 
//      information to the console.
//
//******************************************************************************
VOID
server()
{
    HANDLE              timerEvent;
    LARGE_INTEGER       timerAwakenTime;

    // Return value of WSAStartup
    WSADATA             wsaData;



    // IPv6 wildcard address and port number 40007 to listen on at the server
    SOCKADDR_IN6 IPv6ListeningAddress = { AF_INET6,
                                            PORT,
                                            0,
                                            IN6ADDR_ANY_INIT,
                                            0 };

    // Result value from the various API calls
    DWORD               result;

    // Used to specify an option to setsocktopt
    ULONG               optionValue;

    // Overlapped structure used to post asynchronous receive calls
    WSAOVERLAPPED       recvOverlapped;

    // Counters of the number of bytes and packets received over 
    // a period of time.
    DWORD               numPackets, numBytes;



    // Initialize Winsock
    result = WSAStartup(MAKEWORD(2, 2), &wsaData);

    if (result != 0) {
        printf("%s:%d - WSAStartup failed (%d)\n",
            __FILE__, __LINE__, GetLastError());
        exit(1);
    }

    // 创建 TCP 监听套接字
    SOCKET listener = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (listener == INVALID_SOCKET) {
        printf("Error creating socket: %ld\n", WSAGetLastError());
        WSACleanup();
        exit(1);
    }

    // 绑定到本地地址和端口
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    addr.sin_port = PORT;
    int res = bind(listener, (struct sockaddr*)&addr, sizeof(addr));
    if (res == SOCKET_ERROR) {
        printf("Error binding socket: %ld\n", WSAGetLastError());
        closesocket(listener);
        WSACleanup();
        exit(1);
    }
    else {
        printf("begin listen on %d", PORT);
    }

    // 开始监听传入的连接
    res = listen(listener, SOMAXCONN);
    if (res == SOCKET_ERROR) {
        printf("Error listening on socket: %ld\n", WSAGetLastError());
        closesocket(listener);
        WSACleanup();
        exit(1);
    }

    // 接受连接并处理请求
    while (1) {
        struct sockaddr_in client_addr;
        int client_addr_len = sizeof(client_addr);
        SOCKET client = accept(listener, (struct sockaddr*)&client_addr, &client_addr_len);
        if (client == INVALID_SOCKET) {
            printf("Error accepting client: %ld\n", WSAGetLastError());
            closesocket(listener);
            WSACleanup();
            exit(1);
        }

        while (true) {
            // 读取请求数据
            char buf[1024];
            int bytes_read = recv(client, buf, sizeof(buf), 0);
            if (bytes_read == SOCKET_ERROR)
            {
                printf("Error read: %ld\n", WSAGetLastError());
                break;
            }
            // 打印请求数据
            printf("Received request:%d \n", bytes_read);
        }
        printf("client closed!\n");
        // 关闭连接
        closesocket(client);
    }
    // 关闭监听套接字和清理 Winsock
    closesocket(listener);
    WSACleanup();
}

//******************************************************************************
// Routine: 
//      help
//
// Description:
//      This routine prints out the usage information for the program
//
//******************************************************************************
VOID
help()
{
    printf("USAGE:\n"
        "\tSERVER: qossample -s\n"
        "\tCLIENT: qossample -c IP-ADDRESS BIT-RATE\n\n"
        "\tIn the following example, the application would try to send\n"
        "\t20 Mb of traffic to the host at 10.0.0.1:\n"
        "\tqossample -c 10.0.0.1 20000000\n");
    return;
}

Rust写的TCP版本

这段代码是一个 Rust 程序,它使用 Windows 系统的 QoS API 来实现一个简单的客户端,连接到指定的服务器并发送一条消息。

程序的主要步骤如下:

  1. 连接到服务器,创建一个 TCP 流。

  2. 获取流的原始套接字句柄。

  3. 使用 QOSCreateHandle 函数创建一个 QoS 句柄。

  4. 使用 QOSAddSocketToFlow 函数将套接字添加到 QoS 流中,指定流量类型为音视频流量。

  5. 发送一条消息到服务器。

  6. 从服务器接收一条消息。

  7. 使用 QOSCloseHandle 函数关闭 QoS 句柄。

在程序中,使用了 Rust 的标准库中的 io、net 和 os 模块,以及第三方库 windows-sys,该库提供了对 Windows 系统 API 的 Rust 绑定。

程序使用了 unsafe 关键字,这是因为它调用了 Windows 系统的 API,这些 API 不是 Rust 的安全代码。程序还使用了 panic! 宏来处理错误情况,这将导致程序崩溃并打印错误消息。

总的来说,这段代码演示了如何使用 Rust 和 Windows 系统的 QoS API 来实现一个简单的客户端,连接到服务器并发送一条消息。这条消息的DSCP值是被设置过的。

rust 复制代码
// 导入必要的库
use std::io::{Read, Write};
use std::net::TcpStream;
use std::os::windows::io::AsRawSocket;

// 导入 Windows API
use windows_sys::Win32::Foundation::{GetLastError, FALSE, HANDLE};
use windows_sys::Win32::NetworkManagement::QoS::{
    QOSAddSocketToFlow, QOSCloseHandle, QOSCreateHandle, QOSTrafficTypeAudioVideo,
    QOS_NON_ADAPTIVE_FLOW, QOS_VERSION,
};
use windows_sys::Win32::Networking::WinSock::SOCKET;

fn main() -> std::io::Result<()> {
    // 连接到服务器
    let mut stream = TcpStream::connect("118.24.148.75:8080")?; // 连接到服务器
    let socket = stream.as_raw_socket(); // 获取 socket

    // 创建 QoS 句柄
    let version = QOS_VERSION {
        MajorVersion: 1,
        MinorVersion: 0,
    };
    let mut qos_handle: HANDLE = 0;
    let res = unsafe { QOSCreateHandle(&version, &mut qos_handle) }; // 创建 QoS 句柄
    if res == FALSE {
        let error = unsafe { GetLastError() };
        panic!("Error creating QoS handle: {}, {}", res, error);
    }
    let mut flowid = 0;

    //会将dscp值设置为0x28
    let res = unsafe {
        QOSAddSocketToFlow(
            qos_handle,
            socket as SOCKET,
            std::ptr::null_mut(),
            QOSTrafficTypeAudioVideo,
            QOS_NON_ADAPTIVE_FLOW,
            &mut flowid,
        )
    };
    if res == FALSE {
        panic!("Error adding socket to flow: {}", res);
    }

    // 发送数据
    stream.write_all(b"Hello, server!")?;

    // 接收数据
    let mut buf = [0; 512];
    let size = stream.read(&mut buf)?;
    let response = String::from_utf8_lossy(&buf[..size]);
    println!("Received: {}", response);

    let ret = unsafe { QOSCloseHandle(qos_handle) };
    if ret == FALSE {
        panic!("Error closing QoS handle: {}", ret);
    }

    Ok(())
}

Cargo.toml如下

rust 复制代码
[package]
name = "hello"
version = "0.1.0"
edition = "2021"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
#rand = "0.7"
#bytes = "1"
#tokio = { version = "1.0", features = ["full"] }
#socket2 = {}
#once_cell = "1.8"
#signal-hook = {}
#nix = {features = ["net"]}
windows-sys = {version = "*",features = ["Win32_Foundation","Win32_Networking_WinSock","Win32_NetworkManagement","Win32_NetworkManagement_QoS"]}
相关推荐
戎梓漩1 小时前
windows下安装curl,并集成到visual studio
ide·windows·visual studio
一个小坑货2 小时前
Cargo Rust 的包管理器
开发语言·后端·rust
bluebonnet272 小时前
【Rust练习】22.HashMap
开发语言·后端·rust
蓝田~3 小时前
观察者模式和订阅模式
windows·观察者模式
VertexGeek5 小时前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
梓仁沐白10 小时前
ubuntu+windows双系统切换后蓝牙设备无法连接
windows·ubuntu
九鼎科技-Leo14 小时前
什么是 WPF 中的依赖属性?有什么作用?
windows·c#·.net·wpf
Yang.9916 小时前
基于Windows系统用C++做一个点名工具
c++·windows·sql·visual studio code·sqlite3
我不瘦但很逗16 小时前
Windows下使用DBeaver连接云数据库(MySQL)
数据库·windows
ashane131417 小时前
Java list
java·windows·list