1 #pragma ident	"%Z%%M%	%I%	%E% SMI"
2 
3 /*
4  * lib/krb5/os/genaddrs.c
5  *
6  * Copyright 1995 by the Massachusetts Institute of Technology.
7  * All Rights Reserved.
8  *
9  * Export of this software from the United States of America may
10  *   require a specific license from the United States Government.
11  *   It is the responsibility of any person or organization contemplating
12  *   export to obtain such a license before exporting.
13  *
14  * WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
15  * distribute this software and its documentation for any purpose and
16  * without fee is hereby granted, provided that the above copyright
17  * notice appear in all copies and that both that copyright notice and
18  * this permission notice appear in supporting documentation, and that
19  * the name of M.I.T. not be used in advertising or publicity pertaining
20  * to distribution of the software without specific, written prior
21  * permission.  Furthermore if you modify this software you must label
22  * your software as modified software and not distribute it in such a
23  * fashion that it might be confused with the original M.I.T. software.
24  * M.I.T. makes no representations about the suitability of
25  * this software for any purpose.  It is provided "as is" without express
26  * or implied warranty.
27  *
28  *
29  * Take an IP addr & port and generate a full IP address.
30  */
31 
32 #define NEED_SOCKETS
33 #include <k5-int.h>
34 #include "os-proto.h"
35 
36 #if !defined(_WINSOCKAPI_) && !defined(HAVE_MACSOCK_H)
37 #include <netinet/in.h>
38 #endif
39 #include <inet/ip.h>
40 #include <inet/ip6.h>
41 
42 #ifndef GETPEERNAME_ARG2_TYPE
43 #define GETPEERNAME_ARG2_TYPE struct sockaddr
44 #endif
45 #ifndef GETPEERNAME_ARG3_TYPE
46 #define GETPEERNAME_ARG3_TYPE size_t
47 #endif
48 #ifndef GETSOCKNAME_ARG2_TYPE
49 #define GETSOCKNAME_ARG2_TYPE struct sockaddr
50 #endif
51 #ifndef GETSOCKNAME_ARG3_TYPE
52 #define GETSOCKNAME_ARG3_TYPE size_t
53 #endif
54 
55 struct addrpair {
56     krb5_address addr, port;
57 };
58 
59 #define SET(TARG, THING, TYPE) \
60 	((TARG).contents = (krb5_octet *) &(THING),	\
61 	 (TARG).length = sizeof (THING),		\
62 	 (TARG).addrtype = (TYPE))
63 
64 static void *cvtaddr (struct sockaddr_storage *a, struct addrpair *ap)
65 {
66     switch (ss2sa(a)->sa_family) {
67     case AF_INET:
68 	SET (ap->port, ss2sin(a)->sin_port, ADDRTYPE_IPPORT);
69 	SET (ap->addr, ss2sin(a)->sin_addr, ADDRTYPE_INET);
70 	return a;
71 #ifdef KRB5_USE_INET6
72     case AF_INET6:
73 	SET (ap->port, ss2sin6(a)->sin6_port, ADDRTYPE_IPPORT);
74 	if (IN6_IS_ADDR_V4MAPPED (&ss2sin6(a)->sin6_addr)) {
75 	    ap->addr.addrtype = ADDRTYPE_INET;
76 	    ap->addr.contents = (IPV6_ADDR_LEN - IPV4_ADDR_LEN) +
77 		(krb5_octet *) &ss2sin6(a)->sin6_addr;
78 	    ap->addr.length = IPV4_ADDR_LEN;
79 	} else
80 	    SET (ap->addr, ss2sin6(a)->sin6_addr, ADDRTYPE_INET6);
81 	return a;
82 #endif
83     default:
84 	return 0;
85     }
86 }
87 
88 krb5_error_code KRB5_CALLCONV
89 krb5_auth_con_genaddrs(krb5_context context, krb5_auth_context auth_context, int infd, int flags)
90 {
91     krb5_error_code 	  retval;
92     krb5_address	* laddr;
93     krb5_address	* lport;
94     krb5_address	* raddr;
95     krb5_address	* rport;
96     SOCKET		fd = (SOCKET) infd;
97     struct addrpair laddrs, raddrs;
98 
99 #ifdef HAVE_NETINET_IN_H
100     struct sockaddr_storage lsaddr, rsaddr;
101     GETSOCKNAME_ARG3_TYPE ssize;
102 
103     ssize = sizeof(struct sockaddr_storage);
104     if ((flags & KRB5_AUTH_CONTEXT_GENERATE_LOCAL_FULL_ADDR) ||
105 	(flags & KRB5_AUTH_CONTEXT_GENERATE_LOCAL_ADDR)) {
106     	if ((retval = getsockname(fd, (GETSOCKNAME_ARG2_TYPE *) &lsaddr,
107 				  &ssize)))
108 	    return retval;
109 
110 	if (cvtaddr (&lsaddr, &laddrs)) {
111 	    laddr = &laddrs.addr;
112 	    if (flags & KRB5_AUTH_CONTEXT_GENERATE_LOCAL_FULL_ADDR)
113 		lport = &laddrs.port;
114 	    else
115 		lport = 0;
116 	} else
117 	    return KRB5_PROG_ATYPE_NOSUPP;
118     } else {
119 	laddr = NULL;
120 	lport = NULL;
121     }
122 
123     ssize = sizeof(struct sockaddr_storage);
124     if ((flags & KRB5_AUTH_CONTEXT_GENERATE_REMOTE_FULL_ADDR) ||
125 	(flags & KRB5_AUTH_CONTEXT_GENERATE_REMOTE_ADDR)) {
126         if ((retval = getpeername(fd, (GETPEERNAME_ARG2_TYPE *) &rsaddr,
127 				  &ssize)))
128 	    return errno;
129 
130 	if (cvtaddr (&rsaddr, &raddrs)) {
131 	    raddr = &raddrs.addr;
132 	    if (flags & KRB5_AUTH_CONTEXT_GENERATE_REMOTE_FULL_ADDR)
133 		rport = &raddrs.port;
134 	    else
135 		rport = 0;
136 	} else
137 	    return KRB5_PROG_ATYPE_NOSUPP;
138     } else {
139 	raddr = NULL;
140 	rport = NULL;
141     }
142 
143     if (!(retval = krb5_auth_con_setaddrs(context, auth_context, laddr, raddr)))
144     	return (krb5_auth_con_setports(context, auth_context, lport, rport));
145     return retval;
146 #else
147     return KRB5_PROG_ATYPE_NOSUPP;
148 #endif
149 }
150