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