📄 initusbdisktestdlg.cpp

📁 实现了对U盘里存储内容的加密
💻 CPP
📖 第 1 页 / 共 2 页
字号:
12 下一页
// InitUsbDiskTestDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "InitUsbDiskTest.h"
#include "InitUsbDiskTestDlg.h"
#include ".\initusbdisktestdlg.h"

#include "JasPrivacyOutLib.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


#define DISK_BIGSECTOR_SIZE                             512 * 8


typedef BOOL (* PFINDUSBDISKBOXNUMBER)(DWORD *);
typedef BOOL (* PGETLASTUSBDISKBOXPHYSICALDEVICENUMBER)(DWORD *);

typedef BOOL (* PINITUSBDISKDRIVER)(int, BYTE *, int, int, BYTE *, int, BYTE *, int, int *, int *, DWORD *);
typedef BOOL (* PUNINITUSBDISKDRIVER)(int, BYTE *, int, int, BYTE *, int, BYTE *, int, int *, int *, DWORD *);
typedef BOOL (* PCHECKINITUSBDISKDRIVER)(int, int, DWORD *);
typedef BOOL (* PBUILDSECTRYUSBDISKDRIVER)(int, BOOL, BYTE *, int, int);

typedef BOOL (* PCONTRLDISKSYSTEMFILTERDRIVER)(BOOL);
typedef BOOL (* PCONTRLFILESYSTEMFILTERDRIVER)(BOOL);

typedef BOOL (* PQUERYDISKSYSTEMFILTERDRIVER)(DWORD *);
typedef BOOL (* PQUERYFILESYSTEMFILTERDRIVER)(DWORD *);

typedef BOOL (* PSETUSBHARDDEVICESMID)(int, DWORD, int);
typedef DWORD (* PGETUSBHARDDEVICESMID)(int, int, DWORD *);

typedef BOOL (* VERIFYWRITEUSBDISKDATA)(int, int, BYTE *, int, BYTE *, int, int *, int *);

typedef BOOL (* CONTROLJASLOCKCLIENT)(BOOL);


PFINDUSBDISKBOXNUMBER FindUsbDiskBoxNumber;
PGETLASTUSBDISKBOXPHYSICALDEVICENUMBER GetLastUsbDiskBoxPhysicalDeviceNumber;

PINITUSBDISKDRIVER InitUsbDiskDriver;
PUNINITUSBDISKDRIVER UnInitUsbDiskDriver;
PCHECKINITUSBDISKDRIVER CheckInitUsbDiskDriver;
PBUILDSECTRYUSBDISKDRIVER BuildSectryUsbDiskDriver;

PCONTRLDISKSYSTEMFILTERDRIVER ContrlDiskSystemFilterDriver;
PCONTRLFILESYSTEMFILTERDRIVER ContrlFileSystemFilterDriver;

PQUERYDISKSYSTEMFILTERDRIVER QueryDiskSystemDriverState;
PQUERYFILESYSTEMFILTERDRIVER QueryFileSystemDriverState;

PSETUSBHARDDEVICESMID SetUsbHardDeviceSMID;
PGETUSBHARDDEVICESMID GetUsbHardDeviceSMID;

VERIFYWRITEUSBDISKDATA VerifyWriteUsbDiskData;

CONTROLJASLOCKCLIENT ControlJasLockClient;


typedef BOOL (* ENCRYPTUSBFILE)(char *, char *, BYTE *, int);

ENCRYPTUSBFILE EncryptUsbFile;


BYTE key[32] = {0x1f, 0xf1, 0x01, 0x10, 0xdf, 0xca, 0x19, 0x2a, 
                0xc2, 0x31, 0x02, 0x1a, 0xb9, 0xc0, 0x89, 0xfa,
                0x21, 0xd7, 0x31, 0x2f, 0xfe, 0xff, 0xe1, 0x3b, 
                0xc2, 0x09, 0xc2, 0x8a, 0x19, 0xec, 0xff, 0x12};


void DoEvents(void)
{
	MSG msg0;
	
	if(::PeekMessage(&msg0, NULL, 0, 0, PM_REMOVE))
	{
		::TranslateMessage(&msg0);
		::DispatchMessage(&msg0);
	}
}

void Delay(unsigned int tm)
{
	unsigned int j0, j1;

	j0 = GetTickCount();
	
	while(true)
	{
		j1 = GetTickCount();
		if(j1 - j0 > tm)
		{
			break;
		}
        DoEvents();
	}
}


// CInitUsbDiskTestDlg 对话框


CInitUsbDiskTestDlg::CInitUsbDiskTestDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CInitUsbDiskTestDlg::IDD, pParent)
    , m_type(0)
    , m_nb0(0)
    , m_nb1(0)
    , m_pdnb(0)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CInitUsbDiskTestDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    DDX_Text(pDX, IDC_EDIT1, m_type);
    DDV_MinMaxInt(pDX, m_type, 0, 3);
    DDX_Text(pDX, IDC_EDIT2, m_nb0);
    DDX_Text(pDX, IDC_EDIT3, m_nb1);
    DDX_Text(pDX, IDC_EDIT4, m_pdnb);
}

