开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 嵌入式/单片机编程 > Windows CE > 查看源码
association_win32.c
资源名称：tcpmp.rar　[点击查看]
上传用户：wstnjxml
上传日期：2014-04-03
资源大小：7248k
文件大小：11k
源码类别：

Windows CE

开发平台：

C/C++

association_win32.c：源码内容
  1. /*****************************************************************************
  2.  *
  3.  * This program is free software ; you can redistribute it and/or modify
  4.  * it under the terms of the GNU General Public License as published by
  5.  * the Free Software Foundation; either version 2 of the License, or
  6.  * (at your option) any later version.
  7.  *
  8.  * This program is distributed in the hope that it will be useful,
  9.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11.  * GNU General Public License for more details.
  12.  *
  13.  * You should have received a copy of the GNU General Public License
  14.  * along with this program; if not, write to the Free Software
  15.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16.  *
  17.  * $Id: association_win32.c 603 2006-01-19 13:00:33Z picard $
  18.  *
  19.  * The Core Pocket Media Player
  20.  * Copyright (c) 2004-2005 Gabor Kovacs
  21.  *
  22.  ****************************************************************************/
  24. #include "../common.h"
  26. #if defined(TARGET_WIN32) || defined(TARGET_WINCE)
  28. #define REG_ASSOCIATIONS 0x2C00
  30. #ifndef STRICT
  31. #define STRICT
  32. #endif
  33. #include <windows.h>
  35. static void SetURLProtocol(const tchar_t* Base)
  36. {
  37. const tchar_t* Name = T("URL Protocol");
  38. tchar_t Value[MAXPATH];
  39. DWORD RegType;
  40. DWORD RegSize;
  41. DWORD Disp=0;
  42. HKEY Key;
  43. if (RegCreateKeyEx(HKEY_CLASSES_ROOT, Base, 0, NULL, 0, KEY_READ|KEY_WRITE, NULL, &Key, &Disp) == ERROR_SUCCESS)
  44. {
  45. RegSize=sizeof(Value);
  46. if (RegQueryValueEx(Key, NULL, 0, &RegType, (LPBYTE)Value, &RegSize) != ERROR_SUCCESS || 
  47. tcsnicmp(Value,T("URL:"),4)!=0)
  48. {
  49. stprintf_s(Value,TSIZEOF(Value),T("URL:%s protocol"),Base);
  50. RegSetValueEx(Key, NULL, 0, REG_SZ, (LPBYTE)Value, (DWORD)((tcslen(Value)+1)*sizeof(tchar_t)));
  51. }
  53. RegSize=sizeof(Value);
  54. if (RegQueryValueEx(Key, Name, 0, &RegType, (LPBYTE)Value, &RegSize) != ERROR_SUCCESS)
  55. RegSetValueEx(Key, Name, 0, REG_SZ, (LPBYTE)T(""), sizeof(tchar_t));
  57. RegCloseKey(Key);
  58. }
  59. }
  61. static void SetEditFlags(const tchar_t* Base,DWORD Value)
  62. {
  63. DWORD Disp=0;
  64. HKEY Key;
  65. if (RegCreateKeyEx(HKEY_CLASSES_ROOT, Base, 0, NULL, 0, KEY_READ|KEY_WRITE, NULL, &Key, &Disp) == ERROR_SUCCESS)
  66. {
  67. RegSetValueEx(Key, T("EditFlags"), 0, REG_DWORD, (LPBYTE)&Value, sizeof(Value));
  68. RegCloseKey(Key);
  69. }
  70. //BrowserFlags=8 (?)
  71. }
  73. static void SetReg(const tchar_t* Base,const tchar_t* New,bool_t State)
  74. {
  75. tchar_t Old[MAXPATH];
  76. tchar_t Backup[MAXPATH];
  77. DWORD Disp=0;
  78. HKEY Key;
  79. DWORD RegSize;
  80. DWORD RegType;
  82. stprintf_s(Backup,TSIZEOF(Backup),T("%s.bak"),Context()->ProgramName);
  84. if (RegOpenKeyEx(HKEY_CLASSES_ROOT, Base, 0, KEY_READ|KEY_WRITE, &Key) == ERROR_SUCCESS)
  85. {
  86. RegSize=sizeof(Old);
  87. if (RegQueryValueEx(Key, NULL, 0, &RegType, (LPBYTE)Old, &RegSize) != ERROR_SUCCESS)
  88. Old[0] = 0;
  90. if (tcsicmp(Old,New)!=0)
  91. {
  92. if (State)
  93. {
  94. RegSetValueEx(Key, NULL, 0, REG_SZ, (LPBYTE)New, (DWORD)((tcslen(New)+1)*sizeof(tchar_t)));
  95. RegSetValueEx(Key, Backup, 0, REG_SZ, (LPBYTE)Old, RegSize);
  96. }
  97. }
  98. else 
  99. if (!State)
  100. {
  101. RegSize = sizeof(Old);
  102. if (RegQueryValueEx(Key, Backup, 0, &RegType, (LPBYTE)Old, &RegSize) == ERROR_SUCCESS)
  103. {
  104. RegSetValueEx(Key, NULL, 0, REG_SZ, (LPBYTE)Old, RegSize);
  105. RegDeleteValue(Key, Backup);
  106. }
  107. }
  109. RegCloseKey(Key);
  110. }
  111. else 
  112. if (State)
  113. {
  114. if (RegCreateKeyEx(HKEY_CLASSES_ROOT, Base, 0, NULL, 0, KEY_READ|KEY_WRITE, NULL, &Key, &Disp) == ERROR_SUCCESS)
  115. {
  116. RegSetValueEx(Key, NULL, 0, REG_SZ, (LPBYTE)New, (DWORD)((tcslen(New)+1)*sizeof(New)));
  117. RegSetValueEx(Key, Backup, 0, REG_SZ, (LPBYTE)T(""), sizeof(tchar_t));
  118. RegCloseKey(Key);
  119. }
  120. }
  121. }
  123. static bool_t CmpReg(const tchar_t* Base, const tchar_t* Value)
  124. {
  125. bool_t Result = 0;
  126. HKEY Key;
  127. DWORD RegSize;
  128. DWORD RegType;
  129. tchar_t s[MAXPATH];
  131. if (RegOpenKeyEx(HKEY_CLASSES_ROOT, Base, 0, KEY_READ, &Key) == ERROR_SUCCESS)
  132. {
  133. RegSize = sizeof(s);
  134. if (RegQueryValueEx(Key, NULL, 0, &RegType, (LPBYTE)s, &RegSize) == ERROR_SUCCESS && RegType == REG_SZ)
  135. Result = tcsicmp(s,Value)==0;
  137. RegCloseKey(Key);
  138. }
  139. return Result;
  140. }
  142. static void SetFileAssociation(const tchar_t* Ext,bool_t State,bool_t MIME,bool_t PlayList)
  143. {
  144. tchar_t Base[64];
  145. tchar_t Type[64];
  146. tchar_t Path[MAXPATH];
  147. tchar_t Open[MAXPATH];
  148. tchar_t Icon[MAXPATH];
  150. GetModuleFileName(NULL,Path,MAXPATH);
  151. if (tcschr(Path,' '))
  152. stprintf_s(Open,TSIZEOF(Open),T(""%s" "%%1""),Path);
  153. else
  154. stprintf_s(Open,TSIZEOF(Open),T("%s "%%1""),Path);
  155. stprintf_s(Icon,TSIZEOF(Icon),T("%s, -%d"),Path,1000);
  157. stprintf_s(Base,TSIZEOF(Base),T(".%s"),Ext);
  158. stprintf_s(Type,TSIZEOF(Type),MIME?T("%s"):T("%sFile"),Ext);
  159. tcsupr(Type);
  161. if (!MIME)
  162. {
  163. SetReg(Base,Type,State);
  164. if (State)
  165. SetEditFlags(Type,PlayList?0x10010000:0x10000); 
  166. }
  167. else
  168. if (State)
  169. SetURLProtocol(Type);
  171. stprintf_s(Base,TSIZEOF(Base),T("%s\DefaultIcon"),Type);
  172. SetReg(Base,Icon,State);
  174. stprintf_s(Base,TSIZEOF(Base),T("%s\Shell\Open\Command"),Type);
  175. SetReg(Base,Open,State);
  176. }
  178. #if defined(NDEBUG) && !defined(TARGET_WIN32)
  179. static bool_t GetReg(const tchar_t* Base, tchar_t* Value,DWORD ValueSize)
  180. {
  181. bool_t Result = 0;
  182. HKEY Key;
  183. DWORD RegType;
  185. if (RegOpenKeyEx(HKEY_CLASSES_ROOT, Base, 0, KEY_READ, &Key) == ERROR_SUCCESS)
  186. {
  187. if (RegQueryValueEx(Key, NULL, 0, &RegType, (LPBYTE)Value, &ValueSize) == ERROR_SUCCESS && RegType == REG_SZ && Value[0])
  188. Result = 1;
  190. RegCloseKey(Key);
  191. }
  192. return Result;
  193. }
  195. static bool_t EmptyFileAssociation(const tchar_t* Ext,bool_t MIME)
  196. {
  197. tchar_t Base[MAXPATH];
  198. tchar_t Type[MAXPATH];
  200. stprintf_s(Base,TSIZEOF(Base),T(".%s"),Ext);
  201. if (MIME || GetReg(Base,Type,sizeof(Type)))
  202. {
  203. if (MIME)
  204. tcscpy_s(Type,TSIZEOF(Type),Ext);
  205. stprintf_s(Base,TSIZEOF(Base),T("%s\Shell\Open\Command"),Type);
  206. if (GetReg(Base,Type,sizeof(Type)))
  207. return 0;
  208. }
  209. return 1;
  210. }
  211. #endif
  213. static bool_t GetFileAssociation(const tchar_t* Ext,bool_t MIME)
  214. {
  215. tchar_t Base[64];
  216. tchar_t Type[64];
  218. stprintf_s(Base,TSIZEOF(Base),T(".%s"),Ext);
  219. stprintf_s(Type,TSIZEOF(Type),MIME?T("%s"):T("%sFile"),Ext);
  220. tcsupr(Type);
  222. if (MIME || CmpReg(Base,Type))
  223. {
  224. tchar_t Path[MAXPATH];
  225. tchar_t Open[MAXPATH];
  227. GetModuleFileName(NULL,Path,MAXPATH);
  228. if (tcschr(Path,' '))
  229. stprintf_s(Open,TSIZEOF(Open),T(""%s" "%%1""),Path);
  230. else
  231. stprintf_s(Open,TSIZEOF(Open),T("%s "%%1""),Path);
  233. stprintf_s(Base,TSIZEOF(Base),T("%s\Shell\Open\Command"),Type);
  234. if (CmpReg(Base,Open))
  235. return 1;
  236. }
  238. return 0;
  239. }
  241. static int Enum( node* p, int* No, datadef* Param )
  242. {
  243. array List;
  245. NodeEnumClass(&List,MEDIA_CLASS);
  246. if (*No>=0 && *No<ARRAYCOUNT(List,int))
  247. {
  248. memset(Param,0,sizeof(datadef));
  249. Param->No = ARRAYBEGIN(List,int)[*No];
  250. Param->Name = LangStr(Param->No,NODE_NAME);
  251. Param->Type = TYPE_BOOL;
  252. Param->Size = sizeof(bool_t);
  253. Param->Flags = DF_SETUP|DF_NOSAVE|DF_CHECKLIST;
  254. Param->Class = ASSOCIATION_ID;
  255. if (!UniqueExts(ARRAYBEGIN(List,int),ARRAYBEGIN(List,int)+*No))
  256. Param->Flags = DF_HIDDEN;
  257. ArrayClear(&List);
  258. return ERR_NONE;
  259. }
  260. *No -= ARRAYCOUNT(List,int);
  261. ArrayClear(&List);
  263. NodeEnumClass(&List,STREAM_CLASS);
  264. if (*No>=0 && *No<ARRAYCOUNT(List,int))
  265. {
  266. tchar_t Mime[MAXPATH];
  267. bool_t HasHost;
  269. memset(Param,0,sizeof(datadef));
  270. Param->No = ARRAYBEGIN(List,int)[*No];
  271. Param->Name = LangStr(Param->No,NODE_NAME);
  272. Param->Type = TYPE_BOOL;
  273. Param->Size = sizeof(bool_t);
  274. Param->Flags = DF_SETUP|DF_NOSAVE|DF_CHECKLIST;
  275. Param->Class = ASSOCIATION_ID;
  277. stprintf_s(Mime,TSIZEOF(Mime),T("%s://"),LangStr(Param->No,NODE_CONTENTTYPE));
  278. GetMime(Mime,NULL,0,&HasHost);
  279. if (!HasHost || tcsicmp(Mime,T("://"))==0 || tcsicmp(Mime,T("http://"))==0)
  280. Param->Flags = DF_HIDDEN;
  281. ArrayClear(&List);
  282. return ERR_NONE;
  283. }
  284. *No -= ARRAYCOUNT(List,int);
  285. ArrayClear(&List);
  287. return ERR_INVALID_PARAM;
  288. }
  290. static int Get(node* p, int No, void* Data, int Size)
  291. {
  292. if (NodeIsClass(No,STREAM_CLASS))
  293. {
  294. assert(Size==sizeof(bool_t));
  295. *(bool_t*)Data = GetFileAssociation(LangStr(No,NODE_CONTENTTYPE),1);
  296. return ERR_NONE;
  297. }
  298. else
  299. if (NodeIsClass(No,MEDIA_CLASS))
  300. {
  301. const tchar_t* Exts = LangStr(No,NODE_EXTS);
  302. if (Exts && Exts[0])
  303. {
  304. tchar_t s[16];
  305. const tchar_t *r,*q;
  307. assert(Size==sizeof(bool_t));
  308. *(bool_t*)Data = 1;
  309. for (r=Exts;r;)
  310. {
  311. q = tcschr(r,':');
  312. if (q)
  313. {
  314. tcsncpy_s(s,TSIZEOF(s),r,q-r);
  315. if (!GetFileAssociation(s,0))
  316. {
  317. *(bool_t*)Data = 0;
  318. break;
  319. }
  320. }
  321. r = tcschr(r,';');
  322. if (r) ++r;
  323. }
  324. return ERR_NONE;
  325. }
  326. }
  327. return ERR_INVALID_PARAM;
  328. }
  330. static int Set(node* p, int No, const void* Data, int Size)
  331. {
  332. if (NodeIsClass(No,STREAM_CLASS))
  333. {
  334. assert(Size==sizeof(bool_t));
  335. SetFileAssociation(LangStr(No,NODE_CONTENTTYPE),*(bool_t*)Data,1,0);
  336. return ERR_NONE;
  337. }
  338. else
  339. if (NodeIsClass(No,MEDIA_CLASS))
  340. {
  341. const tchar_t* Exts = LangStr(No,NODE_EXTS);
  342. if (Exts && Exts[0])
  343. {
  344. tchar_t s[16];
  345. const tchar_t *r,*q;
  346. bool_t State;
  348. assert(Size==sizeof(bool_t));
  349. State = *(bool_t*)Data;
  351. for (r=Exts;r;)
  352. {
  353. q = tcschr(r,':');
  354. if (q)
  355. {
  356. tcsncpy_s(s,TSIZEOF(s),r,q-r);
  357. SetFileAssociation(s,State,0,q[1]==FTYPE_PLAYLIST);
  358. }
  359. r = tcschr(r,';');
  360. if (r) ++r;
  361. }
  362. return ERR_NONE;
  363. }
  364. }
  365. return ERR_INVALID_PARAM;
  366. }
  368. static void AssignEmpty(node* p)
  369. {
  370. #if defined(NDEBUG) && !defined(TARGET_WIN32)
  371. int v;
  372. if (!NodeRegLoadValue(0,REG_ASSOCIATIONS,&v,sizeof(v),TYPE_INT) || !v)
  373. {
  374. // set default associations for empty extensions
  375. array List;
  376. int* i;
  378. NodeEnumClass(&List,MEDIA_CLASS);
  380. for (i=ARRAYBEGIN(List,int);i!=ARRAYEND(List,int);++i)
  381. {
  382. const tchar_t* Exts = LangStr(*i,NODE_EXTS);
  383. if (Exts && Exts[0])
  384. {
  385. bool_t Empty = 1;
  386. tchar_t s[16];
  387. const tchar_t *r,*q;
  389. for (r=Exts;r;)
  390. {
  391. q = tcschr(r,':');
  392. if (q)
  393. {
  394. tcsncpy_s(s,TSIZEOF(s),r,q-r);
  395. if (!EmptyFileAssociation(s,0))
  396. {
  397. Empty = 0;
  398. break;
  399. }
  400. }
  401. r = tcschr(r,';');
  402. if (r) ++r;
  403. }
  405. if (Empty)
  406. {
  407. bool_t True = 1;
  408. Set(p,*i,&True,sizeof(True));
  409. }
  410. }
  411. }
  413. ArrayClear(&List);
  415. v = 1;
  416. NodeRegSaveValue(0,REG_ASSOCIATIONS,&v,sizeof(v),TYPE_INT);
  417. }
  418. #endif
  419. }
  421. static int Create(node* p)
  422. {
  423. p->Enum = (nodeenum)Enum;
  424. p->Get = (nodeget)Get;
  425. p->Set = (nodeset)Set;
  426. AssignEmpty(p);
  427. return ERR_NONE;
  428. }
  430. static const nodedef Association =
  431. {
  432. sizeof(node)|CF_GLOBAL|CF_SETTINGS,
  433. ASSOCIATION_ID,
  434. NODE_CLASS,
  435. PRI_MAXIMUM+500,
  436. (nodecreate)Create,
  437. };
  439. void Association_Init()
  440. {
  441. NodeRegisterClass(&Association);
  442. }
  444. void Association_Done()
  445. {
  446. NodeUnRegisterClass(ASSOCIATION_ID);
  447. }
  449. #endif