a1433fcc25897c54a057bf667046e868ddd81615
[lcore.git] / binipstuff.pas
1 { Copyright (C) 2005 Bas Steendijk and Peter Green\r
2   For conditions of distribution and use, see copyright notice in zlib_license.txt\r
3   which is included in the package\r
4   ----------------------------------------------------------------------------- }\r
5 unit binipstuff;\r
6 \r
7 interface\r
8 \r
9 {$include lcoreconfig.inc}\r
10 \r
11 {$ifndef win32}\r
12 {$ifdef ipv6}\r
13 uses sockets;\r
14 {$endif}\r
15 {$endif}\r
16 \r
17 {$ifdef fpc}\r
18   {$mode delphi}\r
19 {$endif}\r
20 {$ifdef cpu386}{$define i386}{$endif}\r
21 {$ifdef i386}{$define ENDIAN_LITTLE}{$endif}\r
22 \r
23 {$include uint32.inc}\r
24 \r
25 const\r
26   hexchars:array[0..15] of char='0123456789abcdef';\r
27   AF_INET=2;\r
28   {$ifdef win32}\r
29     AF_INET6=23;\r
30   {$else}\r
31     AF_INET6=10;\r
32   {$endif}\r
33 \r
34 type\r
35   {$ifdef ipv6}\r
36     \r
37     {$ifdef win32}\r
38       {$define want_Tin6_addr}\r
39     {$endif}\r
40     {$ifdef ver1_0}\r
41       {$define want_Tin6_addr}\r
42     {$endif}\r
43     {$ifdef want_Tin6_addr}\r
44       Tin6_addr = packed record\r
45         case byte of\r
46           0: (u6_addr8  : array[0..15] of byte);\r
47           1: (u6_addr16 : array[0..7] of Word);\r
48           2: (u6_addr32 : array[0..3] of uint32);\r
49           3: (s6_addr8  : array[0..15] of shortint);\r
50           4: (s6_addr   : array[0..15] of shortint);\r
51           5: (s6_addr16 : array[0..7] of smallint);\r
52           6: (s6_addr32 : array[0..3] of LongInt);\r
53       end;\r
54     {$endif}\r
55   {$endif}\r
56 \r
57   tbinip=record\r
58     family:integer;\r
59     {$ifdef ipv6}\r
60       case integer of\r
61         0: (ip:longint);\r
62         1: (ip6:tin6_addr);\r
63     {$else}\r
64       ip:longint;\r
65     {$endif}\r
66   end;\r
67 \r
68   {$ifdef win32}\r
69     TInetSockAddr = packed Record\r
70       family:Word;\r
71       port  :Word;\r
72       addr  :uint32;\r
73       pad   :array [1..8] of byte;\r
74     end;\r
75     {$ifdef ipv6}\r
76 \r
77       TInetSockAddr6 = packed record\r
78         sin6_family: word;\r
79         sin6_port: word;\r
80         sin6_flowinfo: uint32;\r
81         sin6_addr: tin6_addr;\r
82         sin6_scope_id: uint32;\r
83       end;\r
84     {$endif}\r
85   {$endif}\r
86 \r
87 \r
88 \r
89     {$ifdef ipv6}\r
90     {$ifdef ver1_0}\r
91       cuint16=word;\r
92       cuint32=dword;\r
93       sa_family_t=word;\r
94 \r
95     {$endif}\r
96   {$endif}\r
97   TinetSockAddrv = packed record\r
98     case integer of\r
99       0: (InAddr:TInetSockAddr);\r
100       {$ifdef ipv6}\r
101       1: (InAddr6:TInetSockAddr6);\r
102       {$endif}\r
103   end;\r
104   Pinetsockaddrv = ^Tinetsockaddrv;\r
105 \r
106   type\r
107     tsockaddrin=TInetSockAddr;\r
108 \r
109 \r
110 \r
111 {\r
112 bin IP list code, by beware\r
113 while this is really just a string, on the interface side it must be treated\r
114 as an opaque var which is passed as "var" when it needs to be modified}\r
115 \r
116   tbiniplist=string;\r
117 \r
118 function biniplist_new:tbiniplist;\r
119 procedure biniplist_add(var l:tbiniplist;ip:tbinip);\r
120 function biniplist_getcount(const l:tbiniplist):integer;\r
121 function biniplist_get(const l:tbiniplist;index:integer):tbinip;\r
122 procedure biniplist_set(var l:tbiniplist;index:integer;ip:tbinip);\r
123 procedure biniplist_setcount(var l:tbiniplist;newlen:integer);\r
124 procedure biniplist_free(var l:tbiniplist);\r
125 procedure biniplist_addlist(var l:tbiniplist;const l2:tbiniplist);\r
126 function biniplist_tostr(const l:tbiniplist):string;\r
127 function isbiniplist(const l:tbiniplist):boolean;\r
128 \r
129 function htons(w:word):word;\r
130 function htonl(i:uint32):uint32;\r
131 \r
132 function ipstrtobin(const s:string;var binip:tbinip):boolean;\r
133 function ipstrtobinf(const s:string):tbinip;\r
134 function ipbintostr(const binip:tbinip):string;\r
135 {$ifdef ipv6}\r
136 function ip6bintostr(const bin:tin6_addr):string;\r
137 function ip6strtobin(const s:string;var bin:tin6_addr):boolean;\r
138 {$endif}\r
139 \r
140 function comparebinip(const ip1,ip2:tbinip):boolean;\r
141 procedure maskbits(var binip:tbinip;bits:integer);\r
142 function comparebinipmask(ip1,ip2:tbinip;bits:integer):boolean;\r
143 \r
144 procedure addipsoffamily(var l:tbiniplist;const l2:tbiniplist;family:integer);\r
145 \r
146 {deprecated}\r
147 function longip(s:string):longint;\r
148 \r
149 procedure converttov4(var ip:tbinip);\r
150 \r
151 function inaddrvtobinip(inaddrv:tinetsockaddrv):tbinip;\r
152 function makeinaddrv(addr:tbinip;port:string;var inaddr:tinetsockaddrv):integer;\r
153 function inaddrsize(inaddr:tinetsockaddrv):integer;\r
154 \r
155 implementation\r
156 \r
157 uses sysutils;\r
158 \r
159 function htons(w:word):word;\r
160 begin\r
161   {$ifdef ENDIAN_LITTLE}\r
162   result := ((w and $ff00) shr 8) or ((w and $ff) shl 8);\r
163   {$else}\r
164   result := w;\r
165   {$endif}\r
166 end;\r
167 \r
168 function htonl(i:uint32):uint32;\r
169 begin\r
170   {$ifdef ENDIAN_LITTLE}\r
171   result := (i shr 24) or (i shr 8 and $ff00) or (i shl 8 and $ff0000) or (i shl 24 and $ff000000);\r
172   {$else}\r
173   result := i;\r
174   {$endif}\r
175 end;\r
176 \r
177 \r
178 function inaddrvtobinip(inaddrv:tinetsockaddrv):tbinip;\r
179 begin\r
180   result.family := inaddrv.inaddr.family;\r
181   if result.family = AF_INET then result.ip := inaddrv.inaddr.addr;\r
182   {$ifdef ipv6}\r
183   if result.family = AF_INET6 then result.ip6 := inaddrv.inaddr6.sin6_addr;\r
184   {$endif}\r
185 end;\r
186 \r
187 function makeinaddrv(addr:tbinip;port:string;var inaddr:tinetsockaddrv):integer;\r
188 begin\r
189   result := 0;\r
190 {  biniptemp := forwardlookup(addr,10);}\r
191   fillchar(inaddr,sizeof(inaddr),0);\r
192   //writeln('converted address '+addr+' to binip '+ipbintostr(biniptemp));\r
193   if addr.family = AF_INET then begin\r
194     inAddr.InAddr.family:=AF_INET;\r
195     inAddr.InAddr.port:=htons(strtointdef(port,0));\r
196     inAddr.InAddr.addr:=addr.ip;\r
197     result := sizeof(tinetsockaddr);\r
198   end else\r
199   {$ifdef ipv6}\r
200   if addr.family = AF_INET6 then begin\r
201     inAddr.InAddr6.sin6_family:=AF_INET6;\r
202     inAddr.InAddr6.sin6_port:=htons(strtointdef(port,0));\r
203     inAddr.InAddr6.sin6_addr:=addr.ip6;\r
204     result := sizeof(tinetsockaddr6);\r
205   end;\r
206   {$endif}\r
207 end;\r
208 \r
209 function inaddrsize(inaddr:tinetsockaddrv):integer;\r
210 begin\r
211   {$ifdef ipv6}\r
212   if inaddr.inaddr.family = AF_INET6 then result := sizeof(tinetsockaddr6) else\r
213   {$endif}\r
214   result := sizeof(tinetsockaddr);\r
215 end;\r
216 \r
217 {internal}\r
218 {converts dotted v4 IP to longint. returns host endian order}\r
219 function longip(s:string):longint;\r
220 var\r
221   l:longint;\r
222   a,b:integer;\r
223 function convertbyte(const s:string):integer;\r
224 begin\r
225   result := strtointdef(s,-1);\r
226   if result < 0 then begin\r
227     result := -1;\r
228     exit;\r
229   end;\r
230   if result > 255 then begin\r
231     result := -1;\r
232     exit;\r
233   end;\r
234   {01 exception}\r
235   if (result <> 0) and (s[1] = '0') then begin\r
236     result := -1;\r
237     exit;\r
238   end;\r
239   {+1 exception}\r
240   if not (s[1] in ['0'..'9']) then begin\r
241     result := -1;\r
242     exit\r
243   end;\r
244 end;\r
245 \r
246 begin\r
247   result := 0;\r
248   a := pos('.',s);\r
249   if a = 0 then exit;\r
250   b := convertbyte(copy(s,1,a-1));if (b < 0) then exit;\r
251   l := b shl 24;\r
252   s := copy(s,a+1,256);\r
253   a := pos('.',s);\r
254   if a = 0 then exit;\r
255   b := convertbyte(copy(s,1,a-1));if (b < 0) then exit;\r
256   l := l or b shl 16;\r
257   s := copy(s,a+1,256);\r
258   a := pos('.',s);\r
259   if a = 0 then exit;\r
260   b := convertbyte(copy(s,1,a-1));if (b < 0) then exit;\r
261   l := l or b shl 8;\r
262   s := copy(s,a+1,256);\r
263   b := convertbyte(copy(s,1,256));if (b < 0) then exit;\r
264   l := l or b;\r
265   result := l;\r
266 end;\r
267 \r
268 \r
269 function ipstrtobinf;\r
270 begin\r
271   ipstrtobin(s,result);\r
272 end;\r
273 \r
274 function ipstrtobin(const s:string;var binip:tbinip):boolean;\r
275 begin\r
276   binip.family := 0;\r
277   result := false;\r
278   {$ifdef ipv6}\r
279   if pos(':',s) <> 0 then begin\r
280     {try ipv6. use builtin routine}\r
281     result := ip6strtobin(s,binip.ip6);\r
282     if result then binip.family := AF_INET6;\r
283     exit;\r
284   end;\r
285   {$endif}\r
286 \r
287   {try v4}\r
288   binip.ip := htonl(longip(s));\r
289   if (binip.ip <> 0) or (s = '0.0.0.0') then begin\r
290     result := true;\r
291     binip.family := AF_INET;\r
292     exit;\r
293   end;\r
294 end;\r
295 \r
296 function ipbintostr(const binip:tbinip):string;\r
297 var\r
298   a:integer;\r
299 begin\r
300   result := '';\r
301   {$ifdef ipv6}\r
302   if binip.family = AF_INET6 then begin\r
303     result := ip6bintostr(binip.ip6);\r
304   end else\r
305   {$endif}\r
306   if binip.family = AF_INET then begin\r
307     a := htonl(binip.ip);\r
308     result := inttostr(a shr 24)+'.'+inttostr((a shr 16) and $ff)+'.'+inttostr((a shr 8) and $ff)+'.'+inttostr(a and $ff);\r
309   end;\r
310 end;\r
311 \r
312 \r
313 {------------------------------------------------------------------------------}\r
314 \r
315 {$ifdef ipv6}\r
316 \r
317 {\r
318 IPv6 address binary to/from string conversion routines\r
319 written by beware (steendijk at xs4all dot nl)\r
320 \r
321 - implementation does not depend on other ipv6 code such as the tin6_addr type,\r
322   the parameter can also be untyped.\r
323 - it is host endian neutral - binary format is aways network order\r
324 - it supports compression of zeroes\r
325 - it supports ::ffff:192.168.12.34 style addresses\r
326 - they are made to do the Right Thing, more efficient implementations are possible\r
327 }\r
328 \r
329 {fpc has hostaddrtostr6 and strtohostaddr6 but the later isnt implemented yet}\r
330 \r
331 \r
332 function ip6bintostr(const bin:tin6_addr):string;\r
333 {base16 with lowercase output}\r
334 function makehex(w:word):string;\r
335 begin\r
336   result := '';\r
337   if w >= 4096 then result := result + hexchars[w shr 12];\r
338   if w >= 256 then result := result + hexchars[w shr 8 and $f];\r
339   if w >= 16 then result := result + hexchars[w shr 4 and $f];\r
340   result := result + hexchars[w and $f];\r
341 end;\r
342 \r
343 var\r
344   a,b,c,addrlen:integer;\r
345   runbegin,runlength:integer;\r
346   bytes:array[0..15] of byte absolute bin;\r
347   words:array[0..7] of word;\r
348   dwords:array[0..3] of integer absolute words;\r
349 begin\r
350   for a := 0 to 7 do begin\r
351     words[a] := bytes[a shl 1] shl 8 or bytes[a shl 1 or 1];\r
352   end;\r
353   if (dwords[0] = 0) and (dwords[1] = 0) and (words[4] = 0) and (words[5] = $ffff) then begin\r
354     {::ffff:/96 exception: v4 IP}\r
355     addrlen := 6;\r
356   end else begin\r
357     addrlen := 8;\r
358   end;\r
359   {find longest run of zeroes}\r
360   runbegin := 0;\r
361   runlength := 0;\r
362   for a := 0 to addrlen-1 do begin\r
363     if words[a] = 0 then begin\r
364       c := 0;\r
365       for b := a to addrlen-1 do if words[b] = 0 then begin\r
366         inc(c);\r
367       end else break;\r
368       if (c > runlength) then begin\r
369         runlength := c;\r
370         runbegin := a;\r
371       end;\r
372     end;\r
373   end;\r
374   result := '';\r
375   for a := 0 to runbegin-1 do begin\r
376     if (a <> 0) then result := result + ':';\r
377     result := result + makehex(words[a]);\r
378   end;\r
379   if runlength > 0 then result := result + '::';\r
380   c := runbegin+runlength;\r
381   for a := c to addrlen-1 do begin\r
382     if (a > c) then result := result + ':';\r
383     result := result + makehex(words[a]);\r
384   end;\r
385   if addrlen = 6 then begin\r
386     result := result + ':'+inttostr(bytes[12])+'.'+inttostr(bytes[13])+'.'+inttostr(bytes[14])+'.'+inttostr(bytes[15]);\r
387   end;\r
388 end;\r
389 \r
390 function ip6strtobin(const s:string;var bin:tin6_addr):boolean;\r
391 var\r
392   a,b:integer;\r
393   fields:array[0..7] of string;\r
394   fieldcount:integer;\r
395   emptyfield:integer;\r
396   wordcount:integer;\r
397   words:array[0..7] of word;\r
398   bytes:array[0..15] of byte absolute bin;\r
399 begin\r
400   result := false;\r
401   for a := 0 to 7 do fields[a] := '';\r
402   fieldcount := 0;\r
403   for a := 1 to length(s) do begin\r
404     if s[a] = ':' then inc(fieldcount) else fields[fieldcount] := fields[fieldcount] + s[a];\r
405     if fieldcount > 7 then exit;\r
406   end;\r
407   if fieldcount < 2 then exit;\r
408 \r
409   {find the empty field (compressed zeroes), not counting the first and last there may be at most 1}\r
410   emptyfield := -1;\r
411   for a := 1 to fieldcount-1 do begin\r
412     if fields[a] = '' then begin\r
413       if emptyfield = -1 then emptyfield := a else exit;\r
414     end;\r
415   end;\r
416 \r
417   {check if last field is a valid v4 IP}\r
418   a := longip(fields[fieldcount]);\r
419   if (a <> 0) or (fields[fieldcount] = '0.0.0.0') then wordcount := 6 else wordcount := 8;\r
420   {0:1:2:3:4:5:6.6.6.6\r
421    0:1:2:3:4:5:6:7}\r
422   fillchar(words,sizeof(words),0);\r
423   if wordcount = 6 then begin\r
424     if fieldcount > 6 then exit;\r
425     words[6] := a shr 16;\r
426     words[7] := a and $ffff;\r
427   end;\r
428   if emptyfield = -1 then begin\r
429     {no run length: must be an exact number of fields}\r
430     if wordcount = 6 then begin\r
431       if fieldcount <> 6 then exit;\r
432       emptyfield := 5;\r
433     end else if wordcount = 8 then begin\r
434       if fieldcount <> 7 then exit;\r
435       emptyfield := 7;\r
436     end else exit;\r
437   end;\r
438   for a := 0 to emptyfield do begin\r
439     if fields[a] = '' then b := 0 else b := strtointdef('$'+fields[a],-1);\r
440     if (b < 0) or (b > $ffff) then exit;\r
441     words[a] := b;\r
442   end;\r
443   if wordcount = 6 then dec(fieldcount);\r
444   for a := wordcount-1 downto wordcount-(fieldcount-emptyfield) do begin\r
445     b := a+fieldcount-wordcount+1;\r
446     if fields[b] = '' then b := 0 else b := strtointdef('$'+fields[b],-1);\r
447     if (b < 0) or (b > $ffff) then exit;\r
448     words[a] := b;\r
449   end;\r
450   for a := 0 to 7 do begin\r
451     bytes[a shl 1] := words[a] shr 8;\r
452     bytes[a shl 1 or 1] := words[a] and $ff;\r
453   end;\r
454   result := true;\r
455 end;\r
456 {$endif}\r
457 \r
458 function comparebinip(const ip1,ip2:tbinip):boolean;\r
459 begin\r
460   if (ip1.ip <> ip2.ip) then begin\r
461     result := false;\r
462     exit;\r
463   end;\r
464 \r
465   {$ifdef ipv6}\r
466   if ip1.family = AF_INET6 then begin\r
467     if (ip1.ip6.s6_addr32[1] <> ip2.ip6.s6_addr32[1])\r
468     or (ip1.ip6.s6_addr32[2] <> ip2.ip6.s6_addr32[2])\r
469     or (ip1.ip6.s6_addr32[3] <> ip2.ip6.s6_addr32[3]) then begin\r
470       result := false;\r
471       exit;\r
472     end;\r
473   end;\r
474   {$endif}\r
475 \r
476   result := (ip1.family = ip2.family);\r
477 end;\r
478 \r
479 procedure maskbits(var binip:tbinip;bits:integer);\r
480 const\r
481   ipmax={$ifdef ipv6}15{$else}3{$endif};\r
482 type tarr=array[0..ipmax] of byte;\r
483 var\r
484   arr:^tarr;\r
485   a,b:integer;\r
486 begin\r
487   arr := @binip.ip;\r
488   if bits = 0 then b := 0 else b := ((bits-1) div 8)+1;\r
489   for a := b to ipmax do begin\r
490     arr[a] := 0;\r
491   end;\r
492   if (bits and 7 <> 0) then begin\r
493     arr[bits shr 3] := arr[bits div 8] and not ($ff shr (bits and 7))\r
494   end;\r
495 end;\r
496 \r
497 function comparebinipmask;\r
498 begin\r
499   maskbits(ip1,bits);\r
500   maskbits(ip2,bits);\r
501   result := comparebinip(ip1,ip2);\r
502 end;\r
503 \r
504 {converts a binary IP to v4 if it is a v6 IP in the v4 range}\r
505 procedure converttov4(var ip:tbinip);\r
506 begin\r
507   {$ifdef ipv6}\r
508   if ip.family = AF_INET6 then begin\r
509     if (ip.ip6.u6_addr32[0] = 0) and (ip.ip6.u6_addr32[1] = 0) and\r
510     (ip.ip6.u6_addr16[4] = 0) and (ip.ip6.u6_addr16[5] = $ffff) then begin\r
511       ip.family := AF_INET;\r
512       ip.ip := ip.ip6.s6_addr32[3];\r
513     end;\r
514   end;\r
515   {$endif}\r
516 end;\r
517 \r
518 {-----------biniplist stuff--------------------------------------------------}\r
519 \r
520 const\r
521   biniplist_prefix='bipl'#0;\r
522   biniplist_prefixlen=length(biniplist_prefix);\r
523 \r
524 function biniplist_new:tbiniplist;\r
525 begin\r
526   result := biniplist_prefix;\r
527 end;\r
528 \r
529 procedure biniplist_add(var l:tbiniplist;ip:tbinip);\r
530 var\r
531   a:integer;\r
532 begin\r
533   a := biniplist_getcount(l);\r
534   biniplist_setcount(l,a+1);\r
535   biniplist_set(l,a,ip);\r
536 end;\r
537 \r
538 function biniplist_getcount(const l:tbiniplist):integer;\r
539 begin\r
540   result := (length(l)-biniplist_prefixlen) div sizeof(tbinip);\r
541 end;\r
542 \r
543 function biniplist_get(const l:tbiniplist;index:integer):tbinip;\r
544 begin\r
545   if (index >= biniplist_getcount(l)) then begin\r
546     fillchar(result,sizeof(result),0);\r
547     exit;\r
548   end;\r
549   move(l[index*sizeof(tbinip)+1+biniplist_prefixlen],result,sizeof(result));\r
550 end;\r
551 \r
552 procedure biniplist_set(var l:tbiniplist;index:integer;ip:tbinip);\r
553 begin\r
554   uniquestring(l);\r
555   move(ip,l[index*sizeof(tbinip)+1+biniplist_prefixlen],sizeof(ip));\r
556 end;\r
557 \r
558 procedure biniplist_setcount(var l:tbiniplist;newlen:integer);\r
559 begin\r
560   setlength(l,(sizeof(tbinip)*newlen)+biniplist_prefixlen);\r
561 end;\r
562 \r
563 procedure biniplist_free(var l:tbiniplist);\r
564 begin\r
565   l := '';\r
566 end;\r
567 \r
568 procedure biniplist_addlist;\r
569 begin\r
570   l := l + copy(l2,biniplist_prefixlen+1,maxlongint);\r
571 end;\r
572 \r
573 function biniplist_tostr(const l:tbiniplist):string;\r
574 var\r
575   a:integer;\r
576 begin\r
577   result := '(';\r
578   for a := 0 to biniplist_getcount(l)-1 do begin\r
579     if result <> '(' then result := result + ', ';\r
580     result := result + ipbintostr(biniplist_get(l,a));\r
581   end;\r
582   result := result + ')';\r
583 end;\r
584 \r
585 function isbiniplist(const l:tbiniplist):boolean;\r
586 var\r
587   i : integer;\r
588 begin\r
589   for i := 1 to biniplist_prefixlen do begin\r
590     if biniplist_prefix[i] <> l[i] then begin\r
591       result := false;\r
592       exit;\r
593     end;\r
594   end;\r
595   result := true;\r
596 end;\r
597 \r
598 procedure addipsoffamily(var l:tbiniplist;const l2:tbiniplist;family:integer);\r
599 var\r
600   a:integer;\r
601   biniptemp:tbinip;\r
602 begin\r
603   for a := biniplist_getcount(l2)-1 downto 0 do begin\r
604     biniptemp := biniplist_get(l2,a);\r
605     if (biniptemp.family = family) then biniplist_add(l,biniptemp);\r
606   end;\r
607 end;\r
608 \r
609 \r
610 end.\r