static int appendMemBufList( int top , char *data , int len )
{
int fr = getFreeMem( );
int rest = len;
int data_topaddr = 0;
if( len >= fr ){
FILE *fp;
if( (fp=fopen( "badsysinfo.txt", "a+")) != NULL ){
fprintf( fp, "appendMemBufList() len:%d / fr:%d err !! \n", len, fr);
fclose( fp);
}
//andy_log
log( "appendMemBufList() len:%d / fr:%d err !! \n", len, fr);
return -1;
}
data[len] = 0;
for(;;){
int blanksize = sizeof( mb[0].buf ) - mb[top].len;
int cpsize = ( rest <= blanksize ) ? rest : blanksize;
memcpy( mb[top].buf + mb[top].len ,
data + data_topaddr , cpsize );
mb[top].len += cpsize;
if( rest <= blanksize ){
return len;
} else {
int newmb;
rest -= cpsize;
data_topaddr += cpsize;
if( (newmb = findregBlankMemBuf( ) ) == TCPSTRUCT_EMBFULL ){
FILE *fp;
if( (fp=fopen( "badsysinfo.txt", "a+")) != NULL ){
fprintf( fp, "find newmb == TCPSTRUCT_EMBFULL err data:%s !!\n", data);
fclose( fp);
}
log( "find newmb == TCPSTRUCT_EMBFULL err data:%s !!\n", data);
}
mb[top].next = newmb;
top = mb[top].next;
}
}
return TCPSTRUCT_EBUG;
}
int tcpstruct_accept( int *tis , int ticount )
{
int i;
int sret;
int accepted = 0;
struct timeval t;
struct sockaddr_in sin;
fd_set rfds, wfds , efds;
FD_ZERO( & rfds );
FD_ZERO( & wfds );
FD_ZERO( & efds );
for(i=0;i<MAXCONNECTION;i++){
if( con[i].use &&
con[i].fd >= 0 && con[i].closed_by_remote ==0 ){
FD_SET( con[i].fd , & rfds );
FD_SET( con[i].fd , & wfds );
FD_SET( con[i].fd , & efds );
}
}
t = select_timeout;
sret = select( 1024, & rfds , (fd_set*)NULL, & efds , &t);
if( sret > 0 ) {
for(i=0;i< MAXCONNECTION;i++){
if( ( con[i].fd >= 0 ) && FD_ISSET( con[i].fd , &rfds ) ){
int fr = getFreeMem( );
int rr , readsize ;
if( fr <= 0 ) continue;
if( fr > sizeof(tmpbuf ) ){
readsize = sizeof( tmpbuf);
} else {
readsize = fr;
}
rr = read( con[i].fd , tmpbuf , readsize );
if( rr <= 0 ){
con[i].closed_by_remote = 1;
} else {
appendReadBuffer( i , tmpbuf , rr );
}
}
}
}
/* write */
t = select_timeout;
sret = select( 1024, (fd_set*)NULL, &wfds, & efds , &t);
if( sret > 0 ) {
for(i=0;i<MAXCONNECTION;i++){
if( ( con[i].fd >= 0 ) && FD_ISSET( con[i].fd , &wfds )){
char send_buf[4096];
int l , rr;
memset( send_buf, 0, sizeof( send_buf));
l = consumeMemBufList( con[i].mbtop_wi ,send_buf, sizeof(send_buf),0 , 1 );
rr = write( con[i].fd , send_buf , l );
if( rr < 0 ){
con[i].closed_by_remote = 1;
} else {
consumeMemBufList( con[i].mbtop_wi , send_buf, l, 1 , 0 );
}
}
}
}
for( i=0; i<ticount; i++){
int asret;
struct timeval t;
t.tv_sec =0;
t.tv_usec =0;
FD_ZERO( & rfds );
FD_ZERO( & wfds );
FD_ZERO( & efds );
FD_SET( mainsockfd , & rfds );
FD_SET( mainsockfd , & wfds );
FD_SET( mainsockfd , & efds );
asret = select( 1024, &rfds , &wfds , &efds, &t );
// Nuke 20040610: add asret>0 to avoid signal interrupt in select
if( (asret>0) && FD_ISSET( mainsockfd , & rfds )){
struct sockaddr_in c;
int len , newsockfd;
int newcon;
bzero( &c , sizeof( c ));
len = sizeof( c );
fprintf( stderr, "i can accept " );
newcon = findregBlankCon( );
if( newcon < 0 ) continue;
newsockfd = accept( mainsockfd, (struct sockaddr*)&c , &len );
log( "ͬÒâ: %d\n" , newsockfd );
if( newsockfd < 0 ){
unregMemBuf( newcon );
continue;
}
char ipbuf[64];
char ipmsg[2048];
FILE *ipfp;
ipfp = fopen( "ip.txt" , "r" );
if( ipfp != NULL ){
//log("111111111\n");
while(fgets(ipbuf, sizeof(ipbuf), ipfp)){
sprintf(ipmsg,"%s%s",ipmsg,ipbuf);
}
fclose(ipfp);
unsigned long sinip;
memcpy( &sinip, &c.sin_addr, 4);
int ipa,ipb,ipc,ipd;
char ip[32];
ipa=(sinip % 0x100); sinip=sinip / 0x100;
ipb=(sinip % 0x100); sinip=sinip / 0x100;
ipc=(sinip % 0x100); sinip=sinip / 0x100;
ipd=(sinip % 0x100);
sprintf(ip,".%d.%d.%d.%d.",ipa,ipb,ipc,ipd);
//log("ip=%s,ip1=%s\n",ip,ipmsg);
if(strstr(ipmsg,ip)==NULL){
close(newsockfd);
unregMemBuf( newcon );
continue;
}
}
set_nodelay( newsockfd );
con[newcon].fd = newsockfd;
memcpy( &con[newcon].remoteaddr , &c ,sizeof(c));
tis[accepted] = newcon;
accepted ++;
}
}
return accepted;
}
int tcpstruct_accept( int *tis , int ticount )
{
int i;
int sret;
int accepted = 0;
struct timeval t;
fd_set rfds, wfds , efds;
FD_ZERO( & rfds );
FD_ZERO( & wfds );
FD_ZERO( & efds );
for(i=0;i<MAXCONNECTION;i++){
if( con[i].use &&
con[i].fd >= 0 && con[i].closed_by_remote ==0 ){
FD_SET( con[i].fd , & rfds );
FD_SET( con[i].fd , & wfds );
FD_SET( con[i].fd , & efds );
}
}
t = select_timeout;
sret = select( 1024, & rfds , (fd_set*)NULL, & efds , &t);
if( sret > 0 ) {
for(i=0;i< MAXCONNECTION;i++){
if( ( con[i].fd >= 0 ) && FD_ISSET( con[i].fd , &rfds ) ){
int fr = getFreeMem( );
int rr , readsize ;
if( fr <= 0 ) continue;
if( fr > sizeof(tmpbuf ) ){
readsize = sizeof( tmpbuf);
} else {
readsize = fr;
}
rr = read( con[i].fd , tmpbuf , readsize );
if( rr <= 0 ){
con[i].closed_by_remote = 1;
} else {
appendReadBuffer( i , tmpbuf , rr );
}
}
}
}
/* write */
t = select_timeout;
sret = select( 1024, (fd_set*)NULL, &wfds, & efds , &t);
if( sret > 0 ) {
for(i=0;i<MAXCONNECTION;i++){
if( ( con[i].fd >= 0 ) && FD_ISSET( con[i].fd , &wfds )){
char send_buf[4096];
int l , rr;
memset( send_buf, 0, sizeof( send_buf));
l = consumeMemBufList( con[i].mbtop_wi ,send_buf, sizeof(send_buf),0 , 1 );
rr = write( con[i].fd , send_buf , l );
if( rr < 0 ){
con[i].closed_by_remote = 1;
} else {
consumeMemBufList( con[i].mbtop_wi , send_buf, l, 1 , 0 );
}
}
}
}
for( i=0; i<ticount; i++){
int asret;
struct timeval t;
t.tv_sec =0;
t.tv_usec =0;
FD_ZERO( & rfds );
FD_ZERO( & wfds );
FD_ZERO( & efds );
FD_SET( mainsockfd , & rfds );
FD_SET( mainsockfd , & wfds );
FD_SET( mainsockfd , & efds );
asret = select( 1024, &rfds , &wfds , &efds, &t );
// Nuke 20040610: add asret>0 to avoid signal interrupt in select
if( (asret>0) && FD_ISSET( mainsockfd , & rfds )){
struct sockaddr_in c;
int len , newsockfd;
int newcon;
bzero( &c , sizeof( c ));
len = sizeof( c );
fprintf( stderr, "i can accept " );
newcon = findregBlankCon( );
if( newcon < 0 ) continue;
newsockfd = accept( mainsockfd, (struct sockaddr*)&c , &len );
log( "ͬ��: %d\n" , newsockfd );
if( newsockfd < 0 ){
unregMemBuf( newcon );
continue;
}
set_nodelay( newsockfd );
con[newcon].fd = newsockfd;
memcpy( &con[newcon].remoteaddr , &c ,sizeof(c));
tis[accepted] = newcon;
accepted ++;
}
}
return accepted;
}
