=================================================================== RCS file: /home/cvs/OpenXM/src/kan96xx/plugin/sm1Socket.c,v retrieving revision 1.1.1.1 retrieving revision 1.15 diff -u -p -r1.1.1.1 -r1.15 --- OpenXM/src/kan96xx/plugin/sm1Socket.c 1999/10/08 02:12:05 1.1.1.1 +++ OpenXM/src/kan96xx/plugin/sm1Socket.c 2003/09/16 02:57:40 1.15 @@ -1,3 +1,4 @@ +/* $OpenXM: OpenXM/src/kan96xx/plugin/sm1Socket.c,v 1.14 2002/10/30 13:23:06 takayama Exp $ */ /* msg0s.c */ #include #include @@ -13,7 +14,11 @@ #include "sm1Socket.h" extern int Quiet; +static int Post_debug=0; static void errorMsg1s(char *s); +static int getContentLength(char *s); +static int getReceivedContentLength(char *s); +#define MAX_LISTEN_QUEUE 3 /* [(sm1.socket) (open) [optional integer port, optional string name] ] extension ; */ struct object KsocketOpen(struct object obj) { @@ -26,7 +31,12 @@ struct object KsocketOpen(struct object obj) { int on; int tt; extern int errno; + extern int Post_debug; + if ((char *)getenv("OXWEB_DEBUG") != NULL) { + Post_debug = 1; + } + if (obj.tag != Sarray) { errorMsg1s("KsocketOpen([optional integer,optional string name])"); } @@ -57,7 +67,7 @@ struct object KsocketOpen(struct object obj) { me.sin_family = AF_INET; me.sin_port = htons(portNumber); bcopy(myhost->h_addr, - &me.sin_addr,myhost->h_length); + &me.sin_addr,myhost->h_length); if ((s_waiting = socket(AF_INET,SOCK_STREAM,0)) < 0) { errorMsg1s("Socket allocation is failed."); @@ -77,12 +87,12 @@ struct object KsocketOpen(struct object obj) { } - if (listen(s_waiting,1) < 0) { + if (listen(s_waiting,MAX_LISTEN_QUEUE) < 0) { errorMsg1s("Listen failed"); } /* - fprintf(stderr,"Done the initialization. port =%d\n",ntohs(me.sin_port)); - fprintf(stderr,"socket id = %d",accept(s_waiting,NULL,NULL)); */ + fprintf(stderr,"Done the initialization. port =%d\n",ntohs(me.sin_port)); + fprintf(stderr,"socket id = %d",accept(s_waiting,NULL,NULL)); */ rob = newObjectArray(2); putoa(rob,0,KpoInteger(s_waiting)); putoa(rob,1,KpoInteger(ntohs(me.sin_port))); @@ -131,7 +141,7 @@ struct object KsocketConnect(struct object obj) { server.sin_family = AF_INET; server.sin_port = htons(portNumber); bcopy(servhost->h_addr, - (char *)&server.sin_addr,servhost->h_length); + (char *)&server.sin_addr,servhost->h_length); if ((socketid = socket(AF_INET,SOCK_STREAM,0)) <0) { errorMsg1s("socket allocation is failed.\n"); @@ -169,7 +179,11 @@ struct object KsocketAccept(struct object obj) { } s = KopInteger(obj1); if ((news = accept(s,NULL,NULL)) < 0) { - errorMsg1s("Error in accept."); + fprintf(stderr,"Error in accept. Retrying (KsocketAccept).\n"); + /* Code added for strange behavior on cygwin. */ + if ((news = accept(s,NULL,NULL)) < 0) { + errorMsg1s("Error in accept. Retry failed."); + } } if (close(s) < 0) { errorMsg1s("Error in closing the old socket."); @@ -179,6 +193,46 @@ struct object KsocketAccept(struct object obj) { return(robj); } +/* [ integer socketid ] + [ integer newsocketid ] */ +/* It does not close the listening socket. You can call it as + ls = open. + fd=accept2(ls). close(fd). + fd=accept2(ls). close(fd). + .... + */ +struct object KsocketAccept2(struct object obj) { + struct object obj1; + struct object obj2; + struct object robj; + int s, news; + + if (obj.tag != Sarray) { + errorMsg1s("KsocketAccept([integer socketid])"); + } + if (getoaSize(obj) < 1) { + errorMsg1s("KsocketAccept([integer socketid])"); + } + obj1 = getoa(obj,0); + if (obj1.tag != Sinteger ) { + errorMsg1s("KsocketAccept([integer socketid]), argument must be integer."); + } + s = KopInteger(obj1); + if ((news = accept(s,NULL,NULL)) < 0) { + fprintf(stderr,"Error in accept. Retrying (KsocketAccept2).\n"); + /* Code added for strange behavior on cygwin. */ + if ((news = accept(s,NULL,NULL)) < 0) { + errorMsg1s("Error in accept. Retry failed."); + } + } + if (close(s) < 0) { + errorMsg1s("Error in closing the old socket."); + } + robj = newObjectArray(1); + putoa(robj,0,KpoInteger(news)); + return(robj); +} + int KsocketSelect0(int fd,int t) { fd_set readfds; struct timeval timeout; @@ -236,7 +290,7 @@ struct object KsocketSelect(struct object obj) { } struct object KsocketSelectMulti(struct object obj) -/* [ [integer socketid1, integer socketid2, ...] + /* [ [integer socketid1, integer socketid2, ...] optional integer timeout default 0] [ result1, result2, ....] */ @@ -349,14 +403,14 @@ struct object KsocketRead(struct object obj) { if (datasize - totalsize > 0) { n = read(socketid,data+totalsize,datasize-totalsize); if (n < 0) { - errorMsg1s("Read error."); + errorMsg1s("Read error."); } if (n < datasize-totalsize) { - totalsize += n; - break; + totalsize += n; + break; }else{ totalsize += n; } if (totalsize == 0) { - errorMsg1s("Select returns 1, but there is no data to read."); + errorMsg1s("Select returns 1, but there is no data to read."); } }else { /* increase the datasize */ tmp = (char *)GC_malloc(sizeof(char)*2*datasize); @@ -368,10 +422,12 @@ struct object KsocketRead(struct object obj) { } } - r = (char *)GC_malloc(sizeof(char)*(n+1)); + r = (char *)GC_malloc(sizeof(char)*(totalsize+1)); if (r == (char *)NULL) errorMsg1s("Out of Memory."); bcopy(data,r,totalsize); - robj = KpoString(r); + r[totalsize] = 0; + robj = KpoString(r); /* BUG: it works only for reading string from TCP/IP + stream. */ return(robj); @@ -475,28 +531,180 @@ struct object KsocketWriteByte(struct object obj) { struct object ob1; struct object ob2; int socketid; - int r; - char data[2]; + int r,i,n,kk,r0; +#define DATA_SIZE 1024 + char data[DATA_SIZE]; if (obj.tag != Sarray) { - errorMsg1s("KsocketWriteByte([integer socketid, int])"); + errorMsg1s("KsocketWriteByte([integer socketid, int | array of int])"); } if (getoaSize(obj) < 2) { - errorMsg1s("KsocketWriteByte([integer socketid, int])"); + errorMsg1s("KsocketWriteByte([integer socketid, int | array of int])"); } ob1 = getoa(obj,0); ob2 = getoa(obj,1); if (ob1.tag != Sinteger) { - errorMsg1s("KsocketWriteByte([integer socketid, int]) : the first argument must be an integer."); + errorMsg1s("KsocketWriteByte([integer socketid, int | array of int]) : the first argument must be an integer."); } socketid = KopInteger(ob1); - if (ob2.tag != Sinteger) { - errorMsg1s("KsocketWriteByte([integer socketid, int]) : the second argument must be a string."); + if (ob2.tag != Sinteger && ob2.tag != Sarray) { + errorMsg1s("KsocketWriteByte([integer socketid, int | array of int]) : the second argument must be an integer or an array of integers."); } - data[0] = KopInteger(ob2); - r = write(socketid,data, 1); + if (ob2.tag == Sinteger) { + data[0] = KopInteger(ob2); + r = write(socketid,data, 1); + }else{ + n = getoaSize(ob2); kk = 0; r = 0; + for (i=0; i= DATA_SIZE) { + r0 = write(socketid,data,kk); + if (r0 != kk) { + fprintf(stderr,"Warning: Could not write to the socket.\n"); + return(KpoInteger(r+r0)); + } + r += r0; + kk = 0; + } + } + if (kk > 0) { + r0 = write(socketid,data,kk); + if (r0 != kk) { + fprintf(stderr,"Warning: Could not write to the socket.\n"); + return(KpoInteger(r+r0)); + } + r += r0; + } + } return(KpoInteger(r)); } +struct object KsocketReadHTTP(struct object socketObj) { + /* Read until two empty line appears. */ + struct object ob; + struct object ob1; + struct object nob; + char *s; + char *sss; + char *tmp; + int i; + int flag; + int flagmax = 1; + int datasize; + int last; + int contentLength=-1; + int socketid; + extern int Post_debug; + nob = NullObject; + + if (socketObj.tag != Sarray) { + errorMsg1s("KsocketReadHTTP([integer socketid])"); + } + if (getoaSize(socketObj) < 1) { + errorMsg1s("KsocketReadHTTP([integer socketid])"); + } + ob1 = getoa(socketObj,0); + if (ob1.tag != Sinteger) { + errorMsg1s("KsocketReadHTTP([integer socketid]) : the argument must be integer."); + } + socketid = KopInteger(ob1); + + if (KsocketSelect0(socketid,-1) != 1) { + return(nob); + } + ob = KsocketRead(socketObj); + s = KopString(ob); + if (strncmp(s,"POST",4) == 0) flagmax=2; /* for IE */ + else flagmax=1; + flag = 0; + for (i=strlen(s)-1; i>=0; i--) { + if ((s[i] == '\n') && (i==0)) { + ++flag; + }else if ((s[i] == '\n') && (s[i-1] == '\n')) { + ++flag; + }else if ((s[i] == 0xd) && (s[i+1] == 0xa) && (i == 0)) { + ++flag; + }else if ((s[i] == 0xa) && (s[i-1] == 0xd) && (s[i+1] == 0xd) && (s[i+2] == 0xa)) { + ++flag; + } + } + if (flag >= flagmax) return ob; + datasize = strlen(s); + sss = s; + if ((s[strlen(s)-1] == '\n') || + (s[strlen(s)-2] == 0xd) && (s[strlen(s)-1] == 0xa)) { + last = 1; + }else last = 0; + + while (flag < flagmax) { + contentLength = getContentLength(sss); + if (contentLength != -1) { + if (contentLength <= getReceivedContentLength(sss)) { + break; + } + } + if (Post_debug) { + fprintf(stderr,"Waiting in socketReadBlock. flagmax(0d,0a)=%d, content-length=%d, received content-length=%d\n",flagmax,contentLength,getReceivedContentLength(sss)); + } + if (strlen(s) == 0) { + fprintf(stderr,"No data. Perhaps connection is closed by foreign host.\n"); + return nob; + }else{ + /* for debugging. */ + if (Post_debug) { + for (i=0; i= ' ') && (sss[i] < 0x7f)) { + fprintf(stderr,"%c",sss[i]); + }else{ + fprintf(stderr,"(%3x)",sss[i]); + if (sss[i] == 0xa) fprintf(stderr,"\n"); + } + + } + fprintf(stderr,"\n"); + } + } + + if (KsocketSelect0(socketid,-1) != 1) { + return nob; + } + ob = KsocketRead(socketObj); + s = KopString(ob); + for (i=strlen(s)-1; i>=0; i--) { + if ((s[i] == '\n') && (i==0) && last) { + ++flag; + }else if ((s[i] == '\n') && (s[i-1] == '\n')) { + ++flag; + }else if ((s[i] == 0xd) && (s[i+1] == 0xa) && (i==0) && last) { + ++flag; + }else if ((s[i] == 0xa) && (s[i-1] == 0xd) && (s[i+1] == 0xd) && (s[i+2] == 0xa)) { + ++flag; + } + } + if (datasize-1 <= strlen(sss)+strlen(s)) { + tmp = (char *)GC_malloc(sizeof(char)*2*(datasize+strlen(s))+1); + if (tmp == (char *)NULL) errorMsg1s("Out of Memory."); + strcpy(tmp,sss); + strcat(tmp,s); + datasize = 2*(datasize+strlen(s)); + sss = tmp; + }else{ + strcat(sss,s); + } + + if ((s[strlen(s)-1] == '\n') || + (s[strlen(s)-2] == 0xd) && (s[strlen(s)-1] == 0xa)) { + last = 1; + }else last = 0; + + } + + return KpoString(sss); + +} + struct object Kplugin_sm1Socket(char *key,struct object obj) { struct object robj = NullObject; if (strcmp(key,"open") == 0) { @@ -505,12 +713,18 @@ struct object Kplugin_sm1Socket(char *key,struct objec robj = KsocketConnect(obj); }else if (strcmp(key,"accept") == 0) { robj = KsocketAccept(obj); + }else if (strcmp(key,"accept2") == 0) { + robj = KsocketAccept2(obj); }else if (strcmp(key,"select") == 0) { robj = KsocketSelect(obj); }else if (strcmp(key,"mselect") == 0) { robj = KsocketSelectMulti(obj); }else if (strcmp(key,"read") == 0) { robj = KsocketRead(obj); + }else if (strcmp(key,"readHTTP") == 0) { + robj = KsocketReadHTTP(obj); + }else if (strcmp(key,"gethostname") == 0) { + robj = KsocketGetHostName(); }else if (strcmp(key,"write") == 0) { robj = KsocketWrite(obj); }else if (strcmp(key,"read") == 0) { @@ -528,8 +742,57 @@ struct object Kplugin_sm1Socket(char *key,struct objec } +static int getContentLength(char *s) { + int n; + int i,j; + int len = -1; + char *s1 = "content-length:"; + char s0[256]; + int m; + m = strlen(s1); + n = strlen(s); + for (i=0; i= 'A') && (s0[j] <= 'Z')) s0[j] = s0[j]+0x20; + } + if (strncmp(s0,s1,strlen(s1)) == 0) { + sscanf(&(s[i+strlen(s1)]),"%d",&len); + break; + } + } + return len; +} +static int getReceivedContentLength(char *s) { + int n; + int i; + int start; + start = -1; + n = strlen(s); + for (i=0; i