Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 1 | /* |
Kristofer Jonsson | b42bc0b | 2023-01-04 17:09:28 +0100 | [diff] [blame] | 2 | * Copyright 2022-2023 Arm Limited and/or its affiliates |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 3 | * |
| 4 | * This program is free software and is provided to you under the terms of the |
| 5 | * GNU General Public License version 2 as published by the Free Software |
| 6 | * Foundation, and any use by you of this program is subject to the terms |
| 7 | * of such GNU licence. |
| 8 | * |
| 9 | * This program is distributed in the hope that it will be useful, |
| 10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 | * GNU General Public License for more details. |
| 13 | * |
| 14 | * You should have received a copy of the GNU General Public License |
| 15 | * along with this program; if not, you can access it online at |
| 16 | * http://www.gnu.org/licenses/gpl-2.0.html. |
| 17 | * |
| 18 | * SPDX-License-Identifier: GPL-2.0-only |
| 19 | */ |
| 20 | |
| 21 | /**************************************************************************** |
| 22 | * Includes |
| 23 | ****************************************************************************/ |
| 24 | |
| 25 | #include "ethosu_network_info.h" |
| 26 | |
| 27 | #include "ethosu_device.h" |
| 28 | #include "ethosu_network.h" |
| 29 | #include "ethosu_mailbox.h" |
| 30 | #include "uapi/ethosu.h" |
| 31 | |
Mikael Olsson | ea209b6 | 2023-08-15 17:01:01 +0200 | [diff] [blame] | 32 | #include <linux/bug.h> |
| 33 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 34 | #define NETWORK_INFO_RESP_TIMEOUT_MS 3000 |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 35 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 36 | static inline int ethosu_network_info_send(struct ethosu_network_info *info, |
| 37 | struct ethosu_mailbox *mailbox) |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 38 | { |
Kristofer Jonsson | f5b98c9 | 2022-03-14 16:09:12 +0100 | [diff] [blame] | 39 | /* Send network info request to firmware */ |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 40 | return ethosu_mailbox_network_info_request(mailbox, |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 41 | &info->msg, |
| 42 | info->net->buf, |
| 43 | info->net->index); |
Kristofer Jonsson | 442fefb | 2022-03-17 17:15:52 +0100 | [diff] [blame] | 44 | } |
| 45 | |
| 46 | static void ethosu_network_info_fail(struct ethosu_mailbox_msg *msg) |
| 47 | { |
| 48 | struct ethosu_network_info *info = |
| 49 | container_of(msg, typeof(*info), msg); |
| 50 | |
| 51 | if (completion_done(&info->done)) |
| 52 | return; |
| 53 | |
| 54 | info->errno = -EFAULT; |
| 55 | complete(&info->done); |
| 56 | } |
| 57 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 58 | int ethosu_network_info_request(struct device *dev, |
| 59 | struct ethosu_mailbox *mailbox, |
| 60 | struct ethosu_network *net, |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 61 | struct ethosu_uapi_network_info *uapi) |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 62 | { |
| 63 | struct ethosu_network_info *info; |
| 64 | int ret; |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 65 | int timeout; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 66 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 67 | info = devm_kzalloc(dev, sizeof(*info), GFP_KERNEL); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 68 | if (!info) |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 69 | return -ENOMEM; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 70 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 71 | info->dev = dev; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 72 | info->net = net; |
| 73 | info->uapi = uapi; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 74 | init_completion(&info->done); |
Kristofer Jonsson | 442fefb | 2022-03-17 17:15:52 +0100 | [diff] [blame] | 75 | info->msg.fail = ethosu_network_info_fail; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 76 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 77 | ret = ethosu_mailbox_register(mailbox, &info->msg); |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 78 | if (ret < 0) |
| 79 | goto kfree; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 80 | |
| 81 | /* Get reference to network */ |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 82 | ethosu_network_get(info->net); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 83 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 84 | ret = ethosu_network_info_send(info, mailbox); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 85 | if (ret) |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 86 | goto deregister; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 87 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 88 | dev_info(dev, |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 89 | "Network info create. info=0x%pK, net=0x%pK, msg.id=0x%x\n", |
| 90 | info, info->net, info->msg.id); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 91 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 92 | /* Unlock the device mutex and wait for completion */ |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 93 | device_unlock(dev); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 94 | timeout = wait_for_completion_timeout(&info->done, |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 95 | msecs_to_jiffies( |
| 96 | NETWORK_INFO_RESP_TIMEOUT_MS)); |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 97 | device_lock(dev); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 98 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 99 | if (0 == timeout) { |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 100 | dev_warn(dev, "Network info timed out. info=0x%pK", |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 101 | info); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 102 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 103 | ret = -ETIME; |
| 104 | goto deregister; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 105 | } |
| 106 | |
Kristofer Jonsson | e3e2643 | 2022-06-28 11:29:42 +0200 | [diff] [blame] | 107 | ret = info->errno; |
| 108 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 109 | deregister: |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 110 | ethosu_mailbox_deregister(mailbox, &info->msg); |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 111 | ethosu_network_put(info->net); |
| 112 | |
| 113 | kfree: |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 114 | dev_info(dev, |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 115 | "Network info destroy. info=0x%pK, msg.id=0x%x\n", |
| 116 | info, info->msg.id); |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 117 | devm_kfree(dev, info); |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 118 | |
| 119 | return ret; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 120 | } |
| 121 | |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 122 | void ethosu_network_info_rsp(struct ethosu_mailbox *mailbox, |
Kristofer Jonsson | d779a08 | 2023-01-04 17:09:47 +0100 | [diff] [blame] | 123 | int msg_id, |
| 124 | struct ethosu_core_msg_network_info_rsp *rsp) |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 125 | { |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 126 | int ret; |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 127 | struct device *dev = mailbox->dev; |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 128 | struct ethosu_mailbox_msg *msg; |
| 129 | struct ethosu_network_info *info; |
| 130 | uint32_t i; |
Mikael Olsson | ea209b6 | 2023-08-15 17:01:01 +0200 | [diff] [blame] | 131 | const size_t rsp_desc_size = sizeof(rsp->desc); |
| 132 | |
| 133 | BUILD_BUG_ON(rsp_desc_size != sizeof(info->uapi->desc)); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 134 | |
Mikael Olsson | 09965b0 | 2023-06-13 12:17:04 +0200 | [diff] [blame] | 135 | msg = ethosu_mailbox_find(mailbox, msg_id, |
| 136 | ETHOSU_CORE_MSG_NETWORK_INFO_REQ); |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 137 | if (IS_ERR(msg)) { |
Kristofer Jonsson | ec47704 | 2023-01-20 13:38:13 +0100 | [diff] [blame] | 138 | dev_warn(dev, |
Mikael Olsson | 09965b0 | 2023-06-13 12:17:04 +0200 | [diff] [blame] | 139 | "Id for network info msg not found. Id=0x%x: %ld\n", |
| 140 | msg_id, PTR_ERR(msg)); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 141 | |
| 142 | return; |
| 143 | } |
| 144 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 145 | info = container_of(msg, typeof(*info), msg); |
| 146 | |
Kristofer Jonsson | 442fefb | 2022-03-17 17:15:52 +0100 | [diff] [blame] | 147 | if (completion_done(&info->done)) |
| 148 | return; |
| 149 | |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 150 | info->errno = 0; |
| 151 | |
| 152 | if (rsp->status != ETHOSU_CORE_STATUS_OK) { |
Mikael Olsson | 42699b0 | 2023-08-16 11:25:12 +0200 | [diff] [blame] | 153 | dev_err(dev, "Failed to get information about the network\n"); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 154 | info->errno = -EBADF; |
| 155 | goto signal_complete; |
| 156 | } |
| 157 | |
| 158 | if (rsp->ifm_count > ETHOSU_FD_MAX || rsp->ofm_count > ETHOSU_FD_MAX) { |
Mikael Olsson | 42699b0 | 2023-08-16 11:25:12 +0200 | [diff] [blame] | 159 | dev_err(dev, |
| 160 | "Invalid number of IFMs/OFMs in network info: IFMs=%u OFMs=%u\n", |
| 161 | rsp->ifm_count, rsp->ofm_count); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 162 | info->errno = -ENFILE; |
| 163 | goto signal_complete; |
| 164 | } |
| 165 | |
Mikael Olsson | ea209b6 | 2023-08-15 17:01:01 +0200 | [diff] [blame] | 166 | if (strnlen(rsp->desc, rsp_desc_size) == rsp_desc_size) { |
| 167 | dev_err(dev, |
| 168 | "Description in network info is not null-terminated\n"); |
| 169 | info->errno = -EMSGSIZE; |
| 170 | goto signal_complete; |
| 171 | } |
| 172 | |
Davide Grohmann | 6ab0b6b | 2022-05-17 16:11:14 +0200 | [diff] [blame] | 173 | ret = strscpy(info->uapi->desc, rsp->desc, sizeof(info->uapi->desc)); |
| 174 | if (ret < 0) { |
Mikael Olsson | ea209b6 | 2023-08-15 17:01:01 +0200 | [diff] [blame] | 175 | dev_err(dev, "Failed to copy network info description\n"); |
Davide Grohmann | 6ab0b6b | 2022-05-17 16:11:14 +0200 | [diff] [blame] | 176 | info->errno = ret; |
| 177 | goto signal_complete; |
| 178 | } |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 179 | |
| 180 | info->uapi->ifm_count = rsp->ifm_count; |
| 181 | for (i = 0; i < rsp->ifm_count; i++) |
| 182 | info->uapi->ifm_size[i] = rsp->ifm_size[i]; |
| 183 | |
| 184 | info->uapi->ofm_count = rsp->ofm_count; |
| 185 | for (i = 0; i < rsp->ofm_count; i++) |
| 186 | info->uapi->ofm_size[i] = rsp->ofm_size[i]; |
| 187 | |
| 188 | signal_complete: |
| 189 | complete(&info->done); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 190 | } |