LANShare.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902
  1. // LANShare.cpp : 定义 DLL 的初始化例程。
  2. //
  3. #include "stdafx.h"
  4. #include "LANShare.h"
  5. #ifdef _DEBUG
  6. #define new DEBUG_NEW
  7. #endif
  8. //////////////////////////////////////////////////////////////////////////
  9. #include <shlwapi.h>
  10. #include <Winnetwk.h>
  11. #pragma comment(lib, "Mpr.lib.")
  12. #include <Lm.h>
  13. #include <Lmshare.h>
  14. #pragma comment(lib, "Netapi32.lib")
  15. #include <Aclapi.h>
  16. #pragma comment(lib, "Advapi32.lib")
  17. //////////////////////////////////////////////////////////////////////////
  18. //使用Windows的HeapAlloc函数进行动态内存分配
  19. #define myheapalloc(x) (HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, x))
  20. #define myheapfree(x) (HeapFree(GetProcessHeap(), 0, x))
  21. typedef BOOL (WINAPI *SetSecurityDescriptorControlFnPtr)(
  22. IN PSECURITY_DESCRIPTOR pSecurityDescriptor,
  23. IN SECURITY_DESCRIPTOR_CONTROL ControlBitsOfInterest,
  24. IN SECURITY_DESCRIPTOR_CONTROL ControlBitsToSet);
  25. typedef BOOL (WINAPI *AddAccessAllowedAceExFnPtr)(
  26. IN PACL pAcl,
  27. IN DWORD dwAceRevision,
  28. IN DWORD AceFlags,
  29. IN DWORD AccessMask,
  30. IN PSID pSid
  31. );
  32. //
  33. //TODO: 如果此 DLL 相对于 MFC DLL 是动态链接的,
  34. // 则从此 DLL 导出的任何调入
  35. // MFC 的函数必须将 AFX_MANAGE_STATE 宏添加到
  36. // 该函数的最前面。
  37. //
  38. // 例如:
  39. //
  40. // extern "C" BOOL PASCAL EXPORT ExportedFunction()
  41. // {
  42. // AFX_MANAGE_STATE(AfxGetStaticModuleState());
  43. // // 此处为普通函数体
  44. // }
  45. //
  46. // 此宏先于任何 MFC 调用
  47. // 出现在每个函数中十分重要。这意味着
  48. // 它必须作为函数中的第一个语句
  49. // 出现,甚至先于所有对象变量声明,
  50. // 这是因为它们的构造函数可能生成 MFC
  51. // DLL 调用。
  52. //
  53. // 有关其他详细信息,
  54. // 请参阅 MFC 技术说明 33 和 58。
  55. //
  56. // CLANShareApp
  57. BEGIN_MESSAGE_MAP(CLANShareApp, CWinApp)
  58. END_MESSAGE_MAP()
  59. // CLANShareApp 构造
  60. CLANShareApp::CLANShareApp()
  61. {
  62. // TODO: 在此处添加构造代码,
  63. // 将所有重要的初始化放置在 InitInstance 中
  64. }
  65. // 唯一的一个 CLANShareApp 对象
  66. CLANShareApp theApp;
  67. // CLANShareApp 初始化
  68. BOOL CLANShareApp::InitInstance()
  69. {
  70. CWinApp::InitInstance();
  71. return TRUE;
  72. }
  73. //////////////////////////////////////////////////////////////////////////
  74. /************************************************************************/
  75. /* 函数:MultiToWide[5/14/2016 home];
  76. /* 描述:将多字节字符串转成宽字节字符串;
  77. /* 参数:;
  78. /* [IN] pszMultiString:多字节字符串;
  79. /* 返回:若转换成功,返回宽字节字符串, 否则返回NULL;
  80. /* 注意:;
  81. /* 示例:;
  82. /*
  83. /* 修改:;
  84. /* 日期:;
  85. /* 内容:;
  86. /************************************************************************/
  87. WCHAR* MultiToWide(IN const CHAR* pszMultiString)
  88. {
  89. int wSize = MultiByteToWideChar(CP_ACP, 0, pszMultiString, -1, NULL, 0);
  90. if (wSize == ERROR_NO_UNICODE_TRANSLATION || wSize == 0 )
  91. {
  92. return NULL;
  93. }
  94. WCHAR *punicode = new WCHAR[wSize];
  95. int convresult = MultiByteToWideChar(CP_ACP, 0, pszMultiString, -1, punicode, wSize);
  96. if (convresult != wSize)
  97. {
  98. if (punicode) delete []punicode;
  99. return NULL;
  100. }
  101. return punicode;
  102. }
  103. INT WideToMulti(IN const WCHAR* pWideString, IN LPCH pMultiString, IN const int& nBufLen)
  104. {
  105. INT nSize = WideCharToMultiByte(CP_OEMCP, 0, pWideString, -1, NULL, 0, NULL, NULL);
  106. if (nSize == ERROR_NO_UNICODE_TRANSLATION || nSize == 0 )
  107. {
  108. return 0;
  109. }
  110. if ( nSize > nBufLen )
  111. return 0;
  112. INT nResult = WideCharToMultiByte(CP_OEMCP, 0, pWideString, -1, pMultiString, nSize, NULL, NULL);
  113. if (nResult != nSize)
  114. {
  115. return 0;
  116. }
  117. return nSize;
  118. }
  119. /************************************************************************/
  120. /* 函数:DisConnectSession[5/14/2016 home];
  121. /* 描述:断开指定的局域网共享连接;
  122. /* 参数:;
  123. /* [IN] lpHostAddr:要断开连接的局域网地址;
  124. /* 返回:void;
  125. /* 注意:;
  126. /* 示例:;
  127. /*
  128. /* 修改:;
  129. /* 日期:;
  130. /* 内容:;
  131. /************************************************************************/
  132. extern "C" BOOL PASCAL EXPORT DisConnectSession( IN LPCTSTR lpHostAddr )
  133. {
  134. if (NULL == lpHostAddr || lpHostAddr[0] == _T('\0'))
  135. return FALSE;
  136. DWORD dwRes = WNetCancelConnection2(lpHostAddr, CONNECT_UPDATE_PROFILE, TRUE);
  137. if ( ERROR_SUCCESS == dwRes || ERROR_NOT_CONNECTED == dwRes)
  138. _tprintf_s(_T("关掉局域网共享成功[%s]\n"), lpHostAddr);
  139. else
  140. _tprintf_s(_T("关掉局域网共享失败[%s]Error:%d\n"), lpHostAddr, dwRes);
  141. return ( ERROR_SUCCESS == dwRes || ERROR_NOT_CONNECTED == dwRes ) ? TRUE : FALSE;
  142. }
  143. /************************************************************************/
  144. /* 函数:[5/14/2016 home];
  145. /* 描述:取消共享,但并不是删除这个目录;
  146. /* 参数:;
  147. /* [IN] lpShareName:要取消的共享名;
  148. /* 返回:void;
  149. /* 注意:;
  150. /* 示例:;
  151. /*
  152. /* 修改:;
  153. /* 日期:;
  154. /* 内容:;
  155. /************************************************************************/
  156. extern "C" BOOL PASCAL EXPORT DelShareDirectoryW(IN LPCWSTR lpShareName)
  157. {
  158. if (NULL == lpShareName || lpShareName[0] == _T('\0'))
  159. return FALSE;
  160. DWORD dwRes = NetShareDel(NULL, const_cast<LPWSTR>(lpShareName), NULL);
  161. if ( dwRes == NERR_Success )
  162. _tprintf_s(_T("删除共享成功"));
  163. else
  164. _tprintf_s(_T("删除共享失败,Error %d\n"), dwRes);
  165. return (dwRes == NERR_Success) ? TRUE : FALSE;
  166. }
  167. extern "C" BOOL PASCAL EXPORT DelShareDirectoryA(IN LPCCH lpShareName)
  168. {
  169. if ( lpShareName == NULL )
  170. return TRUE;
  171. WCHAR* pShareName = MultiToWide(lpShareName);
  172. BOOL bRes = DelShareDirectoryW(pShareName);
  173. if ( pShareName )
  174. delete []pShareName;
  175. return bRes;
  176. }
  177. /************************************************************************/
  178. /* 函数:IsTheDirectoryShared[5/12/2016 IT];
  179. /* 描述:判断本地目录是否共享,若共享则返回共享名称;
  180. /* 参数:;
  181. /* [IN] lpDirectory:本地路径;
  182. /* [OUT] lpShareName:若本地路径已共享,返回共享名称;
  183. /* [IN] nBufLen:lpShareName缓冲区大小;
  184. /* 返回:若已共享返回TRUE;
  185. /* 注意:该函数字符编号是UICODE;
  186. /* 示例:;
  187. /*
  188. /* 修改:;
  189. /* 日期:;
  190. /* 内容:;
  191. /************************************************************************/
  192. extern "C" BOOL PASCAL EXPORT IsTheDirectorySharedW(IN LPCWSTR lpDirectory, OUT LPWSTR lpShareName = NULL , IN const INT& nBufLen = 0 )
  193. {
  194. if ( lpDirectory == NULL || !PathFileExistsW(lpDirectory) ){
  195. _tprintf_s(_T("要判断的路径无效"));
  196. return FALSE;
  197. }
  198. if ( lpDirectory[0] == L'\\' && lpDirectory[1] == L'\\' ){
  199. _tprintf_s(_T("要判断的路径是网络路径"));
  200. return TRUE;
  201. }
  202. // 枚举本机所有共享目录;
  203. PSHARE_INFO_502 BufPtr, p;
  204. NET_API_STATUS res;
  205. LPTSTR lpszServer = NULL;
  206. DWORD er = 0, tr = 0, resume = 0, i;
  207. do
  208. {
  209. res = NetShareEnum(NULL, 502, (LPBYTE*)&BufPtr, MAX_PREFERRED_LENGTH, &er, &tr, &resume);
  210. if ( res == ERROR_SUCCESS || res == ERROR_MORE_DATA )
  211. {
  212. p = BufPtr;
  213. for ( i = 1; i <= er; i++ )
  214. {
  215. if ( p->shi502_type == STYPE_DISKTREE )
  216. {// 磁盘类型;
  217. printf("%-20S%-30S\n", p->shi502_netname, p->shi502_path);
  218. if ( wcscmp(lpDirectory, (WCHAR*)(p->shi502_path)) == 0 )
  219. {
  220. if ( lpShareName && nBufLen > 0 )
  221. wcscpy_s(lpShareName, nBufLen, (WCHAR*)(p->shi502_netname));
  222. NetApiBufferFree(BufPtr);
  223. return TRUE;
  224. }
  225. }
  226. p++;
  227. }
  228. NetApiBufferFree(BufPtr);
  229. }
  230. } while (res == ERROR_MORE_DATA );
  231. return FALSE;
  232. }
  233. extern "C" BOOL PASCAL EXPORT IsTheDirectorySharedA(IN LPCCH lpDirectory, OUT LPCH lpShareName = NULL , IN const INT& nBufLen = 0 )
  234. {
  235. if ( lpDirectory == NULL )
  236. return FALSE;
  237. WCHAR *pShareName = NULL;
  238. WCHAR *pDirectory = MultiToWide(lpDirectory);
  239. if ( lpShareName )
  240. {
  241. pShareName = new WCHAR[nBufLen + 1];
  242. memset(pShareName, 0, sizeof(WCHAR)*(nBufLen+1));
  243. }
  244. BOOL bRes = IsTheDirectorySharedW(pDirectory, pShareName, nBufLen);
  245. if ( bRes )
  246. WideToMulti(pShareName, lpShareName, nBufLen);
  247. if ( pDirectory )
  248. delete []pDirectory;
  249. if ( pShareName )
  250. delete []pShareName;
  251. return bRes;
  252. }
  253. /************************************************************************/
  254. /* 函数:AccessShareDirectory[5/11/2016 IT];
  255. /* 描述:访问指定账号和密码的局域网共享目录;
  256. /* 参数:;
  257. /* [IN] lpUserName:登陆共享目录的账号;
  258. /* [IN] lpPasswd:登陆共享目录的密码;
  259. /* [IN] lpShareDirectory:要登陆的共享目录("\\it-pc\Test\123");
  260. /* 返回:成功返回true;
  261. /* 注意:;
  262. /* 示例:;
  263. /*
  264. /* 修改:;
  265. /* 日期:;
  266. /* 内容:;
  267. /************************************************************************/
  268. extern "C" BOOL PASCAL EXPORT AccessShareDirectory(IN LPCTSTR lpUserName, IN LPCTSTR lpPasswd, IN LPCTSTR lpShareDirectory)
  269. {
  270. // 参数有效性判断;
  271. if (lpShareDirectory == NULL ||
  272. lpShareDirectory[0] == _T('\0') ||
  273. lpShareDirectory[0] != _T('\\'))
  274. {
  275. return FALSE;
  276. }
  277. int nPos = 0;
  278. DWORD dwError = 0;
  279. CString strHostAddr = lpShareDirectory;
  280. // 从lpShareDirectory解析出网络主机地址,且主机名末尾不能包含'\\';
  281. if ( strHostAddr.Left(2) != _T("\\\\") )
  282. return FALSE;
  283. nPos = strHostAddr.Mid(2).Find(_T("\\"));
  284. if ( nPos != -1 )
  285. {
  286. strHostAddr = strHostAddr.Left(nPos+2);
  287. }
  288. // 断开之前的连接;
  289. dwError = WNetCancelConnection2(strHostAddr, CONNECT_UPDATE_PROFILE, TRUE);
  290. if ( dwError != ERROR_SUCCESS && dwError != ERROR_NOT_CONNECTED)
  291. {
  292. _tprintf_s(_T("断开失败\n"));
  293. }
  294. NETRESOURCE net_Resource;
  295. net_Resource.dwDisplayType = RESOURCEDISPLAYTYPE_DIRECTORY;
  296. net_Resource.dwScope = RESOURCE_CONNECTED;
  297. net_Resource.dwType = RESOURCETYPE_ANY;
  298. net_Resource.dwUsage = 0;
  299. net_Resource.lpComment = TEXT("");
  300. net_Resource.lpLocalName = NULL; // 不映射成本地驱动器;
  301. net_Resource.lpProvider= NULL;
  302. net_Resource.lpRemoteName = const_cast<LPTSTR>(strHostAddr.GetString());
  303. DWORD dwFlags = CONNECT_UPDATE_PROFILE;
  304. dwError = WNetAddConnection2(&net_Resource, lpPasswd, lpUserName, dwFlags);
  305. switch(dwError)
  306. {
  307. case ERROR_SUCCESS:
  308. _tprintf(TEXT("访问共享成功!"));
  309. break;
  310. case ERROR_ACCESS_DENIED:
  311. _tprintf(TEXT("没有权访问!"));
  312. break;
  313. case ERROR_ALREADY_ASSIGNED:
  314. ShellExecute(NULL, TEXT("open"), TEXT("z:"), NULL, NULL, SW_SHOWNORMAL);
  315. break;
  316. case ERROR_INVALID_ADDRESS:
  317. _tprintf(TEXT("IP地址无效"));
  318. break;
  319. case ERROR_NO_NETWORK:
  320. _tprintf(TEXT("网络不可达!"));
  321. break;
  322. }
  323. return dwError;
  324. }
  325. /************************************************************************/
  326. /* 函数:[5/11/2016 Home];
  327. /* 描述:;
  328. /* 参数:;
  329. /* [IN] lpDirectory:要共享的本地目录;
  330. /* [IN] lpShareName:要共享的本地目录的共享名称;
  331. /* 返回:成功返回0;
  332. /* 注意:该函数字符编号是UICODE;
  333. /* 示例:;
  334. /*
  335. /* 修改:;
  336. /* 日期:;
  337. /* 内容:;
  338. /************************************************************************/
  339. extern "C" BOOL PASCAL EXPORT AddShareDirectoryW(IN LPCWSTR lpDirectory, IN LPCWSTR lpShareName)
  340. {
  341. if ( lpDirectory == NULL || !PathFileExistsW(lpDirectory) ){
  342. _tprintf_s(_T("设置共享的目录无效\n"));
  343. return FALSE;
  344. }
  345. if ( lpShareName == NULL || lpShareName[0] == _T('\0') ){
  346. _tprintf_s(_T("设置共享的名称无效\n"));
  347. return FALSE;
  348. }
  349. DWORD dwEr = 0;
  350. DWORD dwRet = 0;
  351. SHARE_INFO_2 si2;
  352. memset(&si2, 0, sizeof(SHARE_INFO_2));
  353. si2.shi2_netname = const_cast<LPWSTR>(lpShareName); // 要设置目录的共享名称(必须为Unicode);
  354. si2.shi2_type = STYPE_DISKTREE; // 要共享的类型STYPE_DISKTREE=磁盘驱动器;
  355. si2.shi2_remark = NULL; // 共享资源的可选注释;
  356. si2.shi2_permissions = ACCESS_ALL; // 共享资源的共享权限(对xp来说,这个值设置成什么都是无效的);
  357. si2.shi2_max_uses = -1; // 共享资源最大可访问用户量;
  358. si2.shi2_current_uses = 0; // 当前可访问量;
  359. si2.shi2_passwd = NULL; // 共享密码 NULL表示没密码;
  360. si2.shi2_path = const_cast<LPWSTR>(lpDirectory); // 要共享的目录;
  361. dwRet = NetShareAdd(NULL, 2, (LPBYTE)&si2, &dwEr);
  362. if ( dwRet == NERR_Success || dwRet == NERR_DuplicateShare )
  363. {
  364. _tprintf_s(_T("设置共享成功\n"));
  365. // 刷新文件列表;
  366. SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
  367. }
  368. else
  369. {
  370. _tprintf_s(_T("设置共享失败 Error %u\n"), dwRet);
  371. }
  372. return ( dwRet == NERR_Success || dwRet == NERR_DuplicateShare ) ? TRUE : FALSE;
  373. //return SetNetShareSecurity(lpShareName, _T("Everyone"));
  374. }
  375. extern "C" BOOL PASCAL EXPORT AddShareDirectoryA(IN LPCCH lpDirectory, IN LPCCH lpShareName)
  376. {
  377. WCHAR *pShareName = NULL;
  378. WCHAR *pDirectory = NULL;
  379. if ( lpDirectory )
  380. pDirectory = MultiToWide(lpDirectory);
  381. if ( lpShareName )
  382. pShareName = MultiToWide(lpShareName);
  383. BOOL bRes = AddShareDirectoryW(pDirectory, pShareName);
  384. if ( pDirectory )
  385. delete []pDirectory;
  386. if ( pShareName )
  387. delete []pShareName;
  388. return bRes;
  389. }
  390. /************************************************************************/
  391. /* 函数:SetNetShareSecurity[5/12/2016 IT];
  392. /* 描述:设置共享目录的安全选项;
  393. /* 参数:;
  394. /* [IN] lpShareName:要设置的共享名称;
  395. /* [OUT] lpUsser:安全选项用户;
  396. /* [IN/OUT] :;
  397. /* 返回:void;
  398. /* 注意:;
  399. /* 示例:;
  400. /*
  401. /* 修改:;
  402. /* 日期:;
  403. /* 内容:;
  404. /************************************************************************/
  405. extern "C" BOOL PASCAL EXPORT SetNetShareSecurity(IN LPCTSTR lpShareName, IN LPCTSTR lpUsser, IN const DWORD& dwPermissions = STANDARD_RIGHTS_ALL| SPECIFIC_RIGHTS_ALL )
  406. {
  407. if ( lpShareName == NULL || lpUsser == NULL ){
  408. _tprintf_s(_T("共享名称或用户名无效\n"));
  409. return FALSE;
  410. }
  411. // 设置成功后,给共享目录分配权限;
  412. DWORD dwRes = 0;
  413. PACL pOldDacl = NULL;
  414. PACL pNewDacl = NULL;
  415. EXPLICIT_ACCESS ea;
  416. PSECURITY_DESCRIPTOR psd = NULL;
  417. // 获取指定对象名称的DACL(Discretionary Access Control List任意访问控制列表)指针对象;
  418. dwRes = GetNamedSecurityInfo(const_cast<LPTSTR>(lpShareName), SE_LMSHARE, DACL_SECURITY_INFORMATION, NULL, NULL, &pOldDacl, NULL, &psd);
  419. if ( ERROR_SUCCESS != dwRes ) {
  420. _tprintf_s(_T("GetNamedSecurityInfo Error %u\n"), dwRes); // dwRes = 2310表示没有这个共享名称;
  421. goto Cleanup;
  422. }
  423. // 为新的ACE初始化EXPLICIT_ACCESS结构体;
  424. ZeroMemory(&ea, sizeof(EXPLICIT_ACCESS));
  425. // 设置最重要的访问权限;STANDARD_RIGHTS_ALL全部标准权限;SPECIFIC_RIGHTS_ALL全部特殊权限;
  426. ea.grfAccessPermissions = dwPermissions;
  427. ea.grfAccessMode = SET_ACCESS;
  428. ea.grfInheritance = CONTAINER_INHERIT_ACE;
  429. ea.Trustee.TrusteeForm = TRUSTEE_IS_NAME;
  430. ea.Trustee.ptstrName = const_cast<LPTSTR>(lpUsser);
  431. // 合并新的ACE到现有的DACL来创建新的ACL;
  432. dwRes = SetEntriesInAcl(1, &ea, pOldDacl, &pNewDacl);
  433. if (ERROR_SUCCESS != dwRes) {
  434. _tprintf_s(_T("SetEntriesInAcl Error %u\n"), dwRes );
  435. goto Cleanup;
  436. }
  437. // 用新的ACL附加到DACL对象中;
  438. dwRes = SetNamedSecurityInfo(const_cast<LPTSTR>(lpShareName), SE_LMSHARE, DACL_SECURITY_INFORMATION, NULL, NULL, pNewDacl, NULL);
  439. if (ERROR_SUCCESS != dwRes) {
  440. _tprintf_s( _T("SetNamedSecurityInfo Error %u\n"), dwRes );
  441. goto Cleanup;
  442. }
  443. // 设置成功后,刷新文件夹;
  444. SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
  445. Cleanup:
  446. if ( psd != NULL )
  447. LocalFree((HLOCAL)psd);
  448. if ( pNewDacl != NULL )
  449. LocalFree((HLOCAL)pNewDacl);
  450. if ( pOldDacl != NULL )
  451. LocalFree((HLOCAL)pNewDacl);
  452. return (dwRes == ERROR_SUCCESS) ? TRUE : FALSE;
  453. }
  454. /************************************************************************/
  455. /* 函数:[5/14/2016 home];
  456. /* 描述:给指定的目录添加安全权限;
  457. /* 参数:;
  458. /* [IN] lpszFileName:文件或目录名;
  459. /* [IN] lpszAccountName:要给文件或目录添加的安全账号;
  460. /* [IN] dwAccessMask:要添加的权限值;
  461. /* 返回:void;
  462. /* 注意:;
  463. /* 示例:;
  464. /*
  465. /* 修改:;
  466. /* 日期:;
  467. /* 内容:;
  468. /************************************************************************/
  469. extern "C" BOOL PASCAL EXPORT AddAccessRights(IN LPCTSTR lpszFileName, IN LPCTSTR lpszAccountName, IN const DWORD& dwAccessMask = STANDARD_RIGHTS_ALL| SPECIFIC_RIGHTS_ALL )
  470. {
  471. // 声明SID变量;
  472. SID_NAME_USE snuType;
  473. // 声明和LookupAccountName相关的变量(注意,全为0,要在程序中动态分配);
  474. TCHAR * szDomain = NULL;
  475. DWORD cbDomain = 0;
  476. LPVOID pUserSID = NULL;
  477. DWORD cbUserSID = 0;
  478. // 和文件相关的安全描述符 SD 的变量;
  479. PSECURITY_DESCRIPTOR pFileSD = NULL; // 结构变量;
  480. DWORD cbFileSD = 0; // SD的size;
  481. // 一个新的SD的变量,用于构造新的ACL(把已有的ACL和需要新加的ACL整合起来);
  482. SECURITY_DESCRIPTOR newSD;
  483. // 和ACL 相关的变量;
  484. PACL pACL = NULL;
  485. BOOL fDaclPresent = FALSE;
  486. BOOL fDaclDefaulted = FALSE;
  487. ACL_SIZE_INFORMATION AclInfo;
  488. // 一个新的 ACL 变量;
  489. PACL pNewACL = NULL; //结构指针变量;
  490. DWORD cbNewACL = 0; //ACL的size;
  491. // 一个临时使用的 ACE 变量;
  492. LPVOID pTempAce = NULL;
  493. UINT CurrentAceIndex = 0; //ACE在ACL中的位置;
  494. UINT newAceIndex = 0; //新添的ACE在ACL中的位置;
  495. //API函数的返回值,假设所有的函数都返回失败。
  496. BOOL fResult = FALSE;
  497. BOOL fAPISuccess = FALSE;
  498. SECURITY_INFORMATION secInfo = DACL_SECURITY_INFORMATION;
  499. // 下面的两个函数是新的API函数,仅在Windows 2000以上版本的操作系统支持。
  500. // 在此将从Advapi32.dll文件中动态载入。如果你使用VC++ 6.0编译程序,而且你想
  501. // 使用这两个函数的静态链接。则请为你的编译加上:/D_WIN32_WINNT=0x0500
  502. // 的编译参数。并且确保你的SDK的头文件和lib文件是最新的。
  503. SetSecurityDescriptorControlFnPtr _SetSecurityDescriptorControl = NULL;
  504. AddAccessAllowedAceExFnPtr _AddAccessAllowedAceEx = NULL;
  505. __try {
  506. //
  507. // STEP 1: 通过用户名取得SID
  508. // 在这一步中LookupAccountName函数被调用了两次,第一次是取出所需要
  509. // 的内存的大小,然后,进行内存分配。第二次调用才是取得了用户的帐户信息。
  510. // LookupAccountName同样可以取得域用户或是用户组的信息。(请参看MSDN)
  511. //
  512. fAPISuccess = LookupAccountName(NULL, lpszAccountName,pUserSID, &cbUserSID, szDomain, &cbDomain, &snuType);
  513. // 以上调用API会失败,失败原因是内存不足。并把所需要的内存大小传出。
  514. // 下面是处理非内存不足的错误。
  515. if (fAPISuccess)
  516. __leave;
  517. else if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) {
  518. _tprintf(TEXT("LookupAccountName() failed. Error %d/n"), GetLastError());
  519. __leave;
  520. }
  521. pUserSID = myheapalloc(cbUserSID);
  522. if (!pUserSID) {
  523. _tprintf(TEXT("HeapAlloc() failed. Error %d/n"), GetLastError());
  524. __leave;
  525. }
  526. szDomain = (TCHAR *) myheapalloc(cbDomain * sizeof(TCHAR));
  527. if (!szDomain) {
  528. _tprintf(TEXT("HeapAlloc() failed. Error %d/n"), GetLastError());
  529. __leave;
  530. }
  531. fAPISuccess = LookupAccountName(NULL, lpszAccountName,pUserSID, &cbUserSID, szDomain, &cbDomain, &snuType);
  532. if (!fAPISuccess) {
  533. _tprintf(TEXT("LookupAccountName() failed. Error %d/n"), GetLastError());
  534. __leave;
  535. }
  536. //
  537. // STEP 2: 取得文件(目录)相关的安全描述符SD
  538. // 使用GetFileSecurity函数取得一份文件SD的拷贝,同样,这个函数也
  539. // 是被调用两次,第一次同样是取SD的内存长度。注意,SD有两种格式:自相关的
  540. // (self-relative)和 完全的(absolute),GetFileSecurity只能取到“自
  541. // 相关的”,而SetFileSecurity则需要完全的。这就是为什么需要一个新的SD,
  542. // 而不是直接在GetFileSecurity返回的SD上进行修改。因为“自相关的”信息
  543. // 是不完整的。
  544. fAPISuccess = GetFileSecurity(lpszFileName, secInfo, pFileSD, 0, &cbFileSD);
  545. // 以上调用API会失败,失败原因是内存不足。并把所需要的内存大小传出。
  546. // 下面是处理非内存不足的错误。
  547. if (fAPISuccess)
  548. __leave;
  549. else if (GetLastError() != ERROR_INSUFFICIENT_BUFFER) {
  550. _tprintf(TEXT("GetFileSecurity() failed. Error %d/n"), GetLastError());
  551. __leave;
  552. }
  553. pFileSD = myheapalloc(cbFileSD);
  554. if (!pFileSD) {
  555. _tprintf(TEXT("HeapAlloc() failed. Error %d/n"), GetLastError());
  556. __leave;
  557. }
  558. fAPISuccess = GetFileSecurity(lpszFileName, secInfo, pFileSD, cbFileSD, &cbFileSD);
  559. if (!fAPISuccess) {
  560. _tprintf(TEXT("GetFileSecurity() failed. Error %d/n"), GetLastError());
  561. __leave;
  562. }
  563. //
  564. // STEP 3: 初始化一个新的SD
  565. //
  566. if (!InitializeSecurityDescriptor(&newSD, SECURITY_DESCRIPTOR_REVISION)) {
  567. _tprintf(TEXT("InitializeSecurityDescriptor() failed. Error %d/n"), GetLastError());
  568. __leave;
  569. }
  570. //
  571. // STEP 4: 从GetFileSecurity 返回的SD中取DACL
  572. //
  573. if (!GetSecurityDescriptorDacl(pFileSD, &fDaclPresent, &pACL, &fDaclDefaulted)) {
  574. _tprintf(TEXT("GetSecurityDescriptorDacl() failed. Error %d/n"), GetLastError());
  575. __leave;
  576. }
  577. //
  578. // STEP 5: 取 DACL的内存size
  579. // GetAclInformation可以提供DACL的内存大小。只传入一个类型为
  580. // ACL_SIZE_INFORMATION的structure的参数,需DACL的信息,是为了
  581. // 方便我们遍历其中的ACE。
  582. AclInfo.AceCount = 0; // Assume NULL DACL.
  583. AclInfo.AclBytesFree = 0;
  584. AclInfo.AclBytesInUse = sizeof(ACL);
  585. if (pACL == NULL)
  586. fDaclPresent = FALSE;
  587. // 如果DACL不为空,则取其信息。(大多数情况下“自关联”的DACL为空)
  588. if (fDaclPresent) {
  589. if (!GetAclInformation(pACL, &AclInfo, sizeof(ACL_SIZE_INFORMATION), AclSizeInformation)) {
  590. _tprintf(TEXT("GetAclInformation() failed. Error %d/n"),GetLastError());
  591. __leave;
  592. }
  593. }
  594. //
  595. // STEP 6: 计算新的ACL的size
  596. // 计算的公式是:原有的DACL的size加上需要添加的一个ACE的size,以
  597. // 及加上一个和ACE相关的SID的size,最后减去两个字节以获得精确的大小。
  598. cbNewACL = AclInfo.AclBytesInUse + sizeof(ACCESS_ALLOWED_ACE) + GetLengthSid(pUserSID) - sizeof(DWORD);
  599. //
  600. // STEP 7: 为新的ACL分配内存
  601. //
  602. pNewACL = (PACL) myheapalloc(cbNewACL);
  603. if (!pNewACL) {
  604. _tprintf(TEXT("HeapAlloc() failed. Error %d/n"), GetLastError());
  605. __leave;
  606. }
  607. //
  608. // STEP 8: 初始化新的ACL结构
  609. //
  610. if (!InitializeAcl(pNewACL, cbNewACL, ACL_REVISION2)) {
  611. _tprintf(TEXT("InitializeAcl() failed. Error %d/n"), GetLastError());
  612. __leave;
  613. }
  614. //
  615. // STEP 9 如果文件(目录) DACL 有数据,拷贝其中的ACE到新的DACL中
  616. //
  617. // 下面的代码假设首先检查指定文件(目录)是否存在的DACL,如果有的话,
  618. // 那么就拷贝所有的ACE到新的DACL结构中,我们可以看到其遍历的方法是采用
  619. // ACL_SIZE_INFORMATION结构中的AceCount成员来完成的。在这个循环中,
  620. // 会按照默认的ACE的顺序来进行拷贝(ACE在ACL中的顺序是很关键的),在拷
  621. // 贝过程中,先拷贝非继承的ACE(我们知道ACE会从上层目录中继承下来)
  622. //
  623. newAceIndex = 0;
  624. if (fDaclPresent && AclInfo.AceCount) {
  625. for (CurrentAceIndex = 0; CurrentAceIndex < AclInfo.AceCount;CurrentAceIndex++) {
  626. //
  627. // STEP 10: 从DACL中取ACE
  628. //
  629. if (!GetAce(pACL, CurrentAceIndex, &pTempAce)) {
  630. _tprintf(TEXT("GetAce() failed. Error %d/n"), GetLastError());
  631. __leave;
  632. }
  633. //
  634. // STEP 11: 检查是否是非继承的ACE
  635. // 如果当前的ACE是一个从父目录继承来的ACE,那么就退出循环。
  636. // 因为,继承的ACE总是在非继承的ACE之后,而我们所要添加的ACE
  637. // 应该在已有的非继承的ACE之后,所有的继承的ACE之前。退出循环
  638. // 正是为了要添加一个新的ACE到新的DACL中,这后,我们再把继承的
  639. // ACE拷贝到新的DACL中。
  640. //
  641. if (((ACCESS_ALLOWED_ACE *)pTempAce)->Header.AceFlags & INHERITED_ACE)
  642. break;
  643. //
  644. // STEP 12: 检查要拷贝的ACE的SID是否和需要加入的ACE的SID一样,
  645. // 如果一样,那么就应该废掉已存在的ACE,也就是说,同一个用户的存取
  646. // 权限的设置的ACE,在DACL中应该唯一。这在里,跳过对同一用户已设置
  647. // 了的ACE,仅是拷贝其它用户的ACE。
  648. //
  649. if (EqualSid(pUserSID,&(((ACCESS_ALLOWED_ACE *)pTempAce)->SidStart)))
  650. continue;
  651. //
  652. // STEP 13: 把ACE加入到新的DACL中
  653. // 下面的代码中,注意 AddAce 函数的第三个参数,这个参数的意思是
  654. // ACL中的索引值,意为要把ACE加到某索引位置之后,参数MAXDWORD的
  655. // 意思是确保当前的ACE是被加入到最后的位置。
  656. //
  657. if (!AddAce(pNewACL, ACL_REVISION, MAXDWORD, pTempAce,((PACE_HEADER) pTempAce)->AceSize)) {
  658. _tprintf(TEXT("AddAce() failed. Error %d/n"), GetLastError());
  659. __leave;
  660. }
  661. newAceIndex++;
  662. }
  663. }
  664. //
  665. // STEP 14: 把一个 access-allowed 的ACE 加入到新的DACL中
  666. // 前面的循环拷贝了所有的非继承且SID为其它用户的ACE,退出循环的第一件事
  667. // 就是加入我们指定的ACE。请注意首先先动态装载了一个AddAccessAllowedAceEx
  668. // 的API函数,如果装载不成功,就调用AddAccessAllowedAce函数。前一个函数仅
  669. // 在Windows 2000以后的版本支持,NT则没有,我们为了使用新版本的函数,我们首
  670. // 先先检查一下当前系统中可不可以装载这个函数,如果可以则就使用。使用动态链接
  671. // 比使用静态链接的好处是,程序运行时不会因为没有这个API函数而报错。
  672. //
  673. // Ex版的函数多出了一个参数AceFlag(第三人参数),用这个参数我们可以来设置一
  674. // 个叫ACE_HEADER的结构,以便让我们所设置的ACE可以被其子目录所继承下去,而
  675. // AddAccessAllowedAce函数不能定制这个参数,在AddAccessAllowedAce函数
  676. // 中,其会把ACE_HEADER这个结构设置成非继承的。
  677. //
  678. _AddAccessAllowedAceEx = (AddAccessAllowedAceExFnPtr)GetProcAddress(GetModuleHandle(TEXT("advapi32.dll")),"AddAccessAllowedAceEx");
  679. if (_AddAccessAllowedAceEx) {
  680. if (!_AddAccessAllowedAceEx(pNewACL, ACL_REVISION2,CONTAINER_INHERIT_ACE | OBJECT_INHERIT_ACE ,dwAccessMask, pUserSID)) {
  681. _tprintf(TEXT("AddAccessAllowedAceEx() failed. Error %d/n"), GetLastError());
  682. __leave;
  683. }
  684. }else{
  685. if (!AddAccessAllowedAce(pNewACL, ACL_REVISION2, dwAccessMask, pUserSID)) {
  686. _tprintf(TEXT("AddAccessAllowedAce() failed. Error %d/n"), GetLastError());
  687. __leave;
  688. }
  689. }
  690. //
  691. // STEP 15: 按照已存在的ACE的顺序拷贝从父目录继承而来的ACE
  692. //
  693. if (fDaclPresent && AclInfo.AceCount) {
  694. for (; CurrentAceIndex < AclInfo.AceCount;CurrentAceIndex++) {
  695. //
  696. // STEP 16: 从文件(目录)的DACL中继续取ACE
  697. //
  698. if (!GetAce(pACL, CurrentAceIndex, &pTempAce)) {
  699. _tprintf(TEXT("GetAce() failed. Error %d/n"), GetLastError());
  700. __leave;
  701. }
  702. //
  703. // STEP 17: 把ACE加入到新的DACL中
  704. //
  705. if (!AddAce(pNewACL, ACL_REVISION, MAXDWORD, pTempAce,((PACE_HEADER) pTempAce)->AceSize)) {
  706. _tprintf(TEXT("AddAce() failed. Error %d/n"), GetLastError());
  707. __leave;
  708. }
  709. }
  710. }
  711. //
  712. // STEP 18: 把新的ACL设置到新的SD中
  713. //
  714. if (!SetSecurityDescriptorDacl(&newSD, TRUE, pNewACL, FALSE)) {
  715. _tprintf(TEXT("SetSecurityDescriptorDacl() failed. Error %d/n"),GetLastError());
  716. __leave;
  717. }
  718. //
  719. // STEP 19: 把老的SD中的控制标记再拷贝到新的SD中,我们使用的是一个叫
  720. // SetSecurityDescriptorControl() 的API函数,这个函数同样只存在于
  721. // Windows 2000以后的版本中,所以我们还是要动态地把其从advapi32.dll
  722. // 中载入,如果系统不支持这个函数,那就不拷贝老的SD的控制标记了。
  723. //
  724. _SetSecurityDescriptorControl =(SetSecurityDescriptorControlFnPtr)GetProcAddress(GetModuleHandle(TEXT("advapi32.dll")),"SetSecurityDescriptorControl");
  725. if (_SetSecurityDescriptorControl) {
  726. SECURITY_DESCRIPTOR_CONTROL controlBitsOfInterest = 0;
  727. SECURITY_DESCRIPTOR_CONTROL controlBitsToSet = 0;
  728. SECURITY_DESCRIPTOR_CONTROL oldControlBits = 0;
  729. DWORD dwRevision = 0;
  730. if (!GetSecurityDescriptorControl(pFileSD, &oldControlBits, &dwRevision)) {
  731. _tprintf(TEXT("GetSecurityDescriptorControl() failed. Error %d/n"), GetLastError());
  732. __leave;
  733. }
  734. if (oldControlBits & SE_DACL_AUTO_INHERITED) {
  735. controlBitsOfInterest =
  736. SE_DACL_AUTO_INHERIT_REQ |
  737. SE_DACL_AUTO_INHERITED ;
  738. controlBitsToSet = controlBitsOfInterest;
  739. }
  740. else if (oldControlBits & SE_DACL_PROTECTED) {
  741. controlBitsOfInterest = SE_DACL_PROTECTED;
  742. controlBitsToSet = controlBitsOfInterest;
  743. }
  744. if (controlBitsOfInterest) {
  745. if (!_SetSecurityDescriptorControl(&newSD,controlBitsOfInterest,controlBitsToSet)) {
  746. _tprintf(TEXT("SetSecurityDescriptorControl() failed. Error %d/n"), GetLastError());
  747. __leave;
  748. }
  749. }
  750. }
  751. //
  752. // STEP 20: 把新的SD设置设置到文件的安全属性中(千山万水啊,终于到了)
  753. //
  754. if (!SetFileSecurity(lpszFileName, secInfo,&newSD)) {
  755. _tprintf(TEXT("SetFileSecurity() failed. Error %d/n"), GetLastError());
  756. __leave;
  757. }
  758. fResult = TRUE;
  759. }
  760. __finally
  761. {
  762. //
  763. // STEP 21: 释放已分配的内存,以免Memory Leak
  764. //
  765. if (pUserSID) myheapfree(pUserSID);
  766. if (szDomain) myheapfree(szDomain);
  767. if (pFileSD) myheapfree(pFileSD);
  768. if (pNewACL) myheapfree(pNewACL);
  769. }
  770. return fResult;
  771. }