//-//--------------------------------------------------------------------
// TUCN, Computer Science Department
// Input/Output Systems and Peripheral Devices
//--------------------------------------------------------------------
// http://users.utcluj.ro/~baruch/en/pages/teaching/inputoutput-systems/laboratory.php
//--------------------------------------------------------------------
// File:		AppScroll-e.cpp
// Date:		10.10.2015
// Modified:	21.11.2015
//--------------------------------------------------------------------
// IOS application example with vertical scroll bar
//--------------------------------------------------------------------

#include <windows.h>
#include <iostream>
#include "Hw.h"
#include "PCI-e.h"
#include "PCI-vendor-dev.h"

#define NLIN 500								// number of lines in the display window
#define NCOL 240								// number of columns in the display window
// Global variables
char szBuffer[NLIN][NCOL];						// buffer for the window contents
int  cLine;										// number of current line in the display buffer

// Declarations of external functions
void DisplWindow(HWND hWnd);

//--------------------------------------------------------------------
// Function AppScroll
//--------------------------------------------------------------------
//
// Function:	IOS application example with vertical scroll bar
//
// Parameters:	- Handle to the application window
//
// Returns:		0 - Operation completed successfully
//				1 - Error at initializing the Hw driver
//
//--------------------------------------------------------------------

PPCI_CONFIG0 getHeaderFromPCIe(byte busNr, byte devNr, byte fctNr)
{
	//start address : PCI_CONFIG_START (different)
	DWORD64 qwFuncAddr = PCI_CONFIG_START;
	//shifts for : PCI bus, device, function
	qwFuncAddr = qwFuncAddr | (busNr << 20);//PCI 27...20
	qwFuncAddr = qwFuncAddr | (devNr << 15);//device 19...15
	qwFuncAddr = qwFuncAddr | (fctNr << 12);//function 14...12
	//return pointer
	return (PPCI_CONFIG0)qwFuncAddr;
}

DWORD getDoubleWord(BYTE busNr, BYTE devNr, BYTE fctNr, BYTE dwNumber)
{
	DWORD configAddrPort = 0x800000000 | (busNr << 16) | (devNr << 11) | (fctNr << 8) | (dwNumber << 2);

		__outpd(PCI_CONFIG_ADR, configAddrPort);

		return __inpd(PCI_CONFIG_DATA);


}

int AppScroll(HWND hWnd)
{
	int   i;

	char szMes0[] = "Error initializing the Hw driver";
	char szMes1[] = "IOS Application";

	// Initialize the Hw library
	if (!HwOpen()) {
		wsprintf(szBuffer[0], szMes0);
		MessageBox(NULL, szBuffer[0], "HwOpen", MB_ICONSTOP);
		return 1;
	}

	// Erase the display buffer and the window contents
	for (i = 0; i < NLIN; i++) {
		memset(szBuffer[i], ' ', NCOL);
	}
	cLine = 1;

	// Copy the start message into the display buffer and display the message
	wsprintf(szBuffer[cLine], szMes1);
	cLine += 2;
	DisplWindow(hWnd);

	//--------------------------------------------------------------------
	// To be completed with the application's code
	//--------------------------------------------------------------------

	for (int busNr = 0; busNr < 16; busNr++){
		for (int devNr = 0; devNr < 32; devNr++){
			for (int fctNr = 0; fctNr < 8; fctNr++){
				PPCI_CONFIG0 header = getHeaderFromPCIe(busNr, devNr, fctNr);
				WORD wVendorID = _inmw((DWORD_PTR)&header->VendorID);
				WORD wDeviceID = _inmw((DWORD_PTR)&header->DeviceID);
				if (wVendorID != 0xFFFF){
					wsprintf(szBuffer[cLine], "Bus: %d | Device: %d | Function: %d | Class Code: %x | Sub Class Code: %x | Interface: %x\n | SubSystem Vendor: %x | SubSystem ID: %x\n",
						busNr, devNr, fctNr,
						_inm((DWORD_PTR)&header->BaseClass), _inm((DWORD_PTR)&header->SubClass),
						_inm((DWORD_PTR)&header->ProgInterface), _inm((DWORD_PTR)&header->SubSystVendorID),
						_inm((DWORD_PTR)&header->SubSystID)
						);
					cLine++;
					PciClassTable->Class = _inm((DWORD_PTR)&header->BaseClass);
					PciClassTable->SubClass = _inm((DWORD_PTR)&header->SubClass);
					PciClassTable->ProgIf = _inm((DWORD_PTR)&header->ProgInterface);
					wsprintf(szBuffer[cLine], "Class Desc.: %s | Interface Desc.: %s\n", PciClassTable->ClassDesc, PciClassTable->ProgIfDesc);
					cLine++;
					for (int i = 0; i < PCI_VENTABLE_LEN; i++){
						if (wVendorID == PciVenTable[i].VenId){
							wsprintf(szBuffer[cLine], "Vendor descriptor : %s ", PciVenTable[i].VenFull);
							cLine++;
						}
					}
					for (int i = 0; i < PCI_DEVTABLE_LEN; i++){
						if (wVendorID == PciDevTable[i].VenId && wDeviceID == PciDevTable[i].DevId){
							wsprintf(szBuffer[cLine], "Chip descriptor : %s ", PciDevTable[i].ChipDesc);
							cLine++;


						}


					}
						wsprintf(szBuffer[cLine], "____________________________________________________________________________________________________ \n ");
						cLine++;
				}
			}
		}

	}

	wsprintf(szBuffer[cLine++], "16 dwords from config\n");
	for (BYTE dwOffset = 0x00; dwOffset <= 0x0F; dwOffset += 0x01)
	{
		DWORD aux = getDoubleWord(0x00, 0x1F, 0x03, dwOffset);
		wsprintf(szBuffer[cLine++], "%x\n", aux);
	}
	// Display the messages
	DisplWindow(hWnd);

	HwClose();
	return 0;
}
///WAMq9n5n
//MgqC0AnW
//D8kmw1Pe