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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 打印编程 > 查看源码
Monitor.cpp
资源名称：PopFaxPrinter_src_v2.01.zip　[点击查看]
上传用户：looem2003
上传日期：2014-07-20
资源大小：13733k
文件大小：10k
源码类别：

打印编程

开发平台：

Visual C++

Monitor.cpp：源码内容
  1. /*
  2.    Copyright (C) 2007    Ibadov Tariel   <itk@bk.ru>
  3. */
  6. #define MONITOR_EXPORTS
  8. #include <fstream.h>
  9. #include <iostream.h>
  10. #include "PortList.h"
  11. #include "Monitor.h"
  12. #include "rpcl.h"
  14. #define szLocalMonitor _T("ImagePrinter Port")
  16. HINSTANCE hInst=NULL;
  17. HINSTANCE hSpoolssDll=NULL;
  19. CRITICAL_SECTION SpoolerSection;
  21. HANDLE m_hEvent=NULL;
  22. CPortList *m_portlist=NULL;
  23. TCHAR *m_sRegistryRoot=NULL;
  25. typedef BOOL (WINAPI *fpENUMPORTS)
  26. (
  27. LPWSTR  pName,
  28. DWORD   Level,
  29. LPBYTE  pPorts,
  30. DWORD   cbBuf,
  31. LPDWORD pcbNeeded,
  32. LPDWORD pcReturned
  33. );
  36. BOOL PortExists(TCHAR *pServerName,TCHAR *pPortName,DWORD *pError)
  37. {
  38. BOOL Found = TRUE;
  39. fpENUMPORTS pfnSpoolssEnumPorts;
  40. DWORD cbNeeded;
  41.     DWORD cReturned;
  42.     DWORD cbPorts;
  43.     LPPORT_INFO_1 pPorts;
  44.     *pError = NO_ERROR;
  45.     if (!hSpoolssDll)
  46.         hSpoolssDll = LoadLibrary(L"SPOOLSS.DLL");
  47. if(!hSpoolssDll){
  48. *pError=GetLastError();
  49. }
  50. else
  51. {
  52. pfnSpoolssEnumPorts = (fpENUMPORTS)GetProcAddress(hSpoolssDll,"EnumPortsW");
  53.         if(!pfnSpoolssEnumPorts){
  54. *pError = GetLastError();
  55. FreeLibrary(hSpoolssDll);
  56. hSpoolssDll = NULL;
  57. }
  58. }
  59.     if (!pfnSpoolssEnumPorts)
  60.         Found=TRUE;
  61.     (*pfnSpoolssEnumPorts)(pServerName, 1, NULL, 0, &cbNeeded, &cReturned);
  62.     if (GetLastError() != ERROR_INSUFFICIENT_BUFFER){
  63. Found=TRUE;
  64. }
  65. else{
  66.         cbPorts = cbNeeded;
  67.         pPorts = (LPPORT_INFO_1)new BYTE[cbPorts];
  69.         if(!pPorts){
  70. Found=TRUE;
  71. }
  72. else{
  73.             if(!(*pfnSpoolssEnumPorts)(pServerName,1,(LPBYTE)pPorts,cbPorts,&cbNeeded,&cReturned)){
  74. Found=TRUE;
  75. }
  76. else{
  77.                 Found = FALSE;
  78.                 for (DWORD z = 0; z < cReturned; z++){
  79.                     if(_tcsicmp(pPorts[z].pName, pPortName)==0)
  80.                         Found = TRUE;
  81.                 }
  82.             }
  83.         }
  84.         delete pPorts;
  85.     }
  86.     return Found;
  87. }
  89. BOOL FileExists(TCHAR *sFilePath)
  90. {
  91. WIN32_FIND_DATA w32fd;
  92. HANDLE hFile=FindFirstFile(sFilePath,&w32fd);
  93. if(hFile!=INVALID_HANDLE_VALUE){
  94. FindClose(hFile);
  95. return TRUE;
  96. }
  97. return FALSE; 
  98. }
  100. BOOL APIENTRY DllMain(HINSTANCE hModule, DWORD  dwReason, LPVOID lpRes)
  101. {
  102. switch (dwReason)
  103. {
  104. case DLL_PROCESS_ATTACH:
  105.                 init_log( _T("imgport"), &hModule, 1 ); 
  106. case DLL_THREAD_ATTACH:
  107.         hInst = hModule;
  108.         InitializeCriticalSection(&SpoolerSection);
  109.          DisableThreadLibraryCalls((HINSTANCE)hModule);
  110. m_portlist=new CPortList();
  111.         return TRUE;
  112. case DLL_PROCESS_DETACH:
  113. case DLL_THREAD_DETACH:
  114. if(m_hEvent)
  115. CloseHandle(m_hEvent);
  116. if(m_sRegistryRoot)
  117. free((void*)m_sRegistryRoot);
  118. if(hSpoolssDll)
  119. FreeLibrary(hSpoolssDll);
  120. delete m_portlist;
  121.         return TRUE;
  122.     }
  123.     UNREFERENCED_PARAMETER( lpRes );
  124. return TRUE;
  125. }
  127. void EnterCritSection()
  128. {
  129. EnterCriticalSection(&SpoolerSection);
  130. }
  132. void LeaveCritSection()
  133. {
  134. LeaveCriticalSection(&SpoolerSection);
  135. }
  138. BOOL WINAPI iEnumPorts(LPWSTR pName,DWORD dwLevel,LPBYTE pPorts,DWORD cbBuf,LPDWORD pdwNeeded,LPDWORD pdwReturned)
  139. {
  140. EnterCritSection();
  141. BOOL bResult = m_portlist->EnumPorts(pName,dwLevel,pPorts,cbBuf,pdwNeeded,pdwReturned);
  142. LeaveCritSection();
  143. return bResult;
  144. }
  146. BOOL WINAPI iOpenPortEx(LPWSTR   pName,LPWSTR pPrinterName,PHANDLE pHandle, struct _MONITOR *pMonitor)
  147. {
  148. EnterCritSection();
  149. PORT *pPort=m_portlist->FindPort(pName);
  150. *pHandle=(HANDLE)pPort;
  151. pPort->dwStatus |= PS_OPENED;
  152. LeaveCritSection();
  153. return pPort!=NULL;
  154. }
  156. BOOL WINAPI iOpenPort(LPWSTR   pName,PHANDLE pHandle)
  157. {
  158. EnterCritSection();
  159. PORT *pPort=m_portlist->FindPort(pName);
  160. *pHandle=(HANDLE)pPort;
  161. pPort->dwStatus |= PS_OPENED;
  162. LeaveCritSection();
  163. return pPort!=NULL;
  164. }
  166. BOOL WINAPI iStartDocPort(HANDLE  hPort,LPWSTR  pPrinterName,DWORD   JobId, DWORD   Level, LPBYTE  pDocInfo)
  167. {
  168. EnterCritSection();
  169. PORT *pPort=(PORT*)hPort;
  170. TCHAR filepath[MAX_PATH];
  171.     TCHAR filename[MAX_PATH];
  172. LPTSTR buff = NULL;
  174. if(pPort->dwStatus & PS_STARTDOC)
  175. {
  176. LeaveCritSection();
  177. return FALSE;
  178. }
  180. _tcscpy( filepath, pPort->cPath);
  182. switch(Level)
  183. {
  184. case 1:{
  185. DOC_INFO_1 *pDoc=(DOC_INFO_1*)pDocInfo;
  186. _tcscpy(filename,pDoc->pDocName);
  187. break;
  188. }
  189. case 2:{
  190. DOC_INFO_2 *pDoc=(DOC_INFO_2*)pDocInfo;
  191. _tcscpy(filename,pDoc->pDocName);
  192. break;
  193. }
  194. }
  195. //_tcscat( filepath, filename);
  196. _tcscpy( filepath, _T("fax_image"));
  198.   if( FileExists(filepath) ) 
  199. DeleteFile(filepath);
  200. pPort->hFile = ::CreateFile( filepath, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
  201.         write_log( 3, _T("iStartDocPort"), _T("%s"), filepath );
  202.     if(pPort->hFile==INVALID_HANDLE_VALUE)
  203. {
  204.         _tcscat(filepath,_T(" : ERROR CreateFile !!!"));                                                       
  205.        MessageBox(HWND_DESKTOP,filepath,_T("ImagePrinter") ,MB_OK);
  206. pPort->hFile=0;
  207. HANDLE hPrinter;
  208. if(OpenPrinter(pPrinterName,&hPrinter,NULL))
  209. {
  210. SetJob(hPrinter,JobId,0,NULL,JOB_CONTROL_RESTART);
  211. SetJob(hPrinter,JobId,0,NULL,JOB_CONTROL_PAUSE);
  212. CloseHandle(hPrinter);
  213. LeaveCritSection();
  214. return FALSE;
  215. }
  216. }
  217. else
  218. {
  219. _tcscpy(pPort->sCurDocument,filepath);
  220. _tcscpy(pPort->sPrinter,pPrinterName);
  221. pPort->dwStatus|=PS_STARTDOC;
  222. pPort->dwJobID=JobId;
  223. }
  224. LeaveCritSection();
  225. return TRUE;
  226. }
  228. BOOL WINAPI iWritePort( HANDLE  hPort, LPBYTE  pBuffer,DWORD   cbBuf,LPDWORD pcbWritten)
  229. {
  230. PORT *pPort=(PORT*)hPort;
  231. if(pPort->hFile){
  232. EnterCritSection();
  233. BOOL bResult=WriteFile(pPort->hFile,pBuffer,cbBuf,pcbWritten,NULL);
  234. LeaveCritSection();
  235. if(!bResult){
  236. HANDLE hPrinter;
  237. if(OpenPrinter(pPort->sPrinter,&hPrinter,NULL)){
  238. SetJob(hPrinter,pPort->dwJobID,0,NULL,JOB_CONTROL_RESTART);
  239. SetJob(hPrinter,pPort->dwJobID,0,NULL,JOB_CONTROL_PAUSE);
  240. CloseHandle(hPrinter);
  241. }
  242. }
  243. return TRUE;
  244. }
  245. return TRUE;
  246. }
  248. BOOL WINAPI iReadPort(HANDLE hPort,LPBYTE pBuffer,DWORD  cbBuf,LPDWORD pcbRead)
  249. {
  250. return TRUE;
  251. }
  253. BOOL WINAPI iEndDocPort(HANDLE hPort)
  254. {
  255. PORT *pPort=(PORT*)hPort;
  256. EnterCritSection();
  257. if(pPort->hFile){
  258. FlushFileBuffers(pPort->hFile);
  259.     CloseHandle(pPort->hFile);
  260.         std::string  s_tmp;
  261.         std::string  extApp;
  263.       for (int i=0;i<MAX_PATH;i++)
  264.      s_tmp+=pPort->sCurDocument[i];
  265.         WideCharToMultiByte( CP_ACP, 0, pPort->sCurDocument, -1, (char *)s_tmp.c_str(), MAX_PATH-1, NULL, NULL);
  266.         //s_tmp = pPort->sCurDocument;
  267.         
  268.       for (int i=0;i<MAX_PATH;i++)
  269.      extApp+=pPort->extApp[i];
  270.         int icount = 0, compress = 0;;
  271.         int isFormat=0;
  272.                     if (_tcsncmp(pPort->img_format,_T("tif"), 3) == 0) 
  273.                     {
  274.                        isFormat=1;
  275.                        if (_tcsicmp(pPort->img_format_ext,_T("G4")) == 0)
  276.                            compress = COMPRESSION_CCITTFAX4;
  277.                        else
  278.                            if (_tcsicmp(pPort->img_format_ext,_T("G3")) == 0)
  279.                                compress = COMPRESSION_CCITTFAX3; 
  280.                        else
  281.                            compress = COMPRESSION_LZW;                       
  282.                     }
  283.                     if (_tcsncmp(pPort->img_format,_T("png"), 3) == 0) isFormat=2;                  
  284.                     if (_tcsncmp(pPort->img_format,_T("bmp"), 3) == 0) isFormat=3;      
  285. if (_tcsncmp(pPort->img_format,_T("pdf"), 3) == 0) isFormat=4;      
  286.                 write_log( 3, "iEndDocPort", "%s %s %d %s", pPort->img_format, pPort->img_format_ext, pPort->dither, s_tmp.c_str() );  
  287.            icount = start_convert( (char *)s_tmp.c_str(), (char *)s_tmp.c_str(), isFormat, compress, pPort->dither, (char*)extApp.c_str());
  288.              if(FileExists(pPort->sCurDocument))DeleteFile(pPort->sCurDocument);
  289. PulseEvent(m_hEvent);
  290. _tcscpy(pPort->sCurDocument,_T(""));
  291. pPort->dwStatus &= ~PS_STARTDOC;
  292. pPort->hFile=0;
  293. HANDLE hPrinter;
  294. if(OpenPrinter(pPort->sPrinter,&hPrinter,NULL)){
  295. SetJob(hPrinter,pPort->dwJobID,0,NULL,JOB_CONTROL_DELETE);
  296. CloseHandle(hPrinter);
  297. }
  298. }
  299. LeaveCritSection();
  300. return TRUE;
  301. }
  303. BOOL WINAPI iClosePort(HANDLE hPort)
  304. {
  305. EnterCritSection();
  306. PORT *pPort=(PORT*)hPort;
  307. pPort->dwStatus &= ~PS_OPENED;
  308. LeaveCritSection();
  309. return TRUE;
  310. }
  312. BOOL WINAPI iAddPort(LPWSTR pName,HWND hWnd,LPWSTR pMonitorName)
  313. {
  314.     TCHAR tempdir[MAX_PATH] = _T("");
  315.     GetTempPath( MAX_PATH,tempdir);
  316.     EnterCritSection();
  317. m_portlist->AddPort( _T("tif"), tempdir, _T("ImagePrinter Port"), 0, NULL, FALSE, NULL);
  318. m_portlist->Save(m_sRegistryRoot);
  319. LeaveCritSection();
  320.     return TRUE;
  321. }
  323. BOOL WINAPI iAddPortEx( LPWSTR pName,DWORD Level,LPBYTE pBuffer,LPWSTR pMonitorName)
  324. {
  325. return TRUE;
  326. }
  328. BOOL WINAPI iConfigurePort( LPWSTR pName,HWND hWnd,LPWSTR pPortName)
  329. {
  330. MessageBox(hWnd,_T("This port cannot be configured."),szLocalMonitor,MB_OK);
  331. return TRUE;
  332. }
  334. BOOL WINAPI iDeletePort(LPWSTR pName,HWND hWnd,LPWSTR pPortName)
  335. {
  336. EnterCritSection();
  337. if(m_portlist->DeletePort(pPortName))
  338. {
  339. m_portlist->Save(m_sRegistryRoot);
  340. LeaveCritSection();
  341. TCHAR buf[512];
  342.         _stprintf(buf,_T("ImagePrinter port successfully removed."),pPortName);
  343. MessageBox(hWnd,buf,_T("Port removed"),MB_OK);
  344. return TRUE;
  345. }
  346. LeaveCritSection();
  347. MessageBox(hWnd,_T("An unknown error nn Couldn't remove the port."),_T("Error"),MB_OK);
  348. return TRUE;
  349. }
  351. BOOL IsFileReady(TCHAR *sPath)
  352. {
  353. HANDLE hFile=CreateFile(sPath,GENERIC_READ,FILE_SHARE_WRITE|FILE_SHARE_READ,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
  354. if(hFile==INVALID_HANDLE_VALUE){
  355. return FALSE;
  356. }
  357. else{
  358. CloseHandle(hFile);
  359. return TRUE;
  360. }
  361. }
  364. MONITOREX MonitorEx = {
  365.     sizeof(MONITOR),
  366.     {
  367.         iEnumPorts,
  368.         iOpenPort,
  369.         iOpenPortEx,
  370.         iStartDocPort,
  371.         iWritePort,
  372. iReadPort,
  373.         iEndDocPort,
  374.         iClosePort,
  375.         iAddPort,
  376.         iAddPortEx,
  377.         iConfigurePort,
  378.         iDeletePort,
  379.         NULL, //iGetPrinterDataFromPort
  380.         NULL //iSetPortTimeOuts
  381.     }
  382. };
  384. LPMONITOREX WINAPI InitializePrintMonitor( LPWSTR pRegistryRoot)
  385. {
  386. if(m_sRegistryRoot) free((void*)m_sRegistryRoot);
  387. m_sRegistryRoot=_tcsdup(pRegistryRoot);
  388. m_hEvent=CreateEvent(NULL,FALSE,FALSE,szLocalMonitor);
  389. EnterCritSection();
  390. m_portlist->Load(m_sRegistryRoot);
  391. LeaveCritSection();
  392. return &MonitorEx;
  393. }