BEGIN_MESSAGE_MAP(CInitUsbDiskTestDlg, CDialog)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
    ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
    ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
    ON_BN_CLICKED(IDC_BUTTON3, OnBnClickedButton3)
    ON_BN_CLICKED(IDC_BUTTON4, OnBnClickedButton4)
    ON_BN_CLICKED(IDC_BUTTON5, OnBnClickedButton5)
    ON_BN_CLICKED(IDC_BUTTON6, OnBnClickedButton6)
    ON_BN_CLICKED(IDC_BUTTON7, OnBnClickedButton7)
    ON_BN_CLICKED(IDC_BUTTON8, OnBnClickedButton8)
    ON_BN_CLICKED(IDC_BUTTON9, OnBnClickedButton9)
    ON_BN_CLICKED(IDC_BUTTON10, OnBnClickedButton10)
    ON_BN_CLICKED(IDC_BUTTON11, OnBnClickedButton11)
    ON_BN_CLICKED(IDC_BUTTON13, OnBnClickedButton13)
    ON_BN_CLICKED(IDC_BUTTON12, OnBnClickedButton12)
    ON_BN_CLICKED(IDC_BUTTON14, OnBnClickedButton14)
    ON_BN_CLICKED(IDC_BUTTON15, OnBnClickedButton15)
    ON_BN_CLICKED(IDC_BUTTON16, OnBnClickedButton16)
    ON_BN_CLICKED(IDC_BUTTON17, OnBnClickedButton17)
    ON_BN_CLICKED(IDC_BUTTON18, OnBnClickedButton18)
    ON_BN_CLICKED(IDC_BUTTON19, OnBnClickedButton19)
END_MESSAGE_MAP()

// CInitUsbDiskTestDlg 消息处理程序

BOOL CInitUsbDiskTestDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码

    if(hDll = LoadLibrary(_T("InitUsbDisk.dll")))
	{
	    FindUsbDiskBoxNumber = (PFINDUSBDISKBOXNUMBER)GetProcAddress(hDll, _T("FindUsbDiskBoxNumber"));
	    GetLastUsbDiskBoxPhysicalDeviceNumber = (PGETLASTUSBDISKBOXPHYSICALDEVICENUMBER)GetProcAddress(hDll, 
                                                                _T("GetLastUsbDiskBoxPhysicalDeviceNumber"));

	    InitUsbDiskDriver = (PINITUSBDISKDRIVER)GetProcAddress(hDll, _T("InitUsbDiskDriver"));
	    UnInitUsbDiskDriver = (PUNINITUSBDISKDRIVER)GetProcAddress(hDll, _T("UnInitUsbDiskDriver"));
	    BuildSectryUsbDiskDriver = (PBUILDSECTRYUSBDISKDRIVER)GetProcAddress(hDll, _T("BuildSectryUsbDiskDriver"));
	    
        ContrlDiskSystemFilterDriver = (PCONTRLDISKSYSTEMFILTERDRIVER)GetProcAddress(hDll, _T("ContrlDiskSystemFilterDriver"));
	    ContrlFileSystemFilterDriver = (PCONTRLFILESYSTEMFILTERDRIVER)GetProcAddress(hDll, _T("ContrlFileSystemFilterDriver"));
	    QueryDiskSystemDriverState = (PQUERYDISKSYSTEMFILTERDRIVER)GetProcAddress(hDll, _T("QueryDiskSystemDriverState"));
	    QueryFileSystemDriverState = (PQUERYFILESYSTEMFILTERDRIVER)GetProcAddress(hDll, _T("QueryFileSystemDriverState"));

	    CheckInitUsbDiskDriver = (PCHECKINITUSBDISKDRIVER)GetProcAddress(hDll, _T("CheckInitUsbDiskDriver"));
	    SetUsbHardDeviceSMID = (PSETUSBHARDDEVICESMID)GetProcAddress(hDll, _T("SetUsbHardDeviceSMID"));
	    GetUsbHardDeviceSMID = (PGETUSBHARDDEVICESMID)GetProcAddress(hDll, _T("GetUsbHardDeviceSMID"));
	    VerifyWriteUsbDiskData = (VERIFYWRITEUSBDISKDATA)GetProcAddress(hDll, _T("VerifyWriteUsbDiskData"));
	    ControlJasLockClient = (CONTROLJASLOCKCLIENT)GetProcAddress(hDll, _T("ControlJasLockClient"));

        if(!FindUsbDiskBoxNumber || !GetLastUsbDiskBoxPhysicalDeviceNumber || !InitUsbDiskDriver || !UnInitUsbDiskDriver || 
           !BuildSectryUsbDiskDriver || !ContrlDiskSystemFilterDriver || !ContrlFileSystemFilterDriver || 
           !CheckInitUsbDiskDriver || !SetUsbHardDeviceSMID || !GetUsbHardDeviceSMID || !VerifyWriteUsbDiskData || 
           !ControlJasLockClient || !QueryDiskSystemDriverState || !QueryFileSystemDriverState)
	    {
            ::AfxMessageBox("InitUsbDisk.dll库文件中，函数获取失败。");    
            OnCancel();
	    }
	}
    else
    {
        ::AfxMessageBox("InitUsbDisk.dll库文件加载失败。");
        OnCancel();
    }

    if(hDll1 = LoadLibrary(_T("JasPrivacyOutDll.dll")))
	{
	    EncryptUsbFile = (ENCRYPTUSBFILE)GetProcAddress(hDll1, _T("EncryptUsbRemovDiskFile"));

        if(!EncryptUsbFile)
	    {
            ::AfxMessageBox("JasPrivacyOutDll.dll库文件中，函数获取失败。");    
            OnCancel();
	    }
	}
    else
    {
        ::AfxMessageBox("JasPrivacyOutDll.dll库文件加载失败。");
        OnCancel();
    }

    m_type = 0;
    m_pdnb = 0;

    UpdateData(FALSE);
	
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CInitUsbDiskTestDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CInitUsbDiskTestDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CInitUsbDiskTestDlg::OnBnClickedButton1()
{
    // TODO: 在此添加控件通知处理程序代码

    BYTE sec_data0[DISK_BIGSECTOR_SIZE];
    BYTE sec_data1[DISK_BIGSECTOR_SIZE];
    int sec_pos0, sec_pos1;
    DWORD user_cancel;

    UpdateData(TRUE);

    ::ZeroMemory(sec_data0, sizeof(sec_data0));
    ::ZeroMemory(sec_data1, sizeof(sec_data1));

    // 0-普通, 1-海泰, 2-中正, 3-亚略特
    if(InitUsbDiskDriver(m_pdnb, key, sizeof(key), m_type, sec_data0, sizeof(sec_data0), 
                         sec_data1, sizeof(sec_data1), &sec_pos0, &sec_pos1, &user_cancel))
    {
        ::AfxMessageBox("USB盘初始化加密成功。");

        Delay(1000);

        if(VerifyWriteUsbDiskData(m_pdnb, m_type, sec_data0, sizeof(sec_data0), 
                                  sec_data1, sizeof(sec_data1), &sec_pos0, &sec_pos1))
        {
            ::AfxMessageBox("USB盘数据写入校验成功。");
        }
        else
        {
            ::AfxMessageBox("USB盘数据写入校验失败。");
        }
    }
    else
    {
        if(user_cancel != 0)
        {
            ::AfxMessageBox("USB盘初始化操作已经被用户取消。");
        }
        else
        {
            ::AfxMessageBox("USB盘初始化加密失败。");
        }
    }
}

