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);
}
int MemcachedSet(const char *Key, int TTL, const char *Value)
{
char *Tempstr=NULL;
int result=FALSE;
if (STREAMIsConnected(MCS))
{
Tempstr=FormatStr(Tempstr,"set %s 0 %d %d\r\n%s\r\n",Key,TTL,StrLen(Value),Value);
STREAMWriteLine(Tempstr,MCS);
Tempstr=STREAMReadLine(Tempstr,MCS);
StripTrailingWhitespace(Tempstr);
if (StrLen(Tempstr) && (strcmp(Tempstr,"STORED")==0)) result=TRUE;
}
DestroyString(Tempstr);
return(result);
}
char *MemcachedGet(char *RetStr, const char *Key)
{
char *Tempstr=NULL;
if (STREAMIsConnected(MCS))
{
RetStr=CopyStr(RetStr,"");
Tempstr=FormatStr(Tempstr,"get %s\r\n",Key);
STREAMWriteLine(Tempstr,MCS);
Tempstr=STREAMReadLine(Tempstr,MCS);
while (Tempstr)
{
StripTrailingWhitespace(Tempstr);
if (strcmp(Tempstr,"END")==0) break;
if (strncmp(Tempstr,"VALUE ",6) !=0) RetStr=CopyStr(RetStr,Tempstr);
Tempstr=STREAMReadLine(Tempstr,MCS);
}
}
DestroyString(Tempstr);
return(RetStr);
}
STREAM *HTTPTransact(HTTPInfoStruct *Info)
{
int result=HTTP_NOCONNECT;
while (1)
{
if (! Info->S) Info->S=HTTPConnect(Info);
else if (! (Info->State & HTTP_HEADERS_SENT)) HTTPSendHeaders(Info->S,Info);
if (Info->S && STREAMIsConnected(Info->S))
{
Info->ResponseCode=CopyStr(Info->ResponseCode,"");
if (! (Info->State & HTTP_CLIENTDATA_SENT))
{
//Set this even if no client data to send, so we no we've been
//through here once
Info->State |= HTTP_CLIENTDATA_SENT;
if (StrLen(Info->PostData))
{
STREAMWriteLine(Info->PostData,Info->S);
if (Info->Flags & HTTP_DEBUG) fprintf(stderr,"\n%s\n",Info->PostData);
}
else
{
if (strcasecmp(Info->Method,"POST")==0) break;
if (strcasecmp(Info->Method,"PUT")==0) break;
}
}
//Must clear this once the headers and clientdata sent
Info->State=0;
HTTPReadHeaders(Info->S,Info);
result=HTTPProcessResponse(Info);
STREAMSetValue(Info->S,"HTTP:URL",Info->Doc);
if (Info->Flags & HTTP_CHUNKED) HTTPAddChunkedProcessor(Info->S);
if (Info->Flags & HTTP_GZIP)
{
STREAMAddStandardDataProcessor(Info->S,"compression","gzip","");
}
else if (Info->Flags & HTTP_DEFLATE) STREAMAddStandardDataProcessor(Info->S,"compression","zlib","");
if (result == HTTP_OKAY) break;
if (result == HTTP_NOTFOUND) break;
if (result == HTTP_NOTMODIFIED) break;
if (result == HTTP_ERROR) break;
if (result == HTTP_CIRCULAR_REDIRECTS) break;
if (result == HTTP_AUTH_BASIC)
{
if (
(! Info->Authorization) ||
(
(Info->Authorization->Flags & HTTP_SENT_AUTH) ||
(! Info->Authorization->Logon) ||
(StrLen(Info->Authorization->Logon)==0)
)
)
{
if (result == HTTP_AUTH_BASIC) break;
if (result == HTTP_AUTH_DIGEST) break;
}
}
if (
(result == HTTP_PROXY_AUTH) &&
(
(Info->ProxyAuthorization->Flags & HTTP_SENT_AUTH) ||
(! Info->ProxyAuthorization->Logon) ||
(StrLen(Info->ProxyAuthorization->Logon)==0)
)
)
{
break;
}
STREAMClose(Info->S);
Info->S=NULL;
}
else break;
}
return(Info->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);
}
}
}
