1/*
2 * Copyright (c) 2013, Cisco Systems, Inc. All rights reserved.
3 *
4 * This software is available to you under a choice of one of two
5 * licenses. You may choose to be licensed under the terms of the GNU
6 * General Public License (GPL) Version 2, available from the file
7 * COPYING in the main directory of this source tree, or the
8 * BSD license below:
9 *
10 * Redistribution and use in source and binary forms, with or
11 * without modification, are permitted provided that the following
12 * conditions are met:
13 *
14 * - Redistributions of source code must retain the above
15 * copyright notice, this list of conditions and the following
16 * disclaimer.
17 *
18 * - Redistributions in binary form must reproduce the above
19 * copyright notice, this list of conditions and the following
20 * disclaimer in the documentation and/or other materials
21 * provided with the distribution.
22 *
23 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
24 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
25 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
26 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
27 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
28 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
29 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
30 * SOFTWARE.
31 *
32 */
33#include <linux/bitmap.h>
34#include <linux/file.h>
35#include <linux/slab.h>
36#include <net/inet_sock.h>
37
38#include "usnic_transport.h"
39#include "usnic_log.h"
40
41/* ROCE */
42static unsigned long *roce_bitmap;
43static u16 roce_next_port = 1;
44#define ROCE_BITMAP_SZ ((1 << (8 /*CHAR_BIT*/ * sizeof(u16)))/8 /*CHAR BIT*/)
45static DEFINE_SPINLOCK(roce_bitmap_lock);
46
47const char *usnic_transport_to_str(enum usnic_transport_type type)
48{
49 switch (type) {
50 case USNIC_TRANSPORT_UNKNOWN:
51 return "Unknown";
52 case USNIC_TRANSPORT_ROCE_CUSTOM:
53 return "roce custom";
54 case USNIC_TRANSPORT_IPV4_UDP:
55 return "IPv4 UDP";
56 case USNIC_TRANSPORT_MAX:
57 return "Max?";
58 default:
59 return "Not known";
60 }
61}
62
63int usnic_transport_sock_to_str(char *buf, int buf_sz,
64 struct socket *sock)
65{
66 int err;
67 uint32_t addr;
68 uint16_t port;
69 int proto;
70
71 memset(buf, 0, buf_sz);
72 err = usnic_transport_sock_get_addr(sock, proto: &proto, addr: &addr, port: &port);
73 if (err)
74 return 0;
75
76 return scnprintf(buf, size: buf_sz, fmt: "Proto:%u Addr:%pI4h Port:%hu",
77 proto, &addr, port);
78}
79
80/*
81 * reserve a port number. if "0" specified, we will try to pick one
82 * starting at roce_next_port. roce_next_port will take on the values
83 * 1..4096
84 */
85u16 usnic_transport_rsrv_port(enum usnic_transport_type type, u16 port_num)
86{
87 if (type == USNIC_TRANSPORT_ROCE_CUSTOM) {
88 spin_lock(lock: &roce_bitmap_lock);
89 if (!port_num) {
90 port_num = bitmap_find_next_zero_area(map: roce_bitmap,
91 ROCE_BITMAP_SZ,
92 start: roce_next_port /* start */,
93 nr: 1 /* nr */,
94 align_mask: 0 /* align */);
95 roce_next_port = (port_num & 4095) + 1;
96 } else if (test_bit(port_num, roce_bitmap)) {
97 usnic_err("Failed to allocate port for %s\n",
98 usnic_transport_to_str(type));
99 spin_unlock(lock: &roce_bitmap_lock);
100 goto out_fail;
101 }
102 bitmap_set(map: roce_bitmap, start: port_num, nbits: 1);
103 spin_unlock(lock: &roce_bitmap_lock);
104 } else {
105 usnic_err("Failed to allocate port - transport %s unsupported\n",
106 usnic_transport_to_str(type));
107 goto out_fail;
108 }
109
110 usnic_dbg("Allocating port %hu for %s\n", port_num,
111 usnic_transport_to_str(type));
112 return port_num;
113
114out_fail:
115 return 0;
116}
117
118void usnic_transport_unrsrv_port(enum usnic_transport_type type, u16 port_num)
119{
120 if (type == USNIC_TRANSPORT_ROCE_CUSTOM) {
121 spin_lock(lock: &roce_bitmap_lock);
122 if (!port_num) {
123 usnic_err("Unreserved invalid port num 0 for %s\n",
124 usnic_transport_to_str(type));
125 goto out_roce_custom;
126 }
127
128 if (!test_bit(port_num, roce_bitmap)) {
129 usnic_err("Unreserving invalid %hu for %s\n",
130 port_num,
131 usnic_transport_to_str(type));
132 goto out_roce_custom;
133 }
134 bitmap_clear(map: roce_bitmap, start: port_num, nbits: 1);
135 usnic_dbg("Freeing port %hu for %s\n", port_num,
136 usnic_transport_to_str(type));
137out_roce_custom:
138 spin_unlock(lock: &roce_bitmap_lock);
139 } else {
140 usnic_err("Freeing invalid port %hu for %d\n", port_num, type);
141 }
142}
143
144struct socket *usnic_transport_get_socket(int sock_fd)
145{
146 struct socket *sock;
147 int err;
148 char buf[25];
149
150 /* sockfd_lookup will internally do a fget */
151 sock = sockfd_lookup(fd: sock_fd, err: &err);
152 if (!sock) {
153 usnic_err("Unable to lookup socket for fd %d with err %d\n",
154 sock_fd, err);
155 return ERR_PTR(error: -ENOENT);
156 }
157
158 usnic_transport_sock_to_str(buf, buf_sz: sizeof(buf), sock);
159 usnic_dbg("Get sock %s\n", buf);
160
161 return sock;
162}
163
164void usnic_transport_put_socket(struct socket *sock)
165{
166 char buf[100];
167
168 usnic_transport_sock_to_str(buf, buf_sz: sizeof(buf), sock);
169 usnic_dbg("Put sock %s\n", buf);
170 sockfd_put(sock);
171}
172
173int usnic_transport_sock_get_addr(struct socket *sock, int *proto,
174 uint32_t *addr, uint16_t *port)
175{
176 int err;
177 struct sockaddr_in sock_addr;
178
179 err = sock->ops->getname(sock,
180 (struct sockaddr *)&sock_addr,
181 0);
182 if (err < 0)
183 return err;
184
185 if (sock_addr.sin_family != AF_INET)
186 return -EINVAL;
187
188 if (proto)
189 *proto = sock->sk->sk_protocol;
190 if (port)
191 *port = ntohs(((struct sockaddr_in *)&sock_addr)->sin_port);
192 if (addr)
193 *addr = ntohl(((struct sockaddr_in *)
194 &sock_addr)->sin_addr.s_addr);
195
196 return 0;
197}
198
199int usnic_transport_init(void)
200{
201 roce_bitmap = kzalloc(ROCE_BITMAP_SZ, GFP_KERNEL);
202 if (!roce_bitmap)
203 return -ENOMEM;
204
205 /* Do not ever allocate bit 0, hence set it here */
206 bitmap_set(map: roce_bitmap, start: 0, nbits: 1);
207 return 0;
208}
209
210void usnic_transport_fini(void)
211{
212 kfree(objp: roce_bitmap);
213}
214

source code of linux/drivers/infiniband/hw/usnic/usnic_transport.c