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

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

打印编程

开发平台：

Visual C++

PortList.cpp：源码内容
  2. #include "PortList.h"
  4. CPortList::CPortList(TCHAR *sMonitorName,TCHAR *sPortDesc)
  5. {
  6.         if(sMonitorName) m_sMonitorName=_tcsdup(sMonitorName);
  7.         else m_sMonitorName=_tcsdup(_T(""));
  8.         if(sPortDesc)        m_sPortDesc=_tcsdup(sPortDesc);
  9.         else m_sPortDesc=_tcsdup(_T(""));
  10.         m_pFirst=new PORT;
  11.         m_pFirst->pNext=NULL;
  12.         m_pFirst->dwStatus=0;
  13.         m_pFirst->hFile=0;
  14.         m_pFirst->dwJobID=0;
  15.         m_pFirst->dither = FALSE;
  16.         _tcscpy(m_pFirst->sPath,_T(""));
  17.         _tcscpy(m_pFirst->sCurDocument,_T(""));
  18.         _tcscpy(m_pFirst->sPrinter,_T(""));
  19. }
  21. CPortList::~CPortList()
  22. {
  23.         free(m_sMonitorName);
  24.         free(m_sPortDesc);
  25.         PORT *pNext;
  26.         while(m_pFirst){
  27.                 pNext=m_pFirst->pNext;
  28.                 delete m_pFirst;
  29.                 m_pFirst=pNext;
  30.         }
  31. }
  33. BOOL CPortList::EnumPorts(LPWSTR   pName,DWORD   Level,LPBYTE  pPorts,DWORD   cbBuf,LPDWORD pcbNeeded,        LPDWORD pcReturned)
  34. {
  35.     LPBYTE  pEnd;
  36.     DWORD   LastError=0;
  37.     DWORD cb=0;
  38.         PORT *pPort=m_pFirst;
  39.     while (pPort->pNext){
  40.         cb+=GetPortSize(pPort->sPath, Level);
  41.         pPort=pPort->pNext;
  42.     }
  43.     *pcbNeeded=cb;
  44.     if (cb <= cbBuf){
  45.                 pEnd=pPorts+cbBuf;
  46.         *pcReturned=0;
  48.         pPort=m_pFirst;
  49.         while (pPort->pNext){
  50.             pEnd = CopyPortToBuffer(pPort, Level, pPorts, pEnd);
  51.             switch (Level)
  52.                         {
  53.             case 1:
  54.                 pPorts+=sizeof(PORT_INFO_1);
  55.                 break;
  56.             case 2:
  57.                 pPorts+=sizeof(PORT_INFO_2);
  58.                 break;
  59.             default:
  60.                 LastError = ERROR_INVALID_LEVEL;
  61.                 goto Cleanup;
  62.             }
  63.             (*pcReturned)++;
  65.             pPort=pPort->pNext;
  66.         }
  67.     }
  68.     else LastError = ERROR_INSUFFICIENT_BUFFER;
  71. Cleanup:
  72.     if (LastError)        {
  73.         SetLastError(LastError);
  74.         return FALSE;
  75.     }
  76.         else return TRUE;
  77. }
  79. DWORD CPortList::GetPortSize(TCHAR *pName,DWORD dwLevel)
  80. {
  81.     DWORD   cb;
  82.     switch (dwLevel) {
  84.     case 1:
  86.         cb=sizeof(PORT_INFO_1) +
  87.                         _tcslen(pName)*sizeof(TCHAR) +
  88.                         sizeof(TCHAR);
  89.         break;
  91.     case 2:
  92.         cb = _tcslen(pName) + 1 + _tcslen(m_sMonitorName) + 1 +_tcslen(m_sPortDesc) + 1;
  93.         cb *= sizeof(TCHAR);
  94.         cb += sizeof(PORT_INFO_2);
  95.         break;
  97.     default:
  98.         cb = 0;
  99.         break;
  100.     }
  101.     return cb;
  102. }
  104. LPBYTE CPortList::CopyPortToBuffer(PORT *pPort,DWORD dwLevel,LPBYTE pStart,LPBYTE pEnd)
  105. {
  106.         switch(dwLevel)
  107.         {
  108.         case 1:
  109.                 {
  110.                         PORT_INFO_1 *pPortInfo=(PORT_INFO_1*)pStart;
  111.                         pEnd-=_tcslen(pPort->sPath) * sizeof(TCHAR) + sizeof(TCHAR);
  112.                         pPortInfo->pName=_tcscpy((TCHAR*)pEnd,pPort->sPath);
  113.                         break;
  114.                 }
  115.         case 2:
  116.                 {
  117.                         PORT_INFO_2 *pPortInfo=(PORT_INFO_2*)pStart;
  118.                         pEnd-=_tcslen(m_sMonitorName) * sizeof(TCHAR) + sizeof(TCHAR);
  119.                         pPortInfo->pMonitorName=_tcscpy((TCHAR*)pEnd,m_sMonitorName);
  120.                         pEnd-=_tcslen(m_sPortDesc) * sizeof(TCHAR) + sizeof(TCHAR);
  121.                         pPortInfo->pDescription=_tcscpy((TCHAR*)pEnd,m_sPortDesc);
  122.                         pEnd-=_tcslen(pPort->sPath) * sizeof(TCHAR) + sizeof(TCHAR);
  123.                         pPortInfo->pPortName=_tcscpy((TCHAR*)pEnd,pPort->sPath);
  124.                         pPortInfo->fPortType=0;
  125.                         pPortInfo->Reserved=0;
  126.                         break;
  127.                 }
  128.         default:
  129.                 MessageBox(0,_T("Error"),_T("invalid level"),MB_OK);
  130.         }
  131.     return pEnd;
  132. }
  134. BOOL CPortList::AddPort(TCHAR *img_format, TCHAR *cPath, TCHAR *sPath, DWORD dwStatus, TCHAR* img_format_ext, BOOL dither, TCHAR* extApp)
  135. {
  136.         HKEY hKey;
  137.         bool tmp_bool;
  138.         PORT *pNew=new PORT;
  139.         if(!pNew){
  140.                   return FALSE;
  141.         }
  142.         pNew->pNext=m_pFirst;
  143.         pNew->dwStatus=dwStatus;
  144.         pNew->hFile=0;
  145.         pNew->dwJobID=0;
  146.         pNew->dither = dither;
  147.          _tcscpy(pNew->img_format,img_format);        
  148.          _tcscpy(pNew->img_format_ext,img_format_ext);
  149.          _tcscpy(pNew->extApp,extApp);        
  150.          _tcscpy(pNew->cPath,cPath);
  151.         _tcscpy(pNew->sPath,sPath);
  152.         _tcscpy(pNew->sCurDocument,_T(""));
  153.         _tcscpy(pNew->sPrinter,_T(""));
  154.         m_pFirst=pNew;
  155.         return TRUE;
  156. }
  158. void CPortList::Save(TCHAR *sRoot)
  159. {
  160.         HKEY hKey;
  161.         if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,sRoot,0,KEY_SET_VALUE | KEY_READ,&hKey) != ERROR_SUCCESS)
  162.                         return;
  163.         DWORD z;
  164.         TCHAR sPort[5];
  165.         z=1;
  166.         do{
  167.                 _ultot(z++,sPort,10);
  168.         }while(RegDeleteValue(hKey,sPort)==ERROR_SUCCESS);
  169.         PORT *pPort=m_pFirst;
  170.         z=Count();
  171.         while(pPort->pNext){
  172.                 _ultot(z--,sPort,10);
  173.                 RegSetValueEx(hKey,_T("path"),0,REG_SZ,(BYTE*)pPort->cPath,_tcslen(pPort->cPath)*sizeof(TCHAR));
  174.                 RegSetValueEx(hKey,_T("name"),0,REG_SZ,(BYTE*)_T("ImagePrinter Port"),_tcslen(_T("ImagePrinter Port"))*sizeof(TCHAR));                
  175.                 RegSetValueEx(hKey,_T("format"),0,REG_SZ,(BYTE*)_T("tif"),_tcslen(_T("tif"))*sizeof(TCHAR));                
  176.                 pPort=pPort->pNext;
  177.         }
  178.         RegCloseKey(hKey);
  179. }
  181. #define szLocalMonitor _T("ImagePrinter Port")
  183. void CPortList::Load2(PMONITORINIT pMi)
  184. {
  185.     if( NULL != pMi )
  186.     {
  187. HANDLE phkResult = NULL;
  188. TCHAR sPath[MAX_PATH]=_T("path");
  189. TCHAR sFormat[MAX_PATH]=_T("format");
  190. TCHAR sFormatEx[MAX_PATH]=_T("format_ext");
  191. TCHAR extApp[MAX_PATH] = _T("ext_app");
  192. DWORD dither = 0;
  193. DWORD dwSize;
  194. LONG ret;
  195. if( ERROR_SUCCESS != (ret = pMi->pMonitorReg->fpOpenKey( pMi->hckRegistryRoot, szLocalMonitor, KEY_READ, &phkResult, pMi->hSpooler ) ) )
  196. return;
  198. dwSize=sizeof(dither);
  199. if( ERROR_SUCCESS != (ret = pMi->pMonitorReg->fpQueryValue( phkResult, _T("dither"), 0, (BYTE*)&dither, &dwSize, pMi->hSpooler ) ) )
  200. {
  201. dither = 0;
  202. ret = ERROR_SUCCESS;
  203. }
  204. dwSize=sizeof(extApp);
  205. if( ERROR_SUCCESS != (ret = pMi->pMonitorReg->fpQueryValue( phkResult, extApp, 0, (BYTE*)extApp, &dwSize, pMi->hSpooler) ) )
  206. {
  207. _tcscpy(extApp,_T(""));
  208. ret = ERROR_SUCCESS;
  209. }
  211. dwSize=sizeof(sFormatEx);
  212. if( ERROR_SUCCESS != (ret = pMi->pMonitorReg->fpQueryValue( phkResult, sFormatEx, 0, (BYTE*)sFormatEx, &dwSize, pMi->hSpooler) ) )
  213. {
  214. _tcscpy(sFormatEx,_T(""));
  215. ret = ERROR_SUCCESS;
  216. }
  218. dwSize = sizeof(sPath);
  219. ret = pMi->pMonitorReg->fpQueryValue( phkResult, sPath, 0, (BYTE*)sPath, &dwSize, pMi->hSpooler );
  220. dwSize = sizeof(sFormat);
  221. if( ret == ERROR_SUCCESS )
  222. ret = pMi->pMonitorReg->fpQueryValue( phkResult, sFormat, 0, (BYTE*)sFormat, &dwSize, pMi->hSpooler );
  224. if( ret==ERROR_SUCCESS )
  225. AddPort( sFormat, sPath, _T("ImagePrinter Port"), 0, sFormatEx, (0!=dither), extApp);
  227. pMi->pMonitorReg->fpCloseKey( phkResult, pMi->hSpooler );
  228.     }
  229. }
  231. void CPortList::Load(TCHAR *sRoot)
  232. {
  233. HKEY hKey;
  234. if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,sRoot,0,KEY_QUERY_VALUE,&hKey) != ERROR_SUCCESS){
  235. return;
  236. }
  237. TCHAR sPath[MAX_PATH]=_T("path");
  238. TCHAR sFormat[MAX_PATH]=_T("format");
  239. TCHAR sFormatEx[MAX_PATH]=_T("format_ext");
  240. TCHAR extApp[MAX_PATH] = _T("ext_app");
  241. DWORD dither = 0;
  242. DWORD dwSize;
  243. LONG ret;
  245. dwSize=sizeof(dither);
  246. if( ERROR_SUCCESS != (ret=RegQueryValueEx(hKey,_T("dither"),0,NULL,(BYTE*)&dither,&dwSize) ) )
  247. {
  248. dither = 0;
  249. ret = ERROR_SUCCESS;
  250. }
  252. dwSize=sizeof(extApp);
  253. if( ERROR_SUCCESS != (ret=RegQueryValueEx(hKey,extApp,0,NULL,(BYTE*)extApp,&dwSize) ) )
  254. {
  255. _tcscpy(extApp,_T(""));
  256. ret = ERROR_SUCCESS;
  257. }
  259. dwSize=sizeof(sFormatEx);
  260. if( ERROR_SUCCESS != (ret=RegQueryValueEx(hKey,sFormatEx,0,NULL,(BYTE*)sFormatEx,&dwSize) ) )
  261. {
  262. _tcscpy(sFormatEx,_T(""));
  263. ret = ERROR_SUCCESS;
  264. }
  266. dwSize=sizeof(sPath);
  267. ret=RegQueryValueEx(hKey,sPath,0,NULL,(BYTE*)sPath,&dwSize);
  268. dwSize=sizeof(sFormat);
  269. if(ret==ERROR_SUCCESS)                
  270. ret=RegQueryValueEx(hKey,sFormat,0,NULL,(BYTE*)sFormat,&dwSize);
  273. if(ret==ERROR_SUCCESS){
  274. AddPort( sFormat, sPath, _T("ImagePrinter Port"), 0, sFormatEx, (0!=dither), extApp);
  275. }
  276. RegCloseKey(hKey);
  277. }
  279. PORT *CPortList::FindPort(TCHAR *sPath)
  280. {
  281.         PORT *pPort=m_pFirst;
  282.         while(pPort->pNext){
  283.                 if(_tcsicmp(sPath,pPort->sPath)==0)
  284.                         return pPort;
  285.                 pPort=pPort->pNext;
  286.         }
  287.         return NULL;
  288. }
  290. BOOL CPortList::DeletePort(TCHAR *sPath)
  291. {
  292.         PORT *pPort=m_pFirst;
  293.         PORT *pPrevPort=m_pFirst;
  294.         while(pPort->pNext)        {
  295.                 if(_tcsicmp(sPath,pPort->sPath)==0)                {
  296.                         if(pPort==m_pFirst)                        {
  297.                                 m_pFirst=m_pFirst->pNext;
  298.                                 delete pPort;
  299.                         }
  300.                         else{
  301.                                 pPrevPort->pNext=pPort->pNext;
  302.                                 delete pPort;
  303.                         }
  304.                         return TRUE;
  305.                 }
  306.                 pPrevPort=pPort;
  307.                 pPort=pPort->pNext;
  308.         }
  309.         return FALSE;
  310. }
  312. DWORD CPortList::Count()
  313. {
  314.         DWORD dwCount=0;
  315.         PORT *pPort=m_pFirst;
  316.         while(pPort->pNext){
  317.                 ++dwCount;
  318.                 pPort=pPort->pNext;
  319.         }
  320.         return dwCount;
  321. }