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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 行业应用 > 金融证券系统 > 查看源码
PCTCP.C
资源名称：inet_stock.zip　[点击查看]
上传用户：bjghjy
上传日期：2007-01-07
资源大小：379k
文件大小：8k
源码类别：

金融证券系统

开发平台：

Visual C++

PCTCP.C：源码内容
  2. #include <windows.h>
  3. #include <stdio.h>
  4. #include <string.h>
  5. #include <time.h>
  6. #include <malloc.h>
  7. #include "hq.h"
  8. #include "pctcp.h"
  10. int sdHq =-1, sdJy =-1;
  12. #define WSA_MAKEWORD(x,y)       ((y) * 256 + (x))
  14. BOOL running =FALSE, run_cancelled =FALSE;
  15. int PROTO_USE =PROTO_UDP;
  16. char  * ReadBuf; //[MAX_READ_BUF_SIZE+1];  
  17. BOOL can_read =TRUE, can_write =FALSE;
  18. BOOL fSockError =FALSE;
  19. char HostName[40];
  21. LPSTR GetError(LPSTR err_prefix);
  23. FILE *fp =NULL;
  25. int CalcTime(int jys);
  26. int StringSub(LPSTR lpstr1, LPSTR lpstr2, LPSTR lpres, int s_type);
  27. int CompString(LPSTR, LPSTR);
  28. int     GMinRecalcJgMax(void);
  29. int     GMinRecalcLcMax(void);
  30. int MaxMinChangeHqData(int jys, int maxmin);
  31. void SetStatusText(LPSTR);
  32. int UDPBlockingHook (void);
  34. extern BOOL GetInitString(LPSTR, LPSTR, LPSTR);
  35. extern BOOL PutInitString(LPSTR, LPSTR, LPSTR);
  36. extern BOOL ErrMsg(HWND, LPSTR);
  37. extern HWND ghWndMain, ghWndHq, ghWndStatus, ghWndCj;
  38. extern HINSTANCE ghInstance;
  39. extern HWND ghWndXlt, ghWndJlt, ghWndMaxMin;
  40. extern HWND ghDlgJy;
  42. void WriteMsg(LPSTR msg)
  43. {                
  44. if(fp) fputs(msg, fp);
  45. }
  47. int PCTCPInit(void)
  48. {
  49. char temp[40];
  50. //WORD VersionReqd;        
  51. WSADATA myWSAData;
  52.     int ret;
  53.     
  54. //VersionReqd=WSA_MAKEWORD(MAJOR_VERSION, MINOR_VERSION);
  56. if(!GetInitString("NET", "HOST", HostName))
  57. {
  58. //strcpy(HostName, "172.20.16.68     ");   //汕头 
  59. //strcpy(HostName, "10.17.128.188     ");   //石家庄
  60. //strcpy(HostName, "172.21.112.99    ");   //梅州 
  61. //strcpy(HostName, "10.62.171.68     ");   //长沙
  62. //strcpy(HostName, "10.190.2.103     ");   //银川
  63. strcpy(HostName, "10.29.15.3     ");   //呼市
  64. //strcpy(HostName, "10.157.2.6     ");   //贵阳 
  65. //strcpy(HostName, "10.19.160.30     ");   //邢台
  66. PutInitString("NET", "HOST", HostName);
  67. }
  69. ReadBuf =(char *)_fmalloc(MAX_READ_BUF_SIZE+1);
  71. ret = WSAStartup(0x0101, &myWSAData);
  72.     
  73. sdHq =sdJy =-1;
  74. if (ret != 0)
  75. return -1;
  77. if(GetInitString("NET", "PROTO", temp))
  78. {
  79. if(temp[0] =='U') PROTO_USE =PROTO_UDP;
  80. else PROTO_USE=PROTO_TCP;
  81. }
  82. else 
  83. {
  84. PutInitString("NET", "PROTO", "TCP");      
  85. PROTO_USE=PROTO_TCP;                             
  86. }
  87.                             
  88. return 0;
  89. }
  91. int PCTCPClose(void)
  92. {
  93. int ret;
  95. run_cancelled =TRUE;
  97. //WSACancelBlockingCall();
  98. if(sdHq !=-1)
  99. {
  100. closesocket(sdHq);
  101. sdHq =-1;
  102.     }
  103. if(sdJy !=-1)
  104. {
  105. closesocket(sdJy);
  106. sdJy =-1;
  107.     }
  109. ret = WSACleanup();
  110. if (ret == SOCKET_ERROR && h_errno == WSAEINPROGRESS)
  111. {
  112. //ErrMsg(NULL, "Data transfer in progress.nStop transfer first.");
  113. return -1;
  114. }
  116. if(fp) fclose(fp);
  118. return 0;
  119. }
  121. int PCTCPExit(void)
  122. {
  123. static BOOL fExit =FALSE;
  125. _ffree(ReadBuf);
  126. if(fExit ==TRUE) return 0;
  127. fExit =TRUE;            
  129. PCTCPClose();
  131. if(fp) fclose(fp);
  133. return 0;
  134. }
  136. // used for client
  137. int ConnectHost(int type, int protoUse)
  138. {
  139. LPSTR lpHostName;
  140. struct  hostent FAR *hp;
  141. struct  sockaddr_in server_addr;
  142. SOCKET  hSock;
  143. int iSockType,ret;
  144. //long lret;
  145. time_t time_start, time_end;
  147. lpHostName =&HostName[0];
  149. if(protoUse ==-1) protoUse =PROTO_USE;
  150. memset(&server_addr, 0, sizeof(server_addr));
  151. server_addr.sin_addr.s_addr =/*INADDR_BROADCAST;//ANY;*/(u_long)inet_addr(lpHostName);
  153. if (server_addr.sin_addr.s_addr == (u_long) -1L || 
  154. server_addr.sin_addr.s_addr == 0) 
  155. {
  156. if ((hp = gethostbyname(lpHostName)) == NULL) 
  157. {
  158. //ErrMsg(NULL, "rpcinfo: server address is unknown");
  159. return -1;
  160. }
  161. _fmemcpy((char *)&server_addr.sin_addr, hp->h_addr, hp->h_length);
  162. }
  163. server_addr.sin_family = PF_INET;
  164. if (protoUse == PROTO_TCP)
  165. iSockType = SOCK_STREAM;
  166. else
  167. iSockType = SOCK_DGRAM;
  168.     
  169. if(type !=1)
  170. {
  171. if(sdHq !=-1) closesocket(sdHq);
  173. server_addr.sin_port = htons(PORT_HQ);
  174.     
  175. hSock = socket(PF_INET, iSockType, 0);
  176. if (hSock == INVALID_SOCKET)
  177. return -1;
  178.     
  179. ret =WSAAsyncSelect(hSock, ghWndMain, WM_USER+1, FD_READ);//|FD_CONNECT);
  180. if(ret <0)
  181. {
  182. closesocket(hSock);
  183. return -1;
  184. }
  186. time(&time_start);
  188. retry_hq_connect:
  189. time(&time_end);
  190. if(difftime(time_end, time_start) >10)
  191. {
  192. closesocket(hSock);
  193. return (SOCKET) -1;
  194. }
  195. ret=connect(hSock,(struct sockaddr FAR *)&server_addr,sizeof(server_addr));
  196.     
  197. if (ret == SOCKET_ERROR) {
  198. if (h_errno == WSAEWOULDBLOCK)
  199. goto retry_hq_connect;
  200. }                            
  201. sdHq =hSock;
  202. }
  204. return hSock;
  205. }
  207. int BeginConnect(char *host)
  208. {
  209. LPSTR lpHostName;
  210. struct  hostent FAR *hp;
  211. struct  sockaddr_in server_addr;
  212. SOCKET  hSock;
  213. int iSockType,ret;
  215. lpHostName =host;
  217. memset(&server_addr, 0, sizeof(server_addr));
  218. server_addr.sin_addr.s_addr =(u_long)inet_addr(lpHostName);
  220. if (server_addr.sin_addr.s_addr == (u_long) -1L || 
  221. server_addr.sin_addr.s_addr == 0) 
  222. {
  223. if ((hp = gethostbyname(lpHostName)) == NULL) 
  224. return -1;
  225. _fmemcpy((char *)&server_addr.sin_addr, hp->h_addr, hp->h_length);
  226. }
  227. server_addr.sin_family = PF_INET;
  228. iSockType = SOCK_STREAM;
  229.     
  230. if(sdHq !=-1) closesocket(sdHq);
  232. server_addr.sin_port = htons(PORT_HQ);
  233.     
  234. hSock = socket(PF_INET, iSockType, 0);
  235. if (hSock == INVALID_SOCKET)
  236. return -1;
  237.     
  238. ret =WSAAsyncSelect(hSock, ghWndMain, WM_USER+1, FD_READ|FD_CONNECT);
  239. if(ret <0)
  240. {
  241. closesocket(hSock);
  242. return -1;
  243. }
  246. ret=connect(hSock,(struct sockaddr FAR *)&server_addr,sizeof(server_addr));
  247. if (ret == SOCKET_ERROR) 
  248. {
  249. if (h_errno != WSAEWOULDBLOCK)
  250. return -1;
  251. }                            
  252. sdHq =hSock;
  253. return hSock;
  254. }
  257. int ConnectToJyHost(char *HostName)
  258. {
  259. LPSTR lpHostName;
  260. struct  hostent FAR *hp;
  261. struct  sockaddr_in server_addr;
  262.     int hSock,iSockType,ret;
  264. if(sdJy !=-1)
  265. {
  266. closesocket(sdJy);
  267. sdJy =-1;
  268. }
  269. lpHostName =&HostName[0];
  271. memset(&server_addr, 0, sizeof(server_addr));
  272. server_addr.sin_addr.s_addr =(u_long)inet_addr(lpHostName);
  274. if (server_addr.sin_addr.s_addr == (u_long) -1L || 
  275. server_addr.sin_addr.s_addr == 0) 
  276. {
  277. if ((hp = gethostbyname(lpHostName)) == NULL) 
  278. {
  279. ErrMsg(NULL, "rpcinfo: server address is unknown");
  280. return -1;
  281. }
  282. _fmemcpy((char *)&server_addr.sin_addr, hp->h_addr, hp->h_length);
  283. }
  284. server_addr.sin_family = PF_INET;
  285. iSockType = SOCK_STREAM;
  286.     
  287. server_addr.sin_port = htons((u_int) PORT_JY);
  288.     
  289. hSock = socket(PF_INET, iSockType, 0);
  290. if (hSock == INVALID_SOCKET)
  291. {
  292. ErrMsg(NULL, "socket() failed");
  293. return (SOCKET) -1;
  294. }
  296. ret =WSAAsyncSelect(hSock, ghWndMain, WM_USER+2, FD_READ|FD_CONNECT);
  297. if(ret <0)
  298. {
  299. closesocket(hSock);
  300. return -1;
  301. }
  302.     ret=connect(hSock,(struct sockaddr FAR *)&server_addr,sizeof(server_addr));
  303. if (ret == SOCKET_ERROR)
  304. {
  305. ret =WSAGetLastError();
  306. if(ret ==WSAEWOULDBLOCK)
  307. ret =0;
  308. else
  309. {
  310. closesocket(hSock);
  311. return ret;
  312. }
  313. }
  315. sdJy =hSock;
  317. return ret;
  318. }
  320. int PCTCPStop(void)
  321. {
  322. run_cancelled =TRUE;
  323. can_read =FALSE;
  325. return 0;
  326. }
  328. void PCTCPPause(BOOL fPause)
  329. {
  330. can_read =!fPause;
  331. }                  
  333. int UDPBlockingHook (void)
  334. {
  335. MSG msg;
  337. //run_cancelled =FALSE;
  338. if(run_cancelled ==TRUE) return FALSE;
  339. if (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
  340. {
  341. TranslateMessage(&msg);
  342.  DispatchMessage(&msg);   
  343. if(msg.message ==WM_QUIT)
  344. {
  345.        PostQuitMessage(0);
  346.        run_cancelled =TRUE;               
  347.        WSASetLastError(WSAEINTR);
  348.        return FALSE;
  349. }
  350. //WriteMsg("messagen");
  351. return TRUE;
  352. }
  353. return FALSE;
  354. }
  356. int PASCAL FAR WSAsperror(int errorcode, char far * buf, int len)
  357. {
  358. int err_len;
  359.     
  360. if (errorcode == 0)
  361. errorcode = WSABASEERR;
  362. if (errorcode < WSABASEERR)
  363. return 0;
  364.     
  365. err_len = LoadString(ghInstance,errorcode,buf,len);
  367. return err_len;
  369. }
  371. LPSTR GetError(LPSTR err_prefix)
  372. {
  373. int wsa_err;
  374. char errbuf[1000];
  375. char prbuf[1000];
  377. wsa_err =WSAGetLastError(); 
  379. WSAsperror(wsa_err, (LPSTR)errbuf, sizeof(errbuf));
  380.     
  381. wsprintf((LPSTR)prbuf, "%s:n%s", (LPSTR) err_prefix, (LPSTR)errbuf);
  383. return &prbuf[0];  
  384. }
  386. int check_socket_status(int sd)
  387. {
  388. struct timeval wait;
  389. fd_set send_ready,except;
  391. wait.tv_sec =wait.tv_usec=0;
  392. FD_ZERO(&send_ready);
  393. FD_ZERO(&except);
  394. FD_SET(sd,&send_ready);
  395. FD_SET(sd,&except);
  396. if(select(FD_SETSIZE,(fd_set *)0,&send_ready,
  397. &except,&wait)<0)
  398. return(-1);
  399. if(FD_ISSET(sd,&send_ready))
  400. return(0);
  401. else if(FD_ISSET(sd,&except))
  402. return(-2);
  403. else
  404. return(-3);
  405. }
  407. void SetStatusText(LPSTR tmp)
  408. {                 
  409. HDC hDC;
  411. hDC =GetDC(NULL);
  412. SetBkColor(hDC, RGB(255, 0, 255));
  413. SetBkMode(hDC, TRANSPARENT);
  414. TextOut(hDC, 10, 2, tmp, strlen(tmp));
  415. ReleaseDC(NULL, hDC);
  416. }