diff options
author | Jannis Hoffmann <jannis@fehcom.de> | 2024-07-03 15:52:39 +0200 |
---|---|---|
committer | Jannis Hoffmann <jannis@fehcom.de> | 2024-07-03 15:52:39 +0200 |
commit | a6a7d6ce079cabdaf2fa502b2e2cf15e5428ac6f (patch) | |
tree | b88cc7a8457658d67e0321718556ac807f6bccf3 /src/dns.c | |
parent | 00be7622c428f279872f84569f098ce16150f8a8 (diff) |
format files
Diffstat (limited to 'src/dns.c')
-rw-r--r-- | src/dns.c | 138 |
1 files changed, 78 insertions, 60 deletions
@@ -1,20 +1,25 @@ +#include "dns.h" + +#include <arpa/nameser.h> + #include <netdb.h> -#include <string.h> -#include <sys/types.h> #include <netinet/in.h> -#include <arpa/nameser.h> #include <sys/socket.h> -#include "ip.h" -#include "ipalloc.h" -#include "fmt.h" +#include <sys/types.h> + +#include <string.h> + #include "alloc.h" -#include "str.h" -#include "stralloc.h" -#include "dnsresolv.h" -#include "case.h" -#include "dns.h" #include "buffer.h" +#include "case.h" +#include "dnsresolv.h" #include "exit.h" +#include "fmt.h" +#include "ip.h" +#include "str.h" +#include "stralloc.h" + +#include "ipalloc.h" /** @file dns.c @@ -24,86 +29,96 @@ static stralloc glue = {0}; static stralloc ip = {0}; -static int dns_ipplus(ipalloc *ia,stralloc *sa,int pref) +static int dns_ipplus(ipalloc *ia, stralloc *sa, int pref) { struct ip_mx ix; int error = 0; char ip4[4]; char ip6[16]; - int i; + int i; /* Case 1: sa is just IPv4 */ - if (ip4_scanbracket(sa->s,ip4)) { - if (!stralloc_copys(&glue,sa->s)) return DNS_MEM; + if (ip4_scanbracket(sa->s, ip4)) { + if (!stralloc_copys(&glue, sa->s)) return DNS_MEM; if (!stralloc_0(&glue)) return DNS_MEM; if (glue.s[0]) { ix.pref = 0; ix.af = AF_INET; - byte_copy(&ix.addr,4,ip4); // = ip; //cp - if (!ipalloc_append(ia,&ix)) return DNS_MEM; + byte_copy(&ix.addr, 4, ip4); // = ip; //cp + if (!ipalloc_append(ia, &ix)) return DNS_MEM; return 0; } } /* Case 2: sa is just IPv6 */ - if (ip6_scanbracket(sa->s,ip6)) { - if (!stralloc_copys(&glue,sa->s)) return DNS_MEM; + if (ip6_scanbracket(sa->s, ip6)) { + if (!stralloc_copys(&glue, sa->s)) return DNS_MEM; if (!stralloc_0(&glue)) return DNS_MEM; if (glue.s[0]) { ix.pref = 0; ix.af = AF_INET6; - byte_copy(&ix.addr,16,ip6); // = ip; //cp - if (!ipalloc_append(ia,&ix)) return DNS_MEM; + byte_copy(&ix.addr, 16, ip6); // = ip; //cp + if (!ipalloc_append(ia, &ix)) return DNS_MEM; return 0; } } /* Case 3: sa is fqdn and looking for IPv6 */ - if (dns_ip6(&ip,sa) > 0) { + if (dns_ip6(&ip, sa) > 0) { for (i = 0; i + 16 <= ip.len; i += 16) { if (ip6_isv4mapped(ip.s + i)) continue; - ix.af = AF_INET6; - ix.pref = pref; - byte_copy(&ix.addr,16,ip.s + i); // = ip; //cp - str_copy(ix.mxh,sa->s); // mx hostname - if (!ipalloc_append(ia,&ix)) { error = DNS_MEM; break; } - error = 0; + ix.af = AF_INET6; + ix.pref = pref; + byte_copy(&ix.addr, 16, ip.s + i); // = ip; //cp + str_copy(ix.mxh, sa->s); // mx hostname + if (!ipalloc_append(ia, &ix)) { + error = DNS_MEM; + break; } + error = 0; + } } else error = 1; /* Case 4: sa is fqdn and looking for IPv4 */ - if (dns_ip4(&ip,sa) > 0) { + if (dns_ip4(&ip, sa) > 0) { for (i = 0; i + 4 <= ip.len; i += 4) { ix.af = AF_INET; ix.pref = pref; - byte_copy(&ix.addr,4,ip.s + i); // = ip; //cp - str_copy(ix.mxh,sa->s); // mx hostname - if (!ipalloc_append(ia,&ix)) { error = DNS_MEM; break; } + byte_copy(&ix.addr, 4, ip.s + i); // = ip; //cp + str_copy(ix.mxh, sa->s); // mx hostname + if (!ipalloc_append(ia, &ix)) { + error = DNS_MEM; + break; + } error = 0; } } else - error += 2; + error += 2; return error; } -int dns_ipalloc(ipalloc *ia,stralloc *sa) +int dns_ipalloc(ipalloc *ia, stralloc *sa) { - if (!ipalloc_readyplus(ia,0)) return DNS_MEM; + if (!ipalloc_readyplus(ia, 0)) return DNS_MEM; ia->len = 0; - return dns_ipplus(ia,sa,0); + return dns_ipplus(ia, sa, 0); } /* dns_mxip */ -int dns_mxip(ipalloc *ia,stralloc *sa,unsigned long random) { - struct mx { stralloc sa; unsigned short p; } *mx; +int dns_mxip(ipalloc *ia, stralloc *sa, unsigned long random) +{ + struct mx { + stralloc sa; + unsigned short p; + } *mx; struct ip_mx ix; int nummx; int i; @@ -114,23 +129,25 @@ int dns_mxip(ipalloc *ia,stralloc *sa,unsigned long random) { /* Case 1: sa is just IPv4 or IPv6 */ - if (!ipalloc_readyplus(ia,0)) return DNS_MEM; + if (!ipalloc_readyplus(ia, 0)) return DNS_MEM; ia->len = 0; - if (!stralloc_copys(&glue,sa->s)) return DNS_MEM; + if (!stralloc_copys(&glue, sa->s)) return DNS_MEM; if (!stralloc_0(&glue)) return DNS_MEM; if (glue.s[0]) { ix.pref = 0; - if (!glue.s[ip4_scan(glue.s,(char *)&ix.addr.ip4)] || \ - !glue.s[ip4_scanbracket(glue.s,(char *)&ix.addr.ip4)]) { + if (!glue.s[ip4_scan(glue.s, (char *)&ix.addr.ip4)] + || !glue.s[ip4_scanbracket(glue.s, (char *)&ix.addr.ip4)]) + { ix.af = AF_INET; - if (!ipalloc_append(ia,&ix)) return DNS_MEM; + if (!ipalloc_append(ia, &ix)) return DNS_MEM; return 0; } - if (!glue.s[ip6_scan(glue.s,(char *)&ix.addr.ip6)] || \ - !glue.s[ip6_scanbracket(glue.s,(char *)&ix.addr.ip6)]) { + if (!glue.s[ip6_scan(glue.s, (char *)&ix.addr.ip6)] + || !glue.s[ip6_scanbracket(glue.s, (char *)&ix.addr.ip6)]) + { ix.af = AF_INET6; - if (!ipalloc_append(ia,&ix)) return DNS_MEM; + if (!ipalloc_append(ia, &ix)) return DNS_MEM; return 0; } } @@ -140,26 +157,27 @@ int dns_mxip(ipalloc *ia,stralloc *sa,unsigned long random) { DNS_INIT nummx = 0; len = 0; - i = dns_mx(&ip,sa); - mx = (struct mx *) alloc(i * sizeof(struct mx)); + i = dns_mx(&ip, sa); + mx = (struct mx *)alloc(i * sizeof(struct mx)); if (!mx) return DNS_MEM; if (i) { do { - j = str_chr(ip.s + len + 2,'\0'); /* several answers */ + j = str_chr(ip.s + len + 2, '\0'); /* several answers */ mx[nummx].sa.s = 0; - if (!stralloc_copys(&mx[nummx].sa,ip.s + len + 2)) { /* mxhost name */ - alloc_free(mx); return DNS_MEM; + if (!stralloc_copys(&mx[nummx].sa, ip.s + len + 2)) { /* mxhost name */ + alloc_free(mx); + return DNS_MEM; } ip.s[len + 3] = '\0'; - uint16_unpack_big(ip.s + len,&pref); + uint16_unpack_big(ip.s + len, &pref); mx[nummx].p = pref; len += j + 3; ++nummx; } while (len < ip.len); } - if (!nummx) return dns_ipalloc(ia,sa); /* e.g., CNAME -> A */ + if (!nummx) return dns_ipalloc(ia, sa); /* e.g., CNAME -> A */ flagsoft = 0; while (nummx > 0) { @@ -170,17 +188,17 @@ int dns_mxip(ipalloc *ia,stralloc *sa,unsigned long random) { if (mx[j].p < mx[i].p) { i = j; numsame = 1; - } - else if (mx[j].p == mx[i].p) { + } else if (mx[j].p == mx[i].p) { ++numsame; random = random * 69069 + 1; if ((random / 2) < (2147483647 / numsame)) i = j; } } - switch (dns_ipplus(ia,&mx[i].sa,mx[i].p)) { + switch (dns_ipplus(ia, &mx[i].sa, mx[i].p)) { case -1: return DNS_MEM; - case -2: case -3: flagsoft = -5; break; + case -2: + case -3: flagsoft = -5; break; } alloc_free(mx[i].sa.s); @@ -191,11 +209,11 @@ int dns_mxip(ipalloc *ia,stralloc *sa,unsigned long random) { return flagsoft; } -int dns_ip(ipalloc *ia,stralloc *sa) +int dns_ip(ipalloc *ia, stralloc *sa) { - if (!ipalloc_readyplus(ia,0)) return DNS_MEM; + if (!ipalloc_readyplus(ia, 0)) return DNS_MEM; ia->len = 0; - return dns_ipplus(ia,sa,0); + return dns_ipplus(ia, sa, 0); } |