windows创建虚拟通道

Posted 2022-03-10 道亦无名

TestVChannelDlg.cpp代码如下：

// TestVChannelDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "TestVChannel.h"
#include "TestVChannelDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


#include <wtsapi32.h>
//#include <tchar.h>
//#include <lmcons.h>

#include <string>
using namespace std;

#pragma comment( lib,"Wtsapi32.lib" )



// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialogEx

public:
	CAboutDlg();

// 对话框数据
#ifdef AFX_DESIGN_TIME
	enum  IDD = IDD_ABOUTBOX ;
#endif

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
;

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)



void CAboutDlg::DoDataExchange(CDataExchange* pDX)

	CDialogEx::DoDataExchange(pDX);


BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CTestVChannelDlg 对话框



CTestVChannelDlg::CTestVChannelDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(IDD_TESTVCHANNEL_DIALOG, pParent)

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);


void CTestVChannelDlg::DoDataExchange(CDataExchange* pDX)

	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_LIST1, m_listbox1);


BEGIN_MESSAGE_MAP(CTestVChannelDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDOK, &CTestVChannelDlg::OnBnClickedOk)
	ON_BN_CLICKED(IDCANCEL, &CTestVChannelDlg::OnBnClickedCancel)
END_MESSAGE_MAP()


// CTestVChannelDlg 消息处理程序

BOOL CTestVChannelDlg::OnInitDialog()

	CDialogEx::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		
	

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

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

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE


void CTestVChannelDlg::OnSysCommand(UINT nID, LPARAM lParam)

	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	
	else
	
		CDialogEx::OnSysCommand(nID, lParam);
	


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

void CTestVChannelDlg::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
	
		CDialogEx::OnPaint();
	


//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CTestVChannelDlg::OnQueryDragIcon()

	return static_cast<HCURSOR>(m_hIcon);




void CTestVChannelDlg::OnBnClickedOk()
	

	DWORD dwSessionId = 0;
	ProcessIdToSessionId(GetCurrentProcessId(), &dwSessionId);
	/* 对应freerdp的虚拟通道的名称*/
	HANDLE hVirtChannel = WTSVirtualChannelOpen(
		WTS_CURRENT_SERVER_HANDLE,
		dwSessionId,
		"mytest3");

	if (hVirtChannel == NULL)
		return;

	DWORD ulBytesWritten = 0;

	//WTSVirtualChannelWrite
	//https://docs.microsoft.com/zh-cn/windows/win32/api/wtsapi32/nf-wtsapi32-wtsvirtualchannelwrite

	char testWrite[128] =  0 ;
	lstrcpy(testWrite, "98765432100000");
	if (WTSVirtualChannelWrite(hVirtChannel,
		(PCHAR)testWrite,
		sizeof(testWrite),
		&ulBytesWritten))

	

		DWORD ulBytesRead = 0;
		//WTSVirtualChannelRead
		//https://docs.microsoft.com/zh-cn/windows/win32/api/wtsapi32/nf-wtsapi32-wtsvirtualchannelread
		//

		char testRead[2500] =  0 ;

		if (WTSVirtualChannelRead(
			hVirtChannel,
			1000 * 5, //（毫秒）超时 //
					  //INFINITE,
			(PCHAR)testRead,
			sizeof(testRead),
			&ulBytesRead))
		

			m_listbox1.AddString( testRead );
			OutputDebugString(testRead);

		
		else
		
			m_listbox1.AddString("ReturnCode Error | WTSVirtualChannelRead TimeOut ");
			OutputDebugString("ReturnCode Error | WTSVirtualChannelRead TimeOut ");
		
	

	//WTSVirtualChannelQuery
	//https ://docs.microsoft.com/zh-cn/windows/win32/api/wtsapi32/nf-wtsapi32-wtsvirtualchannelquery

	WTSVirtualChannelClose(hVirtChannel);
	
	//CDialogEx::OnOK();



void CTestVChannelDlg::OnBnClickedCancel()

	// TODO: 在此添加控件通知处理程序代码



	CDialogEx::OnCancel();

TestVChannelDlg.h 头文件如下：

// TestVChannelDlg.h : 头文件
//

#pragma once
#include "afxwin.h"


// CTestVChannelDlg 对话框
class CTestVChannelDlg : public CDialogEx

// 构造
public:
	CTestVChannelDlg(CWnd* pParent = NULL);	// 标准构造函数

// 对话框数据
#ifdef AFX_DESIGN_TIME
	enum  IDD = IDD_TESTVCHANNEL_DIALOG ;