void CInitUsbDiskTestDlg::OnBnClickedButton2()
{
    // TODO: 在此添加控件通知处理程序代码

    BYTE sec_data0[DISK_BIGSECTOR_SIZE];
    BYTE sec_data1[DISK_BIGSECTOR_SIZE];
    int sec_pos0, sec_pos1;
    DWORD user_cancel;

    UpdateData(TRUE);

    ::ZeroMemory(sec_data0, sizeof(sec_data0));
    ::ZeroMemory(sec_data1, sizeof(sec_data1));

    // 0-普通, 1-海泰, 2-中正, 3-亚略特
    if(UnInitUsbDiskDriver(m_pdnb, key, sizeof(key), m_type, sec_data0, sizeof(sec_data0), 
                           sec_data1, sizeof(sec_data1), &sec_pos0, &sec_pos1, &user_cancel))
    {
        ::AfxMessageBox("USB盘初始化加密 - 解除成功。");

        Delay(1000);

        if(VerifyWriteUsbDiskData(m_pdnb, m_type, sec_data0, sizeof(sec_data0), 
                                  sec_data1, sizeof(sec_data1), &sec_pos0, &sec_pos1))
        {
            ::AfxMessageBox("USB盘数据写入校验成功。");
        }
        else
        {
            ::AfxMessageBox("USB盘数据写入校验失败。");
        }
    }
    else
    {
        if(user_cancel != 0)
        {
            ::AfxMessageBox("USB盘解除初始化操作已经被用户取消。");
        }
        else
        {
            ::AfxMessageBox("USB盘解除初始化失败。");
        }
    }
}

void CInitUsbDiskTestDlg::OnBnClickedButton3()
{
    // TODO: 在此添加控件通知处理程序代码
12 下一页
💿 文件大小 62 K
👤 上传用户 hufei108
📂 所属分类 USB编程
🏷️ 相关标签

#U盘 #存储 #加密
⌨️ 快捷键说明

复制代码 Ctrl + C
搜索代码 Ctrl + F
全屏模式 F11
切换主题 Ctrl + Shift + D
显示快捷键 ?
增大字号 Ctrl + =
减小字号 Ctrl + -