STREAM *ConnectManagerAddClient(char *Host, int Port, int Flags, char *Name, CONNECT_FUNC OnConnect, ONDATA_FUNC OnData)
{
STREAM *S;
TConnectManagerItem *Item;
if (! ConnectManClients) ConnectManClients=ListCreate();
S=STREAMCreate();
if (! STREAMConnectToHost(S,Host,Port,Flags))
{
STREAMClose(S);
return(NULL);
}
Item=(TConnectManagerItem *) calloc(1,sizeof(TConnectManagerItem));
Item->OnConnect=OnConnect;
Item->OnData=OnData;
Item->Data=(void *) S;
Item->Name=CopyStr(Item->Name,Name);
Item->Host=CopyStr(Item->Host,Host);
Item->Port=Port;
if (Item->OnConnect && STREAMIsConnected(S)) Item->OnConnect(Item);
ListAddItem(ConnectManClients,Item);
return(S);
}
STREAM *HTTPSetupConnection(HTTPInfoStruct *Info, int ForceHTTPS)
{
char *Proto=NULL, *Host=NULL, *Token=NULL;
int Port=0, Flags=0;
STREAM *S;
S=STREAMCreate();
if (Info->Flags & HTTP_PROXY)
{
if (! Info->ProxyAuthorization)
{
Info->ProxyAuthorization=(HTTPAuthStruct *) calloc(1,sizeof(HTTPAuthStruct));
}
ParseURL(Info->Proxy, &Proto, &Host, &Token, &Info->ProxyAuthorization->Logon, &Info->ProxyAuthorization->Password,NULL,NULL);
Port=atoi(Token);
if (ForceHTTPS) Proto=CopyStr(Proto,"https");
if (strcasecmp(Proto,"https")==0) Flags |= CONNECT_SSL;
}
else
{
Host=CopyStr(Host,Info->Host);
Port=Info->Port;
if (Info->Flags & HTTP_SSL) Flags |= CONNECT_SSL;
if (ForceHTTPS)
{
Flags |= CONNECT_SSL;
}
if (Port==0)
{
if (Flags & CONNECT_SSL) Port=443;
else Port=80;
}
}
if (Info->Flags & HTTP_TUNNEL) STREAMAddConnectionHop(S,Info->Proxy);
if (STREAMConnectToHost(S,Host,Port,Flags))
{
HTTPSendHeaders(S,Info);
}
else
{
STREAMClose(S);
S=NULL;
}
Info->S=S;
DestroyString(Token);
DestroyString(Proto);
DestroyString(Host);
return(S);
}
void ConnectManagerMainLoop()
{
TConnectManagerItem *Item, *NewItem;
ListNode *Curr, *Prev;
int highfd=0;
fd_set ReadSet, WriteSet;
int sock, result, SelectResult, NextTimerFire;
STREAM *S;
time_t Now;
struct timeval tv;
int MoreData=FALSE;
while (1)
{
MoreData=FALSE;
time(&Now);
NextTimerFire=60;
Curr=ListGetNext(Timers);
while (Curr)
{
Item=(TConnectManagerItem *) Curr->Item;
if (Item->LastTimerFire==0) Item->LastTimerFire=Now;
result=(Item->LastTimerFire + Item->TimerVal) - Now;
if (result < NextTimerFire) NextTimerFire=result;
Curr=ListGetNext(Curr);
}
FD_ZERO(&ReadSet);
FD_ZERO(&WriteSet);
Curr=ListGetNext(ConnectManServers);
while (Curr)
{
Item=(TConnectManagerItem *) Curr->Item;
S=(STREAM *) Item->Data;
FD_SET(S->in_fd,&ReadSet);
if (S->in_fd > highfd) highfd=S->in_fd;
Curr=ListGetNext(Curr);
}
Curr=ListGetNext(ConnectManClients);
while (Curr)
{
Item=(TConnectManagerItem *) Curr->Item;
S=(STREAM *) Item->Data;
if (! S)
{
ListDeleteNode(Curr);
continue;
}
if (S->State & SS_CONNECTING)
{
FD_SET(S->in_fd,&WriteSet);
}
if (S->InEnd > S->InStart) MoreData=TRUE;
else
{
//always add to read set
FD_SET(S->in_fd,&ReadSet);
if (S->in_fd > highfd) highfd=S->in_fd;
}
Curr=ListGetNext(Curr);
}
if (MoreData)
{
tv.tv_usec = 10;
tv.tv_sec = 0;
SelectResult=0;
}
else
{
//SELECT!!!
tv.tv_usec = 20000;
tv.tv_sec = NextTimerFire;
}
SelectResult=select(highfd+1,&ReadSet,&WriteSet,NULL,&tv);
if (SelectResult > 0)
{
Curr=ListGetNext(ConnectManServers);
while (Curr)
{
Item=(TConnectManagerItem *) Curr->Item;
S=(STREAM *) Item->Data;
if (! S)
{
ListDeleteNode(Curr);
continue;
}
if (FD_ISSET(S->in_fd,&ReadSet))
{
sock=TCPServerSockAccept(S->in_fd,&S->Path);
if (sock > -1)
{
S=STREAMFromFD(sock);
STREAMSetFlushType(S,FLUSH_LINE,0,0);
NewItem=ConnectManagerAddIncoming(S,Item->Name, Item->OnData);
if (Item->OnConnect) Item->OnConnect(NewItem);
}
}
Curr=ListGetNext(Curr);
}
}
Curr=ListGetNext(ConnectManClients);
while (Curr)
{
Item=(TConnectManagerItem *) Curr->Item;
S=(STREAM *) Item->Data;
if (! S)
{
ListDeleteNode(Curr);
continue;
}
if ((SelectResult > 0) && FD_ISSET(S->in_fd,&WriteSet))
{
if (S->State & SS_CONNECTING)
{
if (STREAMIsConnected(S))
{
STREAMSetFlags(S, 0, SF_NONBLOCK);
if (Item->OnConnect) Item->OnConnect(Item);
}
}
}
if (
(S->InEnd > S->InStart) ||
((SelectResult > 0) && (FD_ISSET(S->in_fd,&ReadSet)))
)
{
if (! (S->State & SS_CONNECTING))
{
if (Item->OnData)
{
result=Item->OnData(S, Item->Name);
if (! result)
{
STREAMClose(S);
Prev=ListGetPrev(Curr);
ListDeleteNode(Curr);
free(Item);
Curr=Prev;
}
else if (result==RECONNECT)
{
STREAMClose(S);
S=STREAMCreate();
STREAMConnectToHost(S,Item->Host,Item->Port,CONNECT_NONBLOCK);
Item->Data=(void *) S;
}
}
}
}
Curr=ListGetNext(Curr);
}
time(&Now);
Curr=ListGetNext(Timers);
while (Curr)
{
Item=(TConnectManagerItem *) Curr->Item;
if ( (Now - Item->LastTimerFire) >= Item->TimerVal )
{
if (Item->OnData) ((ONTIMER_FUNC)Item->OnData)(Item->Data,Item->Name);
Item->LastTimerFire=Now;
}
Curr=ListGetNext(Curr);
}
}
}