#endif

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);	// DDX/DDV 支持


// 实现
protected:
	HICON m_hIcon;

	// 生成的消息映射函数
	virtual BOOL OnInitDialog();
	afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
	afx_msg void OnPaint();
	afx_msg HCURSOR OnQueryDragIcon();
	DECLARE_MESSAGE_MAP()
public:
	afx_msg void OnBnClickedOk();
	afx_msg void OnBnClickedCancel();
	CListBox m_listbox1;
;

TestVChannel.cpp代码如下：

// TestVChannel.cpp : 定义应用程序的类行为。
//

#include "stdafx.h"
#include "TestVChannel.h"
#include "TestVChannelDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CTestVChannelApp

BEGIN_MESSAGE_MAP(CTestVChannelApp, CWinApp)
	ON_COMMAND(ID_HELP, &CWinApp::OnHelp)
END_MESSAGE_MAP()


// CTestVChannelApp 构造

CTestVChannelApp::CTestVChannelApp()

	// 支持重新启动管理器
	m_dwRestartManagerSupportFlags = AFX_RESTART_MANAGER_SUPPORT_RESTART;

	// TODO: 在此处添加构造代码，
	// 将所有重要的初始化放置在 InitInstance 中



// 唯一的一个 CTestVChannelApp 对象

CTestVChannelApp theApp;


// CTestVChannelApp 初始化

BOOL CTestVChannelApp::InitInstance()

	// 如果一个运行在 Windows XP 上的应用程序清单指定要
	// 使用 ComCtl32.dll 版本 6 或更高版本来启用可视化方式，
	//则需要 InitCommonControlsEx()。  否则，将无法创建窗口。
	INITCOMMONCONTROLSEX InitCtrls;
	InitCtrls.dwSize = sizeof(InitCtrls);
	// 将它设置为包括所有要在应用程序中使用的
	// 公共控件类。
	InitCtrls.dwICC = ICC_WIN95_CLASSES;
	InitCommonControlsEx(&InitCtrls);

	CWinApp::InitInstance();


	AfxEnableControlContainer();

	// 创建 shell 管理器，以防对话框包含
	// 任何 shell 树视图控件或 shell 列表视图控件。
	CShellManager *pShellManager = new CShellManager;

	// 激活“Windows Native”视觉管理器，以便在 MFC 控件中启用主题
	CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));

	// 标准初始化
	// 如果未使用这些功能并希望减小
	// 最终可执行文件的大小，则应移除下列
	// 不需要的特定初始化例程
	// 更改用于存储设置的注册表项
	// TODO: 应适当修改该字符串，
	// 例如修改为公司或组织名
	SetRegistryKey(_T("应用程序向导生成的本地应用程序"));

	CTestVChannelDlg dlg;
	m_pMainWnd = &dlg;
	INT_PTR nResponse = dlg.DoModal();
	if (nResponse == IDOK)
	
		// TODO: 在此放置处理何时用
		//  “确定”来关闭对话框的代码
	
	else if (nResponse == IDCANCEL)
	
		// TODO: 在此放置处理何时用
		//  “取消”来关闭对话框的代码
	
	else if (nResponse == -1)
	
		TRACE(traceAppMsg, 0, "警告: 对话框创建失败，应用程序将意外终止。\\n");
		TRACE(traceAppMsg, 0, "警告: 如果您在对话框上使用 MFC 控件，则无法 #define _AFX_NO_MFC_CONTROLS_IN_DIALOGS。\\n");
	

	// 删除上面创建的 shell 管理器。
	if (pShellManager != NULL)
	
		delete pShellManager;
	

#ifndef _AFXDLL
	ControlBarCleanUp();
#endif

	// 由于对话框已关闭，所以将返回 FALSE 以便退出应用程序，
	//  而不是启动应用程序的消息泵。
	return FALSE;

TestVChannel.h 代码如下：

// TestVChannel.h : PROJECT_NAME 应用程序的主头文件
//

#pragma once

#ifndef __AFXWIN_H__
	#error "在包含此文件之前包含“stdafx.h”以生成 PCH 文件"
#endif

#include "resource.h"		// 主符号


// CTestVChannelApp: 
// 有关此类的实现，请参阅 TestVChannel.cpp
//

class CTestVChannelApp : public CWinApp

public:
	CTestVChannelApp();

// 重写
public:
	virtual BOOL InitInstance();

// 实现

	DECLARE_MESSAGE_MAP()
;

extern CTestVChannelApp theApp;

MFC工程跑起来就好了