void CDevicesDlg::OnNMDblclkListDevices(NMHDR *pNMHDR, LRESULT *pResult)
{
LPNMITEMACTIVATE pNMItemActivate = reinterpret_cast<LPNMITEMACTIVATE>(pNMHDR);
// TODO: 在此添加控件通知处理程序代码
BindDevice(pNMItemActivate->iItem, true);
*pResult = 0;
}
void CDevicesDlg::OnBnClickedButtonAdd()
{
int nCur = m_listDevs.GetItemCount();
const char* uuid = newGUID();
int n = m_listDevs.InsertItem(nCur, "");
m_listDevs.SetItemText(n, 1, uuid);
ASSERT(m_Devs[n].Index == 0 && m_Devs[n].CreateTime == 0);
m_Devs[n].Index = -1;
m_Devs[n].CreateTime = time(NULL);
m_Devs[n].ValidTime = 0;
sprintf(m_Devs[n].SN, "%s", uuid);
m_listDevs.SetItemState(n, LVNI_FOCUSED | LVIS_SELECTED, LVNI_FOCUSED | LVIS_SELECTED);
BindDevice(nCur, true);
}
BOOL CDevicesDlg::OnInitDialog()
{
CDialogEx::OnInitDialog();
m_listDevs.InsertColumn(0, "Name", LVCFMT_LEFT, 120);
m_listDevs.InsertColumn(1, "SN", LVCFMT_LEFT, 240);
m_listDevs.InsertColumn(2, "Desc", LVCFMT_LEFT, 240);
m_listDevs.SetExtendedStyle(m_listDevs.GetExtendedStyle() | LVS_EX_FULLROWSELECT);
memset(m_Devs, 0, sizeof(DeviceInfo) * 2048);
int num = m_pAccApi->ListDevices(m_Devs, 2048, false);
for (int i = 0; i < num; ++i) {
int n = m_listDevs.InsertItem(i, m_Devs[i].Name);
m_listDevs.SetItemText(n, 1, m_Devs[i].SN);
m_listDevs.SetItemText(n, 2, m_Devs[i].Desc);
}
m_CurSel = -2;
BindDevice(-1, false);
return TRUE; // return TRUE unless you set the focus to a control
// 异常: OCX 属性页应返回 FALSE
}
void test(const char* exp, const char* dev)
{
int sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_IP));
if (sock<=0)
{
P("socket error");
return;
}
//设置缓冲区
#define PER_PACKET_SIZE 2048
const int BUFFER_SIZE = 1024*1024*16; //16MB的缓冲区
struct tpacket_req req;
req.tp_block_size = 4096;
req.tp_block_nr = BUFFER_SIZE/req.tp_block_size;
req.tp_frame_size = PER_PACKET_SIZE;
req.tp_frame_nr = BUFFER_SIZE/req.tp_frame_size;
if (-1==setsockopt(sock, SOL_PACKET, PACKET_RX_RING, &req, sizeof(struct tpacket_req)))
{
perror("setsockopt");
P("set PACKET_RX_RING error");
close(sock);
return;
}
//映射缓冲区
char* pBuffer = (char*)mmap(0, BUFFER_SIZE, PROT_READ|PROT_WRITE, MAP_SHARED, sock, 0);
if (MAP_FAILED==pBuffer)
{
P("mmap error");
close(sock);
return;
}
//注意：一定要先映射后再绑定，否则会有问题
// 问题的详细描述可参考：https://lists.linux-foundation.org/pipermail/bugme-new/2003-October/009110.html
if (!SetPromisc(sock, dev))
{
P("SetPromisc [%s] error", dev);
memset(&req, 0, sizeof(req));
if (-1==setsockopt(sock, SOL_PACKET, PACKET_RX_RING, &req, sizeof(req)))
{
P("set map buffer to 0 error!");
}
close(sock);
return;
}
if (!BindDevice(sock, dev))
{
P("bind [%s] error", dev);
memset(&req, 0, sizeof(req));
if (-1==setsockopt(sock, SOL_PACKET, PACKET_RX_RING, &req, sizeof(req)))
{
P("set map buffer to 0 error!");
}
close(sock);
return;
}
//设置过滤器
int nExpLen = strlen(exp);
struct sock_fprog Filter;
memset(&Filter, 0, sizeof(struct sock_fprog));
if (nExpLen>0)
{
if (ExpressionToFilter(exp, &Filter))
{
if (setsockopt(sock, SOL_SOCKET, SO_ATTACH_FILTER,
&Filter, sizeof(Filter))<0)
{
perror("setsockopt");
}
}
}
//
struct pollfd pfd;
int nIndex = 0;
for (; ; )
{
for (; ; )
{
struct tpacket_hdr* pHead = (struct tpacket_hdr*)(pBuffer + nIndex*PER_PACKET_SIZE);
if (pHead->tp_len<34 || pHead->tp_len>1614)
{
break;
}
PrintPacket((char*)pHead+pHead->tp_net);
pHead->tp_len = 0;
pHead->tp_status = TP_STATUS_KERNEL;
//注意：pHead->tp_status这个变量并不能真正反应出包的处理状态，
//在有的服务器上 TP_STATUS_USER(1)代表包可用, TP_STATUS_KERNEL(0)代表包不可用
//但是在有的服务器上，这个标志变量无效
//对于这个问题我还没找到原因
nIndex++;
if (nIndex>=BUFFER_SIZE/PER_PACKET_SIZE)
{
nIndex = 0;
}
}
//
pfd.fd = sock;
pfd.events = POLLIN | POLLERR;
pfd.revents = 0;
switch (poll(&pfd, 1, 1000))
{
case -1:
perror("poll");
P("poll error");
goto EndWhile;
break;
case 0:
P("time out");
continue;
break;
}
}
EndWhile:
if (-1==munmap(pBuffer, BUFFER_SIZE))
{
P("unmap error!");
}
memset(&req, 0, sizeof(req));
if (-1==setsockopt(sock, SOL_PACKET, PACKET_RX_RING, &req, sizeof(req)))
{
P("set map buffer to 0 error!");
}
close(sock);
sock = -1;
//
if (nExpLen>0)
{
FreeFilter(&Filter);
}
}
void CDevicesDlg::OnLvnItemchangedListDevices(NMHDR *pNMHDR, LRESULT *pResult)
{
LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
BindDevice(pNMLV->iItem, false);
*pResult = 0;
}
