NILayer.cpp

// NILayer.cpp: implementation of the CNILayer class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "PacketSender.h"
#include "NILayer.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CNILayer::CNILayer(char *pName, LPADAPTER *pAdapterObject, int iNumAdapter)
	: CBaseLayer(pName)
{
	m_AdapterObject = NULL;
	m_iNumAdapter = iNumAdapter;
	m_thrdSwitch = TRUE;
	SetAdapterList(NULL);
}

CNILayer::~CNILayer()
{
	pcap_close(m_AdapterObject);
	m_thrdSwitch = FALSE;
}

void CNILayer::PacketStartDriver()
{
	char errbuf[PCAP_ERRBUF_SIZE];

	if (m_iNumAdapter == -1) {
		AfxMessageBox("Not exist NICard");
		return;
	}

	if (m_AdapterObject == NULL)
		m_AdapterObject = pcap_open_live(m_pAdapterList[m_iNumAdapter]->name, 65536, PCAP_OPENFLAG_PROMISCUOUS, 2000, errbuf);
	else
		m_thrdSwitch = TRUE;

	if (!m_AdapterObject) {
		AfxMessageBox(errbuf);
		return;
	}

	AfxBeginThread(ReadingThread, this);
}

void CNILayer::PacketStopDriver()
{
	m_thrdSwitch = FALSE;
}

pcap_if_t *CNILayer::GetAdapterObject(int iIndex)
{
	return m_pAdapterList[iIndex];
}

void CNILayer::SetAdapterNumber(int iNum)
{
	m_iNumAdapter = iNum;
}

void CNILayer::SetAdapterList(LPADAPTER *plist)
{
	pcap_if_t *alldevs;
	pcap_if_t *d;
	int i = 0;

	char errbuf[PCAP_ERRBUF_SIZE];

	for (int j = 0; j < NI_COUNT_NIC; j++)
	{
		m_pAdapterList[j] = NULL;
	}

	/* Retrieve the device list */
	if (pcap_findalldevs(&alldevs, errbuf) == -1)
	{
		AfxMessageBox("Not exist NICard");
		return;
	}
	if (!alldevs) // 디바이스가 하나도 없을 때
	{
		AfxMessageBox("Not exist NICard");
		return;
	}

	/* Set adapter list */
	for (d = alldevs; d; d = d->next)
	{
		m_pAdapterList[i++] = d;
	}
}

BOOL CNILayer::Send(u_char *ppayload, int nlength)
{
	if (pcap_sendpacket(m_AdapterObject, ppayload, nlength))
	{
		AfxMessageBox("패킷 전송 실패");
		return FALSE;
	}
	return TRUE;
}

BOOL CNILayer::Receive(u_char* ppayload, int len)
{
	BOOL bSuccess = FALSE;

	u_char *temp = (u_char*)malloc(sizeof(u_char)*len);
	memset(temp, '\0', len);
	temp = SubStr((char*)ppayload, len);

	// 상위 계층으로 payload 올림
	bSuccess = mp_aUpperLayer[0]->Receive(temp);
	return bSuccess;
}

u_char* CNILayer::SubStr(const char *pnInput, int nLen)
{
	int nLoop;

	u_char *pszOutPut;

	if (pnInput == NULL) {
		return NULL;
	}

	pszOutPut = (u_char *)malloc(sizeof(u_char) * nLen + 1);
	for (nLoop = 0; nLoop < nLen; nLoop++) {
		pszOutPut[nLoop] = pnInput[nLoop];
	}
	pszOutPut[nLen] = '\0';

	return pszOutPut;
}


