214 lines · c
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 two5 * licenses. You may choose to be licensed under the terms of the GNU6 * General Public License (GPL) Version 2, available from the file7 * COPYING in the main directory of this source tree, or the8 * BSD license below:9 *10 * Redistribution and use in source and binary forms, with or11 * without modification, are permitted provided that the following12 * conditions are met:13 *14 * - Redistributions of source code must retain the above15 * copyright notice, this list of conditions and the following16 * disclaimer.17 *18 * - Redistributions in binary form must reproduce the above19 * copyright notice, this list of conditions and the following20 * disclaimer in the documentation and/or other materials21 * 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 OF25 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND26 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS27 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN28 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN29 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE30 * 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, &addr, &port);73 if (err)74 return 0;75 76 return scnprintf(buf, buf_sz, "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 one82 * starting at roce_next_port. roce_next_port will take on the values83 * 1..409684 */85u16 usnic_transport_rsrv_port(enum usnic_transport_type type, u16 port_num)86{87 if (type == USNIC_TRANSPORT_ROCE_CUSTOM) {88 spin_lock(&roce_bitmap_lock);89 if (!port_num) {90 port_num = bitmap_find_next_zero_area(roce_bitmap,91 ROCE_BITMAP_SZ,92 roce_next_port /* start */,93 1 /* nr */,94 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(&roce_bitmap_lock);100 goto out_fail;101 }102 bitmap_set(roce_bitmap, port_num, 1);103 spin_unlock(&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(&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(roce_bitmap, port_num, 1);135 usnic_dbg("Freeing port %hu for %s\n", port_num,136 usnic_transport_to_str(type));137out_roce_custom:138 spin_unlock(&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(sock_fd, &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(-ENOENT);156 }157 158 usnic_transport_sock_to_str(buf, 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, 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(roce_bitmap, 0, 1);207 return 0;208}209 210void usnic_transport_fini(void)211{212 kfree(roce_bitmap);213}214