static int encrypt_file(void)
{
struct enc_param ep;
ssize_t src_len;
unsigned char *buf;
uint32_t hdrlen;
ssize_t totlen = 0;
int err;
int ret = -1;
src_len = get_file_size(ifname);
if (src_len < 0) {
ERR("unable to get size of '%s'", ifname);
goto out;
}
totlen = enc_compute_buf_len(product, version, src_len);
hdrlen = enc_compute_header_len(product, version);
buf = malloc(totlen);
if (buf == NULL) {
ERR("no memory for the buffer");
goto out;
}
err = read_file_to_buf(ifname, &buf[hdrlen], src_len);
if (err) {
ERR("unable to read from file '%s'", ofname);
goto free_buf;
}
memset(&ep, '\0', sizeof(ep));
ep.key = (unsigned char *) crypt_key;
ep.seed = seed;
ep.longstate = longstate;
ep.csum = buffalo_csum(src_len, &buf[hdrlen], src_len);
ep.datalen = src_len;
strcpy((char *) ep.magic, magic);
strcpy((char *) ep.product, product);
strcpy((char *) ep.version, version);
err = encrypt_buf(&ep, buf, &buf[hdrlen]);
if (err) {
ERR("invalid input file");
goto free_buf;
}
err = write_buf_to_file(ofname, buf, totlen);
if (err) {
ERR("unable to write to file '%s'", ofname);
goto free_buf;
}
ret = 0;
free_buf:
free(buf);
out:
return ret;
}
static int decrypt_file(void)
{
struct enc_param ep;
ssize_t src_len;
unsigned char *buf = NULL;
int err;
int ret = -1;
src_len = get_file_size(ifname);
if (src_len < 0) {
ERR("unable to get size of '%s'", ifname);
goto out;
}
buf = malloc(src_len);
if (buf == NULL) {
ERR("no memory for the buffer");
goto out;
}
err = read_file_to_buf(ifname, buf, src_len);
if (err) {
ERR("unable to read from file '%s'", ifname);
goto out;
}
memset(&ep, '\0', sizeof(ep));
ep.key = (unsigned char *) crypt_key;
ep.longstate = longstate;
err = decrypt_buf(&ep, buf, src_len);
if (err) {
ERR("unable to decrypt '%s'", ifname);
goto out;
}
printf("Magic\t\t: '%s'\n", ep.magic);
printf("Seed\t\t: 0x%02x\n", ep.seed);
printf("Product\t\t: '%s'\n", ep.product);
printf("Version\t\t: '%s'\n", ep.version);
printf("Data len\t: %u\n", ep.datalen);
printf("Checksum\t: 0x%08x\n", ep.csum);
err = write_buf_to_file(ofname, buf, ep.datalen);
if (err) {
ERR("unable to write to file '%s'", ofname);
goto out;
}
ret = 0;
out:
free(buf);
return ret;
}
LRESULT CALLBACK stream_WndProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
RECT rect ;
int cxClient, cyClient;
static HMENU hMenu ;
POINT point ;
int ret, idx;
TCHAR buf[64];
char file_name[MAX_FILE_PATH_LEN];
static int edit_iItem=-1 ;
static int edit_iSubItem;
LVITEM lv_item;
switch (message)
{
case WM_CREATE:
hwnd_stream= hwnd;
hMenu = LoadMenu (g_hInstance, TEXT("my_popup_menu")) ;
hMenu = GetSubMenu (hMenu, 0) ;
hwnd_dynamic_edit=CreateWindow (TEXT("edit"), TEXT(""),
WS_CHILD|ES_AUTOHSCROLL,
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
hwnd, (HMENU)ID_DYNAMIC_EDIT, g_hInstance, NULL) ;
SendMessage(hwnd_dynamic_edit, WM_SETFONT, (WPARAM)char_font_2, 0);
hwnd_lv = CreateListView(hwnd);
//InitListViewImageLists(hwnd_lv);
InitListViewColumns(hwnd_lv);
lv_row_color_init();
old_lv_proc = (WNDPROC) SetWindowLong (hwnd_lv,
GWL_WNDPROC, (LONG)my_lv_proc) ;
ShowWindow(hwnd_lv, 1) ;
refresh_window(hwnd_lv) ;
add_tip(hwndTip, hwnd_lv, TEXT("点击鼠标右键进行操作"));
return 0 ;
case WM_SIZE:
cxClient = LOWORD (lParam) ;
cyClient = HIWORD (lParam) ;
MoveWindow(hwnd_lv, 0, 0,
cxClient, cyClient, TRUE) ;
return 0 ;
case WM_NOTIFY:
{
NMHDR *pt_nmhdr=(void *)lParam;
switch(LOWORD(wParam))
{
case ID_LV:
// if code == NM_CLICK - Single click on an item
if(pt_nmhdr->code == NM_CLICK || pt_nmhdr->code == NM_DBLCLK)
{
int iItem = ((LPNMITEMACTIVATE)lParam)->iItem;
int iSubItem=((LPNMITEMACTIVATE)lParam)->iSubItem;
if (iItem>=0 && ((iSubItem>=3 && iSubItem<=5) || iSubItem==7))
{
ListView_GetSubItemRect(hwnd_lv,iItem,iSubItem,LVIR_LABEL,&rect);
ListView_GetItemText(hwnd_lv, iItem, iSubItem, buf, sizeof(buf));
MoveWindow (hwnd_dynamic_edit
,rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, TRUE);
SetWindowText(hwnd_dynamic_edit, buf);
ShowWindow (hwnd_dynamic_edit, 1) ;
SetFocus(hwnd_dynamic_edit);
SendMessage(hwnd_dynamic_edit, EM_SETSEL, (WPARAM)0, (LPARAM)-1);
edit_iItem = iItem;
edit_iSubItem = iSubItem;
}
return 0;
}
else if (pt_nmhdr->code == NM_RCLICK)
{
point = ((LPNMITEMACTIVATE)lParam)->ptAction;
ListView_GetItem(hwnd_lv, &lv_item);
ClientToScreen (hwnd_lv, &point) ;
TrackPopupMenu (hMenu, TPM_LEFTBUTTON, point.x, point.y, 0, hwnd_stream, NULL) ;
return 0;
}
else if (pt_nmhdr->code == LVN_ITEMCHANGED)
{
if (!init_over) break;
if (lv_in_op) break;
//if((((LPNMLISTVIEW)lParam)->uOldState&LVIS_STATEIMAGEMASK) !=
// (((LPNMLISTVIEW)lParam)->uNewState&LVIS_STATEIMAGEMASK))
{
int iItem = ((LPNMITEMACTIVATE)lParam)->iItem;
int selected = ListView_GetCheckState(hwnd_lv, iItem);
if (g_apt_streams[iItem]->selected!=selected)
{
doc_modified=1;
g_apt_streams[iItem]->selected=selected;
update_statusbar();
}
}
return 0;
}
break;
}
break;
}
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDM_STREAM_NEW:
init_stream(>_edit_stream);
gt_edit_stream.len=gat_sample_pkts[0].len;
memcpy(gt_edit_stream.data, gat_sample_pkts[0].pkt_data, gt_edit_stream.len);
ret=DialogBox(g_hInstance, TEXT("STREAM_EDIT_DLG"), hwnd, StreamEditDlgProc);
if (IDOK==ret)
{
add_stream(>_edit_stream);
re_populate_items();
}
return 0 ;
case IDM_STREAM_NEW_HEX:
{
int len;
char buf[MAX_PACKET_LEN];
if (get_open_file_name(file_name, hwnd, ALL_FILE_FILTER))
return 0;
len = read_file_to_buf(buf, sizeof(buf)-1, file_name);
if (len>0)
add_stream_from_hex_text(buf, len);
else
err_msg_box("读取文件内容失败");
return 0 ;
}
case IDM_STREAM_NEW_BIN:
{
int len;
char buf[MAX_PACKET_LEN];
if (get_open_file_name(file_name, hwnd, ALL_FILE_FILTER))
return 0;
len = read_file_to_buf(buf, sizeof(buf)-1, file_name);
if (len>0)
add_stream_from_raw_data(buf, len);
else
err_msg_box("读取文件内容失败");
return 0 ;
}
case IDM_STREAM_NEW_PCAP:
{
if (get_open_file_name(file_name, hwnd, PCAP_FILE_FILTER))
return 0;
add_stream_from_pcap(file_name);
return 0 ;
}
case IDM_STREAM_DEL:
idx=GetIndex(hwnd_lv);
//ListView_DeleteItem(hwnd_lv, idx);
delete_stream(idx);
re_populate_items();
update_statusbar();
return 0 ;
case IDM_STREAM_EDIT:
cur_strm_idx=GetIndex(hwnd_lv);
if (cur_strm_idx<0) return 0;
cpy_stream(>_edit_stream, g_apt_streams[cur_strm_idx]);
ret=DialogBox(g_hInstance, TEXT("STREAM_EDIT_DLG"), hwnd, StreamEditDlgProc);
if (IDOK==ret)
{
doc_modified=1;
cpy_stream(g_apt_streams[cur_strm_idx], >_edit_stream);
re_populate_items();
}
//ListView_DeleteAllItems(hwnd_lv);
return 0 ;
case IDM_STREAM_DEL_SEL:
//DelSel(hwnd_lv);
delete_sel_stream();
re_populate_items();
update_statusbar();
return 0 ;
case IDM_STREAM_SEL_ALL:
SelAll(hwnd_lv);
return 0 ;
case IDM_STREAM_SEL_RVS:
SelRvs(hwnd_lv);
return 0 ;
case IDM_STREAM_COPY:
copy_idx = GetIndex(hwnd_lv);
return 0 ;
case IDM_STREAM_PASTE:
cpy_stream(>_edit_stream, g_apt_streams[copy_idx]);
add_stream(>_edit_stream);
re_populate_items();
return 0 ;
case IDM_STREAM_MAKE_FRAGS:
{
ret=DialogBox(g_hInstance, TEXT("FRAG_DLG"), hwnd, FragDlgProc);
return 0 ;
}
case IDM_STREAM_SEL2PCAP:
{
ret=get_save_file_name(file_name, hwnd, PCAP_FILE_FILTER, PCAP_FILE_SUFFIX);
if (ret) return 0 ;
stream2dump(file_name);
update_pcap_file_history(file_name);
return 0 ;
}
case IDM_STREAM_2_BIN:
{
ret=get_save_file_name(file_name, hwnd, BIN_FILE_FILTER, BIN_FILE_SUFFIX);
if (ret) return 0 ;
stream_2_bin(file_name);
return 0 ;
}
case IDM_STREAM_2_TEXT:
{
ret=get_save_file_name(file_name, hwnd, TEXT_FILE_FILTER, TEXT_FILE_SUFFIX);
if (ret) return 0 ;
stream_2_text(file_name);
return 0 ;
}
case ID_DYNAMIC_EDIT:
if (HIWORD(wParam)==EN_KILLFOCUS)
{
update_grid_from_edit(edit_iItem, edit_iSubItem);
edit_iItem=-1;
return 0 ;
}
}
break;
case WM_KEYDOWN:
if (VK_RETURN==wParam)
{
SetFocus(hwnd);
return 0;
}
break;
case WM_INITMENUPOPUP:
{
int idx=GetIndex(hwnd_lv);
t_stream *pt_stream = g_apt_streams[idx];
int sel_cnt=GetSelCnt(hwnd_lv);
int item_cnt=ListView_GetItemCount(hwnd_lv);
if (lParam == 0)
{
UINT add_stream_menu_state = nr_cur_stream<MAX_STREAM_NUM ? MF_ENABLED : MF_GRAYED;
EnableMenuItem ((HMENU) wParam, IDM_STREAM_NEW, add_stream_menu_state);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_NEW_HEX, add_stream_menu_state);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_NEW_BIN, add_stream_menu_state);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_NEW_PCAP, add_stream_menu_state);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_EDIT, idx>=0 ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_DEL, idx>=0 ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_COPY, idx>=0 ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_PASTE, copy_idx>=0 && item_cnt>copy_idx ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_DEL_SEL, sel_cnt ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_SEL_ALL, item_cnt ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_SEL_RVS, item_cnt ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_MAKE_FRAGS, idx>=0&&stream_fragable(pt_stream) ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_SEL2PCAP, sel_cnt ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_2_BIN, idx>=0 ? MF_ENABLED : MF_GRAYED);
EnableMenuItem ((HMENU) wParam, IDM_STREAM_2_TEXT, idx>=0 ? MF_ENABLED : MF_GRAYED);
return 0;
}
break;
}
}
return DefWindowProc (hwnd, message, wParam, lParam) ;
}
