xref: /freebsd/crypto/krb5/src/lib/krb5/os/addr.c (revision f1c4c3daccbaf3820f0e2224de53df12fc952fcc)
1 /* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
2 /* lib/krb5/os/addr.c - socket address utilities */
3 /*
4  * Copyright (C) 2024 by the Massachusetts Institute of Technology.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * * Redistributions of source code must retain the above copyright
12  *   notice, this list of conditions and the following disclaimer.
13  *
14  * * Redistributions in binary form must reproduce the above copyright
15  *   notice, this list of conditions and the following disclaimer in
16  *   the documentation and/or other materials provided with the
17  *   distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
22  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
23  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
24  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
25  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
26  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
28  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
30  * OF THE POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 #include "k5-int.h"
34 #include "socket-utils.h"
35 
36 const krb5_address k5_addr_directional_init = {
37     KV5M_ADDRESS, ADDRTYPE_DIRECTIONAL, 4, (uint8_t *)"\x00\x00\x00\x00"
38 };
39 const krb5_address k5_addr_directional_accept = {
40     KV5M_ADDRESS, ADDRTYPE_DIRECTIONAL, 4, (uint8_t *)"\x00\x00\x00\x01"
41 };
42 
43 krb5_error_code
k5_sockaddr_to_address(const struct sockaddr * sa,krb5_boolean local_use,krb5_address * out)44 k5_sockaddr_to_address(const struct sockaddr *sa, krb5_boolean local_use,
45                        krb5_address *out)
46 {
47     if (sa->sa_family == AF_INET) {
48         const struct sockaddr_in *sin = sa2sin(sa);
49         out->addrtype = ADDRTYPE_INET;
50         out->length = sizeof(sin->sin_addr);
51         out->contents = (uint8_t *)&sin->sin_addr;
52     } else if (sa->sa_family == AF_INET6) {
53         const struct sockaddr_in6 *sin6 = sa2sin6(sa);
54         if (IN6_IS_ADDR_V4MAPPED(&sin6->sin6_addr)) {
55             out->addrtype = ADDRTYPE_INET;
56             out->contents = (uint8_t *)&sin6->sin6_addr + 12;
57             out->length = 4;
58         } else {
59             out->addrtype = ADDRTYPE_INET6;
60             out->length = sizeof(sin6->sin6_addr);
61             out->contents = (uint8_t *)&sin6->sin6_addr;
62         }
63 #ifndef _WIN32
64     } else if (sa->sa_family == AF_UNIX && local_use) {
65         const struct sockaddr_un *sun = sa2sun(sa);
66         out->addrtype = ADDRTYPE_UNIXSOCK;
67         out->length = strlen(sun->sun_path);
68         out->contents = (uint8_t *)sun->sun_path;
69 #endif
70     } else {
71         return KRB5_PROG_ATYPE_NOSUPP;
72     }
73     out->magic = KV5M_ADDRESS;
74     return 0;
75 }
76 
77 void
k5_print_addr(const struct sockaddr * sa,char * buf,size_t len)78 k5_print_addr(const struct sockaddr *sa, char *buf, size_t len)
79 {
80     if (sa_is_inet(sa)) {
81         if (getnameinfo(sa, sa_socklen(sa), buf, len, NULL, 0,
82                         NI_NUMERICHOST | NI_NUMERICSERV) != 0)
83             strlcpy(buf, "<unknown>", len);
84 #ifndef _WIN32
85     } else if (sa->sa_family == AF_UNIX) {
86         strlcpy(buf, sa2sun(sa)->sun_path, len);
87 #endif
88     } else {
89         strlcpy(buf, "<unknown>", len);
90     }
91 }
92 
93 void
k5_print_addr_port(const struct sockaddr * sa,char * buf,size_t len)94 k5_print_addr_port(const struct sockaddr *sa, char *buf, size_t len)
95 {
96     char addr[64], port[10];
97 
98     if (sa_is_inet(sa)) {
99         if (getnameinfo(sa, sa_socklen(sa), addr, sizeof(addr), port,
100                         sizeof(port), NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
101             strlcpy(buf, "<unknown>", len);
102         } else if (sa->sa_family == AF_INET) {
103             (void)snprintf(buf, len, "%s:%s", addr, port);
104         } else {
105             (void)snprintf(buf, len, "[%s]:%s", addr, port);
106         }
107     } else {
108         k5_print_addr(sa, buf, len);
109     }
110 }
111