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 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 32 | #define NETWORK_INFO_RESP_TIMEOUT_MS 3000 |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 33 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 34 | static inline int ethosu_network_info_send(struct ethosu_network_info *info) |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 35 | { |
Kristofer Jonsson | f5b98c9 | 2022-03-14 16:09:12 +0100 | [diff] [blame] | 36 | /* Send network info request to firmware */ |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 37 | return ethosu_mailbox_network_info_request(&info->edev->mailbox, |
| 38 | &info->msg, |
| 39 | info->net->buf, |
| 40 | info->net->index); |
Kristofer Jonsson | 442fefb | 2022-03-17 17:15:52 +0100 | [diff] [blame] | 41 | } |
| 42 | |
| 43 | static void ethosu_network_info_fail(struct ethosu_mailbox_msg *msg) |
| 44 | { |
| 45 | struct ethosu_network_info *info = |
| 46 | container_of(msg, typeof(*info), msg); |
| 47 | |
| 48 | if (completion_done(&info->done)) |
| 49 | return; |
| 50 | |
| 51 | info->errno = -EFAULT; |
| 52 | complete(&info->done); |
| 53 | } |
| 54 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 55 | int ethosu_network_info_request(struct ethosu_network *net, |
| 56 | struct ethosu_uapi_network_info *uapi) |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 57 | { |
| 58 | struct ethosu_network_info *info; |
| 59 | int ret; |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 60 | int timeout; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 61 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 62 | info = devm_kzalloc(net->edev->dev, sizeof(*info), GFP_KERNEL); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 63 | if (!info) |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 64 | return -ENOMEM; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 65 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 66 | info->edev = net->edev; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 67 | info->net = net; |
| 68 | info->uapi = uapi; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 69 | init_completion(&info->done); |
Kristofer Jonsson | 442fefb | 2022-03-17 17:15:52 +0100 | [diff] [blame] | 70 | info->msg.fail = ethosu_network_info_fail; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 71 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 72 | ret = ethosu_mailbox_register(&info->edev->mailbox, &info->msg); |
| 73 | if (ret < 0) |
| 74 | goto kfree; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 75 | |
| 76 | /* Get reference to network */ |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 77 | ethosu_network_get(info->net); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 78 | |
Kristofer Jonsson | f5b98c9 | 2022-03-14 16:09:12 +0100 | [diff] [blame] | 79 | ret = ethosu_network_info_send(info); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 80 | if (ret) |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 81 | goto deregister; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 82 | |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 83 | dev_info(info->edev->dev, |
| 84 | "Network info create. info=0x%pK, net=0x%pK, msg.id=0x%x\n", |
| 85 | info, info->net, info->msg.id); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 86 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 87 | /* Unlock the device mutex and wait for completion */ |
| 88 | mutex_unlock(&info->edev->mutex); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 89 | timeout = wait_for_completion_timeout(&info->done, |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 90 | msecs_to_jiffies( |
| 91 | NETWORK_INFO_RESP_TIMEOUT_MS)); |
| 92 | mutex_lock(&info->edev->mutex); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 93 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 94 | if (0 == timeout) { |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 95 | dev_warn(info->edev->dev, "Network info timed out. info=0x%pK", |
| 96 | info); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 97 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 98 | ret = -ETIME; |
| 99 | goto deregister; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 100 | } |
| 101 | |
Kristofer Jonsson | e3e2643 | 2022-06-28 11:29:42 +0200 | [diff] [blame] | 102 | ret = info->errno; |
| 103 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 104 | deregister: |
| 105 | ethosu_mailbox_deregister(&info->edev->mailbox, &info->msg); |
| 106 | ethosu_network_put(info->net); |
| 107 | |
| 108 | kfree: |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 109 | dev_info(info->edev->dev, |
| 110 | "Network info destroy. info=0x%pK, msg.id=0x%x\n", |
| 111 | info, info->msg.id); |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 112 | devm_kfree(info->edev->dev, info); |
| 113 | |
| 114 | return ret; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 115 | } |
| 116 | |
| 117 | void ethosu_network_info_rsp(struct ethosu_device *edev, |
Kristofer Jonsson | d779a08 | 2023-01-04 17:09:47 +0100 | [diff] [blame^] | 118 | int msg_id, |
| 119 | struct ethosu_core_msg_network_info_rsp *rsp) |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 120 | { |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 121 | int ret; |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 122 | struct ethosu_mailbox_msg *msg; |
| 123 | struct ethosu_network_info *info; |
| 124 | uint32_t i; |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 125 | |
Kristofer Jonsson | d779a08 | 2023-01-04 17:09:47 +0100 | [diff] [blame^] | 126 | msg = ethosu_mailbox_find(&edev->mailbox, msg_id); |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 127 | if (IS_ERR(msg)) { |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 128 | dev_warn(edev->dev, |
Kristofer Jonsson | 53fd03d | 2022-06-21 16:58:45 +0200 | [diff] [blame] | 129 | "Id for network info msg not found. msg.id=0x%x\n", |
Kristofer Jonsson | d779a08 | 2023-01-04 17:09:47 +0100 | [diff] [blame^] | 130 | msg_id); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 131 | |
| 132 | return; |
| 133 | } |
| 134 | |
Davide Grohmann | 32660f9 | 2022-04-27 16:49:07 +0200 | [diff] [blame] | 135 | info = container_of(msg, typeof(*info), msg); |
| 136 | |
Kristofer Jonsson | 442fefb | 2022-03-17 17:15:52 +0100 | [diff] [blame] | 137 | if (completion_done(&info->done)) |
| 138 | return; |
| 139 | |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 140 | info->errno = 0; |
| 141 | |
| 142 | if (rsp->status != ETHOSU_CORE_STATUS_OK) { |
| 143 | info->errno = -EBADF; |
| 144 | goto signal_complete; |
| 145 | } |
| 146 | |
| 147 | if (rsp->ifm_count > ETHOSU_FD_MAX || rsp->ofm_count > ETHOSU_FD_MAX) { |
| 148 | info->errno = -ENFILE; |
| 149 | goto signal_complete; |
| 150 | } |
| 151 | |
Davide Grohmann | 6ab0b6b | 2022-05-17 16:11:14 +0200 | [diff] [blame] | 152 | ret = strscpy(info->uapi->desc, rsp->desc, sizeof(info->uapi->desc)); |
| 153 | if (ret < 0) { |
| 154 | info->errno = ret; |
| 155 | goto signal_complete; |
| 156 | } |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 157 | |
| 158 | info->uapi->ifm_count = rsp->ifm_count; |
| 159 | for (i = 0; i < rsp->ifm_count; i++) |
| 160 | info->uapi->ifm_size[i] = rsp->ifm_size[i]; |
| 161 | |
| 162 | info->uapi->ofm_count = rsp->ofm_count; |
| 163 | for (i = 0; i < rsp->ofm_count; i++) |
| 164 | info->uapi->ofm_size[i] = rsp->ofm_size[i]; |
| 165 | |
| 166 | signal_complete: |
| 167 | complete(&info->done); |
Kristofer Jonsson | 3c6a260 | 2022-03-10 11:17:29 +0100 | [diff] [blame] | 168 | } |