receive buffer size
[lcore.git] / lsocket.pas
1 {lsocket.pas}\r
2 \r
3 {socket code by plugwash}\r
4 \r
5 {use this code for whatever you like in programs under whater licence you like\r
6  *  This program is distributed in the hope that it will be useful,\r
7  *  but WITHOUT ANY WARRANTY; without even the implied warranty of\r
8  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\r
9 }\r
10 \r
11 {\r
12 changes by plugwash (20030728)\r
13 * created handlefdtrigger virtual method in tlasio (overridden in tlsocket) and moved a lot of code from messageloop into it\r
14 * changed tlasio to tlasio\r
15 * split fdhandle into fdhandlein and fdhandleout\r
16 * i now use fdsrmaster and fdswmaster instead of rebuilding the lists every loop\r
17 * split lsocket.pas into lsocket.pas and lcore.pas\r
18 \r
19 \r
20 changes by beware (20030903)\r
21 * added getxaddr, getxport (local addr, port, as string)\r
22 * added getpeername, remote addr+port as binary\r
23 * added htons and htonl functions (endian swap, same interface as windows API)\r
24 \r
25 beware (20030905)\r
26 * if connect failed (conn refused) set state to connected and call internalclose, to get closed handler (instead of fdclose)\r
27 * (lcore) if closing the fd's in internalcose, set fd's to -1 because closing an fd makes it invalid\r
28 \r
29 beware (20030927)\r
30 * fixed: on connect failed, tried to close fdhandle's which were already set to -1, added check\r
31 \r
32 beware (20031017)\r
33 * added getpeeraddr, getpeerport, remote addr+port as string\r
34 }\r
35 \r
36 \r
37 unit lsocket;\r
38 {$mode delphi}\r
39 interface\r
40   uses\r
41     sysutils,\r
42     {$ifdef VER1_0}\r
43       linux,\r
44     {$else}\r
45       baseunix,unix,\r
46     {$endif}\r
47     sockets,classes,pgdebugout,pgtypes,lcore,fd_utils,binipstuff,dnssync;\r
48 type\r
49   sunB = packed record\r
50     s_b1, s_b2, s_b3, s_b4: byte;\r
51   end;\r
52 \r
53   SunW = packed record\r
54     s_w1, s_w2: word;\r
55   end;\r
56 \r
57   TInAddr = packed record\r
58     case integer of\r
59       0: (S_un_b: SunB);\r
60       1: (S_un_w: SunW);\r
61       2: (S_addr: cardinal);\r
62   end;\r
63   {$ifdef ipv6}\r
64     {$ifdef ver1_0}\r
65       cuint16=word;\r
66       cuint32=dword;\r
67       sa_family_t=word;\r
68 \r
69 \r
70       TInetSockAddr6 = packed Record\r
71         sin6_family   : sa_family_t;\r
72         sin6_port     : cuint16;\r
73         sin6_flowinfo : cuint32;\r
74         sin6_addr     : Tin6_addr;\r
75         sin6_scope_id : cuint32;\r
76       end;\r
77     {$endif}\r
78   {$endif}\r
79   TinetSockAddrv = packed record\r
80     case integer of\r
81       0: (InAddr:TInetSockAddr);\r
82       {$ifdef ipv6}\r
83       1: (InAddr6:TInetSockAddr6);\r
84       {$endif}\r
85   end;\r
86 \r
87   type\r
88     tsockaddrin=TInetSockAddr;\r
89 \r
90   type\r
91     TLsocket = class(tlasio)\r
92     public\r
93       //a: string;\r
94 \r
95       inAddr             : TInetSockAddrV;\r
96 {      inAddrSize:integer;}\r
97 \r
98       //host               : THostentry      ;\r
99 \r
100       //mainthread         : boolean         ; //for debuggin only\r
101       addr:string;\r
102       port:string;\r
103       localaddr:string;\r
104       localport:string;\r
105       proto:string;\r
106       udp:boolean;\r
107       listenqueue:integer;\r
108       function getaddrsize:integer;\r
109       procedure connect; virtual;\r
110       procedure bindsocket;\r
111       procedure listen;\r
112       function accept : longint;\r
113       function sendto(dest:TInetSockAddr;destlen:integer;data:pointer;len:integer):integer; virtual;\r
114       function receivefrom(data:pointer;len:integer;var src:TInetSockAddr;var srclen:integer):integer; virtual;\r
115       //procedure internalclose(error:word);override;\r
116       procedure handlefdtrigger(readtrigger,writetrigger:boolean); override;\r
117       function send(data:pointer;len:integer):integer;override;\r
118       procedure sendstr(const str : string);override;\r
119       function Receive(Buf:Pointer;BufSize:integer):integer; override;\r
120       function getpeername(var addr:tsockaddrin;addrlen:integer):integer; virtual;\r
121       procedure getXaddrbin(var binip:tbinip); virtual;\r
122       procedure getpeeraddrbin(var binip:tbinip); virtual;\r
123       function getXaddr:string; virtual;\r
124       function getpeeraddr:string; virtual;\r
125       function getXport:string; virtual;\r
126       function getpeerport:string; virtual;\r
127       constructor Create(AOwner: TComponent); override;\r
128     end;\r
129     tsocket=longint; // for compatibility with twsocket\r
130 \r
131   twsocket=tlsocket; {easy}\r
132 \r
133 function htons(w:word):word;\r
134 function htonl(i:integer):integer;\r
135 {!!!function longipdns(s:string):longint;}\r
136 \r
137 {$ifdef ipv6}\r
138 const\r
139   v4listendefault:boolean=false;\r
140 {$endif}\r
141 \r
142 \r
143 const\r
144   TCP_NODELAY=1;\r
145   IPPROTO_TCP=6;\r
146 \r
147 implementation\r
148 {$include unixstuff.inc}\r
149 \r
150 function longip(s:string):longint;inline;\r
151 var\r
152   l:longint;\r
153   a,b:integer;\r
154 \r
155 function convertbyte(const s:string):integer;inline;\r
156 begin\r
157   result := strtointdef(s,-1);\r
158   if result < 0 then exit;\r
159   if result > 255 then exit;\r
160 \r
161   {01 exception}\r
162   if (result <> 0) and (s[1] = '0') then begin\r
163     result := -1;\r
164     exit;\r
165   end;\r
166 \r
167   {+1 exception}\r
168   if not (s[1] in ['0'..'9']) then begin\r
169     result := -1;\r
170     exit\r
171   end;\r
172 end;\r
173 \r
174 begin\r
175   result := 0;\r
176   a := pos('.',s);\r
177   if a = 0 then exit;\r
178   b := convertbyte(copy(s,1,a-1));if (b < 0) then exit;\r
179   l := b shl 24;\r
180   s := copy(s,a+1,256);\r
181   a := pos('.',s);\r
182   if a = 0 then exit;\r
183   b := convertbyte(copy(s,1,a-1));if (b < 0) then exit;\r
184   l := l or b shl 16;\r
185   s := copy(s,a+1,256);\r
186   a := pos('.',s);\r
187   if a = 0 then exit;\r
188   b := convertbyte(copy(s,1,a-1));if (b < 0) then exit;\r
189   l := l or b shl 8;\r
190   s := copy(s,a+1,256);\r
191   b := convertbyte(copy(s,1,256));if (b < 0) then exit;\r
192   l := l or b;\r
193   result := l;\r
194 end;\r
195 \r
196 (*!!!\r
197 function longipdns(s:string):longint;\r
198 var\r
199   host : thostentry;\r
200 begin\r
201   if s = '0.0.0.0' then begin\r
202     result := 0;\r
203   end else begin\r
204     result := longip(s);\r
205     if result = 0 then begin\r
206       if gethostbyname(s,host) then begin;\r
207         result := htonl(Longint(Host.Addr));\r
208       end;\r
209       //writeln(inttohex(longint(host.addr),8))\r
210     end;\r
211     if result = 0 then begin\r
212       if resolvehostbyname(s,host) then begin;\r
213         result := htonl(Longint(Host.Addr));\r
214       end;\r
215       //writeln(inttohex(longint(host.addr),8))\r
216     end;\r
217   end;\r
218 end;\r
219 *)\r
220 \r
221 \r
222 function htons(w:word):word;\r
223 begin\r
224   {$ifdef ENDIAN_LITTLE}\r
225   result := ((w and $ff00) shr 8) or ((w and $ff) shl 8);\r
226   {$else}\r
227   result := w;\r
228   {$endif}\r
229 end;\r
230 \r
231 function htonl(i:integer):integer;\r
232 begin\r
233   {$ifdef ENDIAN_LITTLE}\r
234   result := (i shr 24) or (i shr 8 and $ff00) or (i shl 8 and $ff0000) or (i shl 24 and $ff000000);\r
235   {$else}\r
236   result := i;\r
237   {$endif}\r
238 end;\r
239 \r
240 function tlsocket.getaddrsize:integer;\r
241 begin\r
242   {$ifdef ipv6}\r
243   if inaddr.inaddr.family = AF_INET6 then result := sizeof(tinetsockaddr6) else\r
244   {$endif}\r
245   result := sizeof(tinetsockaddr);\r
246 end;\r
247 \r
248 function makeinaddrv(addr,port:string;var inaddr:tinetsockaddrv):integer;\r
249 var\r
250   biniptemp:tbinip;\r
251 begin\r
252   result := 0;\r
253   biniptemp := forwardlookup(addr,10);\r
254   fillchar(inaddr,sizeof(inaddr),0);\r
255   if biniptemp.family = AF_INET then begin\r
256     inAddr.InAddr.family:=AF_INET;\r
257     inAddr.InAddr.port:=htons(strtointdef(port,0));\r
258     inAddr.InAddr.addr:=biniptemp.ip;\r
259     result := sizeof(tinetsockaddr);\r
260   end else\r
261   {$ifdef ipv6}\r
262   if biniptemp.family = AF_INET6 then begin\r
263     inAddr.InAddr6.sin6_family:=AF_INET6;\r
264     inAddr.InAddr6.sin6_port:=htons(strtointdef(port,0));\r
265     inAddr.InAddr6.sin6_addr:=biniptemp.ip6;\r
266     result := sizeof(tinetsockaddr6);\r
267   end else\r
268   {$endif}\r
269   raise esocketexception.create('unable to resolve address: '+addr);\r
270 end;\r
271 \r
272 procedure tlsocket.connect;\r
273 var\r
274   a:integer;\r
275 begin\r
276   if state <> wsclosed then close;\r
277   //prevtime := 0;\r
278   makeinaddrv(addr,port,inaddr);\r
279 \r
280   udp := uppercase(proto) = 'UDP';\r
281   if udp then a := SOCK_DGRAM else a := SOCK_STREAM;\r
282   a := Socket(inaddr.inaddr.family,a,0);\r
283 \r
284   if a = -1 then begin\r
285     lasterror := socketerror;\r
286     raise esocketexception.create('unable to create socket');\r
287   end;\r
288   try\r
289     dup(a);\r
290     bindsocket;\r
291     if udp then begin\r
292       state := wsconnected;\r
293       if assigned(onsessionconnected) then onsessionconnected(self,0);\r
294     end else begin\r
295       state :=wsconnecting;\r
296       sockets.Connect(fdhandlein,inADDR,getaddrsize);\r
297     end;\r
298     rmasterset(fdhandlein);\r
299     if udp then begin\r
300       wmasterclr(fdhandleout);\r
301     end else begin\r
302       wmasterset(fdhandleout);\r
303     end;\r
304     //sendq := '';\r
305   except\r
306     on e: exception do begin\r
307       fdcleanup;\r
308       raise; //reraise the exception\r
309     end;\r
310   end;\r
311 end;\r
312 \r
313 procedure tlsocket.sendstr(const str : string);\r
314 begin\r
315   if udp then begin\r
316     send(@str[1],length(str))\r
317   end else begin\r
318     inherited sendstr(str);\r
319   end;\r
320 end;\r
321 \r
322 function tlsocket.send(data:pointer;len:integer):integer;\r
323 begin\r
324   if udp then begin\r
325     result := sendto(inaddr.inaddr,getaddrsize,data,len)\r
326   end else begin\r
327     result := inherited send(data,len);\r
328   end;\r
329 end;\r
330 \r
331 \r
332 function tlsocket.receive(Buf:Pointer;BufSize:integer):integer;\r
333 begin\r
334   if udp then begin\r
335     result := fdread(self.fdhandlein,buf^,bufsize);\r
336   end else begin\r
337     result := inherited receive(buf,bufsize);\r
338   end;\r
339 end;\r
340 \r
341 procedure tlsocket.bindsocket;\r
342 var\r
343   a:integer;\r
344   inAddrtemp:TInetSockAddrV;\r
345   inaddrtempsize:integer;\r
346 begin\r
347   try\r
348     if (localaddr <> '') or (localport <> '') then begin\r
349       if localaddr = '' then begin\r
350         {$ifdef ipv6}\r
351         if inaddr.inaddr.family = AF_INET6 then localaddr := '::' else\r
352         {$endif}\r
353         localaddr := '0.0.0.0';\r
354       end;\r
355       //gethostbyname(localaddr,host);\r
356 \r
357       inaddrtempsize := makeinaddrv(localaddr,localport,inaddrtemp);\r
358 \r
359       If Not Bind(fdhandlein,inaddrtemp,inaddrtempsize) Then begin\r
360         state := wsclosed;\r
361         lasterror := socketerror;\r
362         raise ESocketException.create('unable to bind, error '+inttostr(lasterror));\r
363       end;\r
364       state := wsbound;\r
365     end;\r
366   except\r
367     on e: exception do begin\r
368       fdcleanup;\r
369       raise; //reraise the exception\r
370     end;\r
371   end;\r
372 end;\r
373 \r
374 procedure tlsocket.listen;\r
375 var\r
376   yes:longint;\r
377   socktype:integer;\r
378   biniptemp:tbinip;\r
379   origaddr:string;\r
380 begin\r
381   if state <> wsclosed then close;\r
382   udp := uppercase(proto) = 'UDP';\r
383   if udp then socktype := SOCK_DGRAM else socktype := SOCK_STREAM;\r
384   origaddr := addr;\r
385 \r
386   if addr = '' then begin\r
387     {$ifdef ipv6}\r
388     if not v4listendefault then begin\r
389       addr := '::';\r
390     end else\r
391     {$endif}\r
392     addr := '0.0.0.0';\r
393   end;\r
394   biniptemp := forwardlookup(addr,10);\r
395   addr := ipbintostr(biniptemp);\r
396   fdhandlein := socket(biniptemp.family,socktype,0);\r
397   {$ifdef ipv6}\r
398   if (addr = '::') and (origaddr = '') and (fdhandlein < 0) then begin\r
399     addr := '0.0.0.0';\r
400     fdhandlein := socket(AF_INET,socktype,0);\r
401   end;\r
402   {$endif}\r
403   if fdhandlein = -1 then raise ESocketException.create('unable to create socket');\r
404   dup(fdhandlein); // sets up maxs and copies handle to fdhandleout among other things\r
405   fdreverse[fdhandlein] := self;\r
406   state := wsclosed; // then set this back as it was an undesired side effect of dup\r
407 \r
408   try\r
409     yes := $01010101;  {Copied this from existing code. Value is empiric,\r
410                     but works. (yes=true<>0) }\r
411     if SetSocketOptions(fdhandlein, SOL_SOCKET, SO_REUSEADDR,yes,sizeof(yes))=-1 then begin\r
412       raise ESocketException.create('unable to set socket options');\r
413     end;\r
414 \r
415     localaddr := addr;\r
416     localport := port;\r
417     bindsocket;\r
418 \r
419     if not udp then begin\r
420       {!!! allow custom queue length? default 5}\r
421       if listenqueue = 0 then listenqueue := 5;\r
422       If Not sockets.Listen(fdhandlein,listenqueue) Then raise esocketexception.create('unable to listen');\r
423       state := wsListening;\r
424     end else begin\r
425       state := wsconnected;\r
426     end;\r
427   finally\r
428     if state = wsclosed then begin\r
429       if fdhandlein >= 0 then begin\r
430         {one *can* get here without fd -beware}\r
431         rmasterclr(fdhandlein);\r
432         fdclose(fdhandlein); // we musnt leak file discriptors\r
433         fdreverse[fdhandlein] := nil;\r
434         fdhandlein := -1;\r
435       end;\r
436     end else begin\r
437       rmasterset(fdhandlein);\r
438     end;\r
439     if fdhandleout >= 0 then wmasterclr(fdhandleout);\r
440   end;\r
441 end;\r
442 \r
443 function tlsocket.accept : longint;\r
444 var\r
445   FromAddrSize     : LongInt;        // i don't realy know what to do with these at this\r
446   FromAddr         : TInetSockAddr;  // at this point time will tell :)\r
447 begin\r
448   rmasterset(fdhandlein);\r
449   FromAddrSize := Sizeof(FromAddr);\r
450   result := sockets.accept(fdhandlein,fromaddr,fromaddrsize);\r
451   if result = -1 then raise esocketexception.create('error '+inttostr(socketerror)+' while accepting');\r
452   if result > absoloutemaxs then begin\r
453     fdclose(result);\r
454     result := -1;\r
455     raise esocketexception.create('file discriptor out of range');\r
456   end;\r
457 end;\r
458 \r
459 function tlsocket.sendto;\r
460 begin\r
461   result := sockets.sendto(self.fdhandleout,data^,len,0,dest,destlen);\r
462 end;\r
463 \r
464 function tlsocket.receivefrom;\r
465 begin\r
466   result := sockets.recvfrom(self.fdhandlein,data^,len,0,src,srclen);\r
467 end;\r
468 \r
469 procedure tlsocket.handlefdtrigger(readtrigger,writetrigger:boolean);\r
470 var\r
471   tempbuf:array[0..receivebufsize-1] of byte;\r
472 begin\r
473   if (state =wslistening) and readtrigger then begin\r
474 {    debugout('listening socket triggered on read');}\r
475     rmasterclr(fdhandlein);\r
476     if assigned(onsessionAvailable) then onsessionAvailable(self,0);\r
477   end;\r
478   if udp and readtrigger then begin\r
479     if assigned(ondataAvailable) then ondataAvailable(self,0);\r
480     {!!!test}\r
481     exit;\r
482   end;\r
483   if (state =wsconnecting) and writetrigger then begin\r
484     // code for dealing with the reults of a non-blocking connect is\r
485     // rather complex\r
486     // if just write is triggered it means connect suceeded\r
487     // if both read and write are suceededed it can mean 2 things\r
488     // 1: connect ok and data availible\r
489     // 2: connect fail\r
490     // to find out which you must read from the socket and look for errors\r
491     // there if we read successfully we drop through into the code for fireing\r
492     // the read event\r
493     if not readtrigger then begin\r
494       state := wsconnected;\r
495       if assigned(onsessionconnected) then onsessionconnected(self,0);\r
496     end else begin\r
497       numread := fdread(fdhandlein,tempbuf,sizeof(tempbuf));\r
498       if numread <> -1 then begin\r
499         state := wsconnected;\r
500         if assigned(onsessionconnected) then onsessionconnected(self,0);\r
501         //connectread := true;\r
502         recvq.add(@tempbuf,numread);\r
503       end else begin\r
504         state := wsconnected;\r
505         if assigned(onsessionconnected) then onsessionconnected(self,linuxerror);\r
506 {        debugout('connect fail');}\r
507         self.internalclose(0);\r
508         recvq.del(maxlongint);\r
509       end;\r
510       // if things went well here we are now in the state wsconnected with data sitting in our receive buffer\r
511       // so we drop down into the processing for data availible\r
512     end;\r
513     if fdhandlein >= 0 then begin\r
514       if state = wsconnected then begin\r
515         rmasterset(fdhandlein);\r
516       end else begin\r
517         rmasterclr(fdhandlein);\r
518       end;\r
519     end;\r
520     if fdhandleout >= 0 then begin\r
521       if sendq.size = 0 then begin\r
522         //don't clear the bit in fdswmaster if data is in the sendq\r
523         wmasterclr(fdhandleout);\r
524       end;\r
525     end;\r
526 \r
527   end;\r
528   inherited handlefdtrigger(readtrigger,writetrigger);\r
529 end;\r
530 \r
531 constructor tlsocket.Create(AOwner: TComponent);\r
532 begin\r
533   inherited create(aowner);\r
534   closehandles := true;\r
535 end;\r
536 \r
537 \r
538 function tlsocket.getpeername(var addr:tsockaddrin;addrlen:integer):integer;\r
539 begin\r
540   result := sockets.getpeername(self.fdhandlein,addr,addrlen);\r
541 end;\r
542 \r
543 procedure tlsocket.getxaddrbin(var binip:tbinip);\r
544 var\r
545   addr:tinetsockaddrv;\r
546   i:integer;\r
547 begin\r
548   i := sizeof(addr);\r
549   fillchar(addr,sizeof(addr),0);\r
550   sockets.getsocketname(self.fdhandlein,addr,i);\r
551 \r
552   binip.family := addr.inaddr.family;\r
553   {$ifdef ipv6}\r
554   if addr.inaddr6.sin6_family = AF_INET6 then begin\r
555     binip.ip6 := addr.inaddr6.sin6_addr;\r
556   end else\r
557   {$endif}\r
558   begin\r
559     binip.ip := addr.inaddr.addr;\r
560   end;\r
561   converttov4(binip);\r
562 end;\r
563 \r
564 procedure tlsocket.getpeeraddrbin(var binip:tbinip);\r
565 var\r
566   addr:tinetsockaddrv;\r
567   i:integer;\r
568 begin\r
569   i := sizeof(addr);\r
570   fillchar(addr,sizeof(addr),0);\r
571   sockets.getpeername(self.fdhandlein,addr,i);\r
572 \r
573   binip.family := addr.inaddr.family;\r
574   {$ifdef ipv6}\r
575   if addr.inaddr6.sin6_family = AF_INET6 then begin\r
576     binip.ip6 := addr.inaddr6.sin6_addr;\r
577   end else\r
578   {$endif}\r
579   begin\r
580     binip.ip := addr.inaddr.addr;\r
581   end;\r
582   converttov4(binip);\r
583 end;\r
584 \r
585 function tlsocket.getXaddr:string;\r
586 var\r
587   biniptemp:tbinip;\r
588 begin\r
589   getxaddrbin(biniptemp);\r
590   result := ipbintostr(biniptemp);\r
591   if result = '' then result := 'error';\r
592 end;\r
593 \r
594 function tlsocket.getpeeraddr:string;\r
595 var\r
596   biniptemp:tbinip;\r
597 begin\r
598   getpeeraddrbin(biniptemp);\r
599   result := ipbintostr(biniptemp);\r
600   if result = '' then result := 'error';\r
601 end;\r
602 \r
603 function tlsocket.getXport:string;\r
604 var\r
605   addr:tinetsockaddr;\r
606   i:integer;\r
607 begin\r
608   i := sizeof(addr);\r
609   sockets.getsocketname(self.fdhandlein,addr,i);\r
610   i := htons(addr.port);\r
611   result := inttostr(i);\r
612 end;\r
613 \r
614 function tlsocket.getpeerport:string;\r
615 var\r
616   addr:tinetsockaddr;\r
617   i:integer;\r
618 begin\r
619   i := sizeof(addr);\r
620   sockets.getpeername(self.fdhandlein,addr,i);\r
621   i := htons(addr.port);\r
622   result := inttostr(i);\r
623 end;\r
624 \r
625 end.\r