Network info
Add UAPI and core message that allows user space space to fetch
information about network models built into the firmware.
Change-Id: Ic92529bce3edd0a5499e691a566bd065da2a72ad
diff --git a/kernel/ethosu_network_info.c b/kernel/ethosu_network_info.c
new file mode 100644
index 0000000..2b1f841
--- /dev/null
+++ b/kernel/ethosu_network_info.c
@@ -0,0 +1,179 @@
+/*
+ * Copyright (c) 2022 ARM Limited.
+ *
+ * This program is free software and is provided to you under the terms of the
+ * GNU General Public License version 2 as published by the Free Software
+ * Foundation, and any use by you of this program is subject to the terms
+ * of such GNU licence.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, you can access it online at
+ * http://www.gnu.org/licenses/gpl-2.0.html.
+ *
+ * SPDX-License-Identifier: GPL-2.0-only
+ */
+
+/****************************************************************************
+ * Includes
+ ****************************************************************************/
+
+#include "ethosu_network_info.h"
+
+#include "ethosu_device.h"
+#include "ethosu_network.h"
+#include "ethosu_mailbox.h"
+#include "uapi/ethosu.h"
+
+/****************************************************************************
+ * Functions
+ ****************************************************************************/
+
+static void ethosu_network_info_destroy(struct kref *kref)
+{
+ struct ethosu_network_info *info =
+ container_of(kref, struct ethosu_network_info, kref);
+
+ dev_info(info->edev->dev, "Network info destroy. handle=0x%pK\n", info);
+
+ list_del(&info->list);
+
+ ethosu_network_put(info->net);
+
+ devm_kfree(info->edev->dev, info);
+}
+
+struct ethosu_network_info *ethosu_network_info_create(
+ struct ethosu_device *edev,
+ struct ethosu_network *net,
+ struct ethosu_uapi_network_info *uapi)
+{
+ struct ethosu_network_info *info;
+ int ret;
+
+ info = devm_kzalloc(edev->dev, sizeof(*info), GFP_KERNEL);
+ if (!info)
+ return ERR_PTR(-ENOMEM);
+
+ info->edev = edev;
+ info->net = net;
+ info->uapi = uapi;
+ kref_init(&info->kref);
+ init_completion(&info->done);
+
+ /* Insert network info to network info list */
+ list_add(&info->list, &edev->network_info_list);
+
+ /* Get reference to network */
+ ethosu_network_get(net);
+
+ /* Send network info request to firmware */
+ ret = ethosu_mailbox_network_info_request(&info->edev->mailbox,
+ info,
+ info->net->buf,
+ info->net->index);
+ if (ret)
+ goto put_info;
+
+ /* Increase reference count for the pending network info response */
+ ethosu_network_info_get(info);
+
+ dev_info(edev->dev, "Network info create. handle=%p\n", info);
+
+ return info;
+
+put_info:
+ ethosu_network_info_put(info);
+
+ return ERR_PTR(ret);
+}
+
+static int ethosu_network_info_find(struct ethosu_network_info *info,
+ struct list_head *network_info_list)
+{
+ struct ethosu_network_info *cur;
+
+ list_for_each_entry(cur, network_info_list, list) {
+ if (cur == info)
+ return 0;
+ }
+
+ return -EINVAL;
+}
+
+void ethosu_network_info_get(struct ethosu_network_info *info)
+{
+ kref_get(&info->kref);
+}
+
+void ethosu_network_info_put(struct ethosu_network_info *info)
+{
+ kref_put(&info->kref, ethosu_network_info_destroy);
+}
+
+int ethosu_network_info_wait(struct ethosu_network_info *info,
+ int timeout_ms)
+{
+ int timeout;
+
+ timeout = wait_for_completion_timeout(&info->done,
+ msecs_to_jiffies(timeout_ms));
+
+ if (!timeout) {
+ dev_warn(info->edev->dev,
+ "Network info timed out.");
+
+ return -ETIME;
+ }
+
+ return info->errno;
+}
+
+void ethosu_network_info_rsp(struct ethosu_device *edev,
+ struct ethosu_core_network_info_rsp *rsp)
+{
+ struct ethosu_network_info *info =
+ (struct ethosu_network_info *)rsp->user_arg;
+ uint32_t i;
+ int ret;
+
+ ret = ethosu_network_info_find(info, &edev->network_info_list);
+ if (0 != ret) {
+ dev_warn(edev->dev,
+ "Handle not found in network info list. handle=0x%p\n",
+ info);
+
+ return;
+ }
+
+ info->errno = 0;
+
+ if (rsp->status != ETHOSU_CORE_STATUS_OK) {
+ info->errno = -EBADF;
+ goto signal_complete;
+ }
+
+ if (rsp->ifm_count > ETHOSU_FD_MAX || rsp->ofm_count > ETHOSU_FD_MAX) {
+ info->errno = -ENFILE;
+ goto signal_complete;
+ }
+
+ strncpy(info->uapi->desc, rsp->desc, sizeof(info->uapi->desc));
+
+ info->uapi->ifm_count = rsp->ifm_count;
+ for (i = 0; i < rsp->ifm_count; i++)
+ info->uapi->ifm_size[i] = rsp->ifm_size[i];
+
+ info->uapi->ofm_count = rsp->ofm_count;
+ for (i = 0; i < rsp->ofm_count; i++)
+ info->uapi->ofm_size[i] = rsp->ofm_size[i];
+
+signal_complete:
+ complete(&info->done);
+
+ ethosu_network_info_put(info);
+}