UINT CNILayer::ReadingThread(LPVOID pParam)
{
	struct pcap_pkthdr *header;
	const u_char *pkt_data;
	int result;

	AfxBeginThread(FileTransferThread, (LPVOID)pParam);
	CNILayer *pNI = (CNILayer *)pParam;

	while (pNI->m_thrdSwitch) // 패킷 체크
	{
		// 패킷 읽어오기
		result = pcap_next_ex(pNI->m_AdapterObject, &header, &pkt_data);

		if (result == 0) {
			//	AfxMessageBox("패킷 없음");
		}
		else if (result == 1) {
			//	AfxMessageBox("패킷 있음");
			pNI->Receive((u_char *)pkt_data, header->len);
		}
		else if (result<0) {
			//	AfxMessageBox("패킷 오류");
		}
	}

	return 0;
}

UINT CNILayer::FileTransferThread(LPVOID pParam)
{
	CNILayer *pNI = (CNILayer *)pParam;

	return 0;
}

CString CNILayer::GetNICardAddress(char* adapter_name)
{
	PPACKET_OID_DATA OidData;
	LPADAPTER lpAdapter;

	OidData = (PPACKET_OID_DATA)malloc(6 + sizeof(PACKET_OID_DATA));
	if (OidData == NULL)
	{
		return "None";
	}

	OidData->Oid = OID_802_3_CURRENT_ADDRESS;
	OidData->Length = 6;
	ZeroMemory(OidData->Data, 6);

	lpAdapter = PacketOpenAdapter(adapter_name);

	CString NICardAddress;

	if (PacketRequest(lpAdapter, FALSE, OidData))
	{
		NICardAddress.Format("%.2x:%.2x:%.2x:%.2x:%.2x:%.2x",
			(OidData->Data)[0],
			(OidData->Data)[1],
			(OidData->Data)[2],
			(OidData->Data)[3],
			(OidData->Data)[4],
			(OidData->Data)[5]);
	}

	PacketCloseAdapter(lpAdapter);
	free(OidData);
	return NICardAddress;
}

LPIP_ADDR CNILayer::GetIpAddress(u_char* ether)
{
	// IP_ADAPTER_INFO struct를 사용하여, IP를 가져온다.
	// 참고 : IP Help Api

	IP_ADAPTER_INFO  *pAdapterInfo;
	IP_ADAPTER_INFO  *pAdapter;
	ULONG  ulOutBufLen;
	DWORD  dwRetVal;

	pAdapterInfo = (IP_ADAPTER_INFO *)malloc(sizeof(IP_ADAPTER_INFO));
	if (pAdapterInfo == NULL) {
		printf("Error allocating memory needed to call GetAdapterInfo\n");
		return NULL;
	}
	ulOutBufLen = sizeof(IP_ADAPTER_INFO);

	if (GetAdaptersInfo(pAdapterInfo, &ulOutBufLen) == ERROR_BUFFER_OVERFLOW) {
		free(pAdapterInfo);
		pAdapterInfo = (IP_ADAPTER_INFO *)malloc(ulOutBufLen);
		if (pAdapterInfo == NULL) {
			printf("Error allocating memory needed to call GetAdapterInfo\n");
			return NULL;
		}
	}

	if ((dwRetVal = GetAdaptersInfo(pAdapterInfo, &ulOutBufLen)) != NO_ERROR) {
		printf("GetAdaptersInfo call failed with %d\n", dwRetVal);
		return NULL;
	}
	pAdapter = pAdapterInfo;

	CString strIpAddress;
	while (pAdapter) {
		if (!memcmp(pAdapter->Address, ether, 6))
		{
			memcpy((void *)strIpAddress.GetString(), pAdapter->IpAddressList.IpAddress.String, 16);
			break;
		}
		pAdapter = pAdapter->Next;
	}

	// 위에서 얻어온 IP주소를 구조에 맞게 포맷 변환
	CString cstr;
	IP_ADDR my_ipaddr;
	for (int i = 0; i<4; i++)
	{
		AfxExtractSubString(cstr, strIpAddress, i, '.');
		my_ipaddr.addrs_i[i] = (u_char)strtoul(cstr.GetString(), NULL, 10);
	}

	return &my_ipaddr;
}