yttrium-virtio-gpu/NetKVM/DebugTools/Netchecksum/netchecksum.cpp

/**********************************************************************
 * Copyright (c) 2008  Red Hat, Inc.
 *
 * File: netchecksum.cpp
 *
 * Defines the entry point for the console application
 *
 * This work is licensed under the terms of the GNU GPL, version 2.  See
 * the COPYING file in the top-level directory.
 *
**********************************************************************/

#define _CRT_SECURE_NO_WARNINGS

#include "stdafx.h"
extern "C" {
#include "ndis56common.h"
}

BYTE buf[0x10000];


bool ProcessFile(FILE *f, ULONG flags, ULONG result[4])
{
    bool bContinue = TRUE;
    UINT offset = 0;
    memset(buf, 0, sizeof(buf));
    tTcpIpPacketParsingResult res;
    while (bContinue)
    {
        char s[3];
        if (fread(s, 1, 1, f) == 1)
        {
            if (isxdigit(s[0]) && fread(s+1, 1, 1, f) == 1 && isxdigit(s[1]))
            {
                ULONG val;
                s[2] = 0;
                sscanf(s, "%x", &val);
                buf[offset++] = (UCHAR)val;
            }
            else if (isalpha(s[0])) bContinue = FALSE;
        }
    }
    bContinue = false;
    if (offset > 14)
    {
        ULONG expected;
        ULONG pass = 0;
        bContinue = true;
        
        if (bContinue)
        {
            pass++;
            expected = result[pass - 1];
            DPrintf(0, ("Pass %d buffer of %d started", pass, offset));
            res = ParaNdis_ReviewIPPacket(buf + 14, offset - 14, __FUNCTION__);
            DPrintf(0, ("Pass %d buffer of %d finished", pass, offset));
            if (res.value != expected)
            {
                DPrintf(0, ("%d pass FAILED: expected %08X, received %08X", pass, expected, res.value));
                bContinue = false;
            }
        }

        if (bContinue)
        {
            pass++;
            expected = result[pass - 1];
            DPrintf(0, ("Pass %d buffer of %d started", pass, offset));
            res = ParaNdis_CheckSumVerify(buf + 14, offset - 14, pcrAnyChecksum, __FUNCTION__);
            DPrintf(0, ("Pass %d buffer of %d finished", pass, offset));
            if (res.value != expected)
            {
                DPrintf(0, ("%d pass FAILED: expected %08X, received %08X", pass, expected, res.value));
                bContinue = false;
            }
        }

        if (bContinue)
        {
            pass++;
            expected = result[pass - 1];
            DPrintf(0, ("Pass %d buffer of %d started", pass, offset));
            res = ParaNdis_CheckSumVerify(buf + 14, offset - 14, pcrAnyChecksum | flags, __FUNCTION__);
            DPrintf(0, ("Pass %d buffer of %d finished", pass, offset));
            if (res.value != expected)
            {
                DPrintf(0, ("%d pass FAILED: expected %08X, received %08X", pass, expected, res.value));
                bContinue = false;
            }
        }       
        
        if (bContinue)
        {
            pass++;
            expected = result[pass - 1];
            DPrintf(0, ("Pass %d buffer of %d started", pass, offset));
            res = ParaNdis_CheckSumVerify(buf + 14, offset - 14, pcrAnyChecksum, __FUNCTION__);
            DPrintf(0, ("Pass %d buffer of %d finished", pass, offset));
            if (res.value != expected)
            {
                DPrintf(0, ("%d pass FAILED: expected %08X, received %08X", pass, expected, res.value));
                bContinue = false;
            }
        }
    }
    return bContinue;
}

struct
{
    LPCSTR file;
    ULONG flags;
    ULONG result[4];
}Jobs[] =
{
    // TCP packet with valid IPCS and PHCS, populate TCP CS
    { "tcp-ph.txt",    pcrFixXxpChecksum, { 0x28140182, 0x2814019A, 0x2814099A, 0x281401AA } },
    // TCP packet with bad IPCS and valid TCPCS, fix IPCS
    { "tcp-short.txt", pcrFixIPChecksum, { 0x28140182, 0x281401AE, 0x281405AE, 0x281401AA }  },
    // TCP packet with valid IPCS and TCPCS, populate PHCS
    { "tcp-cs.txt",    pcrFixPHChecksum, { 0x28140182, 0x281401AA, 0x281409BA, 0x2814019A }  },
    // TCP packet with valid IPCS and bad TCPCS, populate TCPCS
    { "tcp-badcs.txt", pcrFixXxpChecksum, { 0x28140182, 0x281401BA, 0x281409BA, 0x281401AA }  },
    // TCP packet with valid IPCS and bad TCPCS, populate PHCS
    { "tcp-badcs.txt", pcrFixPHChecksum, { 0x28140182, 0x281401BA, 0x281409BA, 0x2814019A }  },
    // TCP packet with valid TCPCS, populate TCPCS
    { "tcpv6-cs.txt",  pcrFixXxpChecksum, { 0x3C28018B, 0x3C2801AB, 0x3C2801AB, 0x3C2801AB }  },
    // TCP packet with valid TCPCS, populate PHCS
    { "tcpv6-cs.txt",  pcrFixPHChecksum, { 0x3C28018B, 0x3C2801AB, 0x3C2809BB, 0x3C28019B }  },
    // TCP packet with valid UDPCS, populate UDPCS
    { "udpv6-cs.txt",  pcrFixXxpChecksum, { 0x3028038B, 0x302803AB, 0x302803AB, 0x302803AB }  },
    // TCP packet with valid UDPCS, populate PHCS
    { "udpv6-cs.txt",  pcrFixPHChecksum | pcrFixIPChecksum, { 0x3028038B, 0x302803AB, 0x30280BBB, 0x3028039B }  },
};

int _tmain(int argc, _TCHAR* argv[])
{
    bool bOK = true;
    int i;
    FILE *f;
    for (i = 0; bOK && i < sizeof(Jobs)/sizeof(Jobs[0]); ++i)
    {
        f = fopen(Jobs[i].file,"rt");
        if (f)
        {
            DPrintf(0, ("Processing file %s started", Jobs[i].file));
            bOK = ProcessFile(f, Jobs[i].flags, Jobs[i].result);
            DPrintf(0, ("Processing file %s finished", Jobs[i].file));
            DPrintf(0, ("===================================="));
            fclose(f);
        }
    }

    DPrintf(0, ("Unit test %s", bOK ? "PASSED" : "FAILED"));
    
    //getchar();

    return 0;
}