新人满分 sock 连接问题求助。。。。
流媒体接收不成功 求解 谢谢
#include "stdafx.h"sock?求助
#include "afxsock.h"
#include "MyProxy.h"
#include "MemPool.h"
#define MAXBUFFERSIZE 20*1024 //缓冲区大小
#define HTTP "http://"
//Proxy 端口
UINT pport = 6000;
UINT ProxyToServer(LPVOID pParam);
UINT UserToProxy (LPVOID pParam);
//extern void Log(CString info);
struct ProxySockets
{
SOCKET ProxyToUserSocket;//本地机器到PROXY 服务机的socket
SOCKET ProxyToServSocket;//PROXY 服务机到远程主机的socket
BOOL IsProxyToUserClosed; // 本地机器到PROXY 服务机状态
BOOL IsProxyToServClosed; // PROXY 服务机到远程主机状态
};
struct ProxyParam
{
char Address[256];// 远程主机地址
HANDLE IsConnectedOK;// PROXY 服务机到远程主机的联结状态
ProxySockets * pPair;// 维护一组SOCKET的指针
int Port;// 用来联结远程主机的端口
}; //结构用来PROXY SERVER与远程主机的信息交换
SOCKET listentsocket; //用来侦听的SOCKET
int StartProxyServer() //启动服务
{
WSADATA wsaData;
sockaddr_in local;
//SOCKET listentsocket;
Log("启动代理服务器");
if(::WSAStartup(0x202,&wsaData)!=0)
{
Log("\nError in Startup session.\n");
WSACleanup();
return -1;
}
local.sin_family = AF_INET;
local.sin_addr.s_addr = INADDR_ANY;
local.sin_port = htons(pport);
//
listentsocket = socket(AF_INET,SOCK_STREAM,0);
Log("创建侦听套接字");
if(listentsocket == INVALID_SOCKET)
{
Log("\nError in New a Socket.");
WSACleanup();
return -2;
}
Log("绑定侦听套接字");
if(::bind(listentsocket,(sockaddr *)&local,sizeof(local))!=0)
{
Log("\n Error in Binding socket.");
WSACleanup();
return -3;
}
Log("建立侦听");
if(::listen(listentsocket,5)!=0)
{
Log("\n Error in Listen.");
WSACleanup();
return -4;
}
//::listentsocket=listentsocket;
Log("启动侦听线程");
AfxBeginThread(UserToProxy,NULL); //启动侦听
return 1;
}
int CloseServer() //关闭服务
{
closesocket(listentsocket);
//WSACleanup();
return 1;
}
//分析接收到的字符，得到远程主机地址
int ResolveInformation( char * str, char *address, int * port)
{
//char buff[MAXBUFFERSIZE], command[512], proto[128], *p;
char *buf = (char*)MallocMem(MAXBUFFERSIZE);
char *command = (char*)MallocMem(512);
char *proto = (char*)MallocMem(128);
char *p = NULL;
CString strLog;
sscanf(str,"%s%s%s",command,buf,proto);
strLog.Format("COMMAND : %s",command);
Log(strLog);
strLog.Format("BUFFER : %s",buf);
Log(strLog);
strLog.Format("PROTOCL : %s",proto);
Log(strLog);
p=strstr(buf,HTTP);
//HTTP
if(p)
{
p+=strlen(HTTP);
UINT i;
for(i=0; i<strlen(p); i++)
{
if( *(p+i)=='/') break;
}
*(p+i)=0;
strcpy(address,p);
p = strstr(str,HTTP);
//去掉远程主机名:
//GET http://www.njust.edu.cn/ HTTP1.1 == > GET / HTTP1.1
for(UINT j=0;j< i+strlen(HTTP);j++)
{
*(p+j)=' ';
}
*port=80;//缺省的 http 端口
strLog.Format("ADDR : %s",address);
Log(strLog);
strLog.Format("PORT : %d",*port);
Log(strLog);
return 1;
}
return 0;
}
// 取到本地的数据，发往远程主机
UINT UserToProxy(void *pParam)
{
sockaddr_in from;
int fromlen = sizeof(from);
char Buffer[MAXBUFFERSIZE];
int Len;
SOCKET ProxyToUserSocket;
ProxySockets SPair;
ProxyParam ProxyP;
CWinThread * pChildThread;
int retval;
Log("正在侦听用户连接");
ProxyToUserSocket = accept(listentsocket,(struct sockaddr*)&from,&fromlen);
Log("接受连接");
Log("启动新的侦听线程");
AfxBeginThread(UserToProxy,pParam); //启动另一侦听.
if( ProxyToUserSocket==INVALID_SOCKET)
{
Log( "\nError in accept ");
return -5;
}
//读客户的第一行数据
SPair.IsProxyToUserClosed = FALSE;
SPair.IsProxyToServClosed = TRUE ;
SPair.ProxyToUserSocket = ProxyToUserSocket;
Log("接收用户数据");
retval = recv(SPair.ProxyToUserSocket,Buffer,sizeof(Buffer),0);
if(retval==SOCKET_ERROR)
{
Log("\nError Recv");
if(SPair.IsProxyToUserClosed == FALSE)
{
closesocket(SPair.ProxyToUserSocket);
SPair.IsProxyToUserClosed = TRUE;
return 0;
}
}
if(retval==0)
{
Log("Client Close connection\n");
if(SPair.IsProxyToUserClosed==FALSE)
{
closesocket(SPair.ProxyToUserSocket);
SPair.IsProxyToUserClosed=TRUE;
return 0;
}
}
Len = retval;
//
SPair.IsProxyToUserClosed = FALSE;
SPair.IsProxyToServClosed = TRUE;
SPair.ProxyToUserSocket = ProxyToUserSocket;
ProxyP.pPair=&SPair;
ProxyP.IsConnectedOK = CreateEvent(NULL,TRUE,FALSE,NULL);
Log("分析用户请求");
ResolveInformation( Buffer,ProxyP.Address,&ProxyP.Port);
Log("创建连接目标地址线程");
pChildThread = AfxBeginThread(ProxyToServer,(LPVOID)&ProxyP);
Log("等待连接目标地址事件");
::WaitForSingleObject(ProxyP.IsConnectedOK,60000);
::CloseHandle(ProxyP.IsConnectedOK);
while(SPair.IsProxyToServClosed == FALSE && SPair.IsProxyToUserClosed == FALSE)
{
Log("向目标地址发送数据");
retval = send(SPair.ProxyToServSocket,Buffer,Len,0);
if(retval==SOCKET_ERROR)
{
if(SPair.IsProxyToServClosed == FALSE)
{
closesocket(SPair.ProxyToServSocket);
SPair.IsProxyToServClosed = TRUE;
}
return 0;
}
Log("从用户地址接收数据");
Sleep(50);
retval = recv(SPair.ProxyToUserSocket,Buffer,sizeof(Buffer),0);
if(retval==SOCKET_ERROR)
{
Log("\nError Recv");
if(SPair.IsProxyToUserClosed==FALSE)
{
closesocket(SPair.ProxyToUserSocket);
SPair.IsProxyToUserClosed=TRUE;
}
continue;
}
if(retval==0)
{
Log("Client Close connection\n");
if(SPair.IsProxyToUserClosed==FALSE)
{
closesocket(SPair.ProxyToUserSocket);
SPair.IsProxyToUserClosed=TRUE;
}
break;
}
Len=retval;
}//End While
if(SPair.IsProxyToServClosed == FALSE)
{
closesocket(SPair.ProxyToServSocket);
SPair.IsProxyToServClosed=TRUE;
}
if(SPair.IsProxyToUserClosed == FALSE)
{
closesocket(SPair.ProxyToUserSocket);
SPair.IsProxyToUserClosed=TRUE;
}
::WaitForSingleObject(pChildThread->m_hThread,20000);
return 0;
}
// 读取远程主机数据，并发往本地客户机
UINT ProxyToServer(LPVOID pParam)
{
ProxyParam * pPar=(ProxyParam*) pParam;
char Buffer[MAXBUFFERSIZE];
char * server_name;
unsigned int addr;
int socket_type ;
unsigned short port ;
int retval,Len;
struct sockaddr_in server;
struct hostent *hp;
SOCKET conn_socket;
CString strLog;
server_name = pPar->Address;
socket_type = SOCK_STREAM;
port = pPar->Port;
if (isalpha(server_name[0]))
{
/* server address is a name */
hp = gethostbyname(server_name);
Log("GetHostByName");
}
else
{
/* Convert nnn.nnn address to a usable one */
addr = inet_addr(server_name);
hp = gethostbyaddr((char *)&addr,4,AF_INET);
Log("GetHostByName");
}
if (hp == NULL )
{
strLog.Format("Client: Cannot resolve address [%s]: Error %d\n",server_name,WSAGetLastError());
Log(strLog);
::SetEvent(pPar->IsConnectedOK);
return 0;
}
//
// Copy the resolved information into the sockaddr_in structure
//
memset(&server,0,sizeof(server));
memcpy(&(server.sin_addr),hp->h_addr,hp->h_length);
server.sin_family = hp->h_addrtype;
server.sin_port = htons(port);
conn_socket = socket(AF_INET,socket_type,0);/* 打开一个 socket */
if (conn_socket < 0 )
{
fprintf(stderr,"Client: Error Opening socket: Error %d\n",WSAGetLastError());
strLog.Format("Client: Error Opening socket: Error %d\n",WSAGetLastError());
Log(strLog);
pPar->pPair->IsProxyToServClosed=TRUE;
::SetEvent(pPar->IsConnectedOK);
return -1;
}
Log("连接服务器. . .");
if (connect(conn_socket,(struct sockaddr*)&server,sizeof(server))== SOCKET_ERROR)
{
strLog.Format("connect() failed: %d\n",WSAGetLastError());
pPar->pPair->IsProxyToServClosed=TRUE;
::SetEvent(pPar->IsConnectedOK);
return -1;
}
Log("连接服务器成功");
pPar->pPair->ProxyToServSocket = conn_socket;
pPar->pPair->IsProxyToServClosed = FALSE;
::SetEvent(pPar->IsConnectedOK);
//
while(!pPar->pPair->IsProxyToServClosed && !pPar->pPair->IsProxyToUserClosed)
{
Sleep(50);
Log("准备接收目标地址的数据");
retval = recv(conn_socket,Buffer,sizeof (Buffer),0 );
Log("接收目标地址的数据");
if (retval == SOCKET_ERROR )
{
fprintf(stderr,"recv() failed: error %d\n",WSAGetLastError());
closesocket(conn_socket);
pPar->pPair->IsProxyToServClosed=TRUE;
break;
}
Len=retval;
if (retval == 0)
{
Log("Server closed connection\n");
closesocket(conn_socket);
pPar->pPair->IsProxyToServClosed=TRUE;
break;
}
Log("准备向用户地址发送数据");
retval = send(pPar->pPair->ProxyToUserSocket,Buffer,Len,0);
if (retval == SOCKET_ERROR)
{
fprintf(stderr,"send() failed: error %d\n",WSAGetLastError());
closesocket(pPar->pPair->ProxyToUserSocket);
pPar->pPair->IsProxyToUserClosed=TRUE;
break;
}
Log("向用户地址发送数据");
}
if(pPar->pPair->IsProxyToServClosed==FALSE)
{
closesocket(pPar->pPair->ProxyToServSocket);
pPar->pPair->IsProxyToServClosed=TRUE;
}
if(pPar->pPair->IsProxyToUserClosed==FALSE)
{
closesocket(pPar->pPair->ProxyToUserSocket);
pPar->pPair->IsProxyToUserClosed=TRUE;
}
return 1;
}
