// Copyright 2017 The Fuchsia Authors
//
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file or at
// https://opensource.org/licenses/MIT

#include <assert.h>
#include <stdio.h>

#include <lk/init.h>
#include <pdev/driver.h>
#include <pdev/pdev.h>

static const zbi_header_t* driver_zbi = NULL;

extern const struct lk_pdev_init_struct __start_lk_pdev_init[];
extern const struct lk_pdev_init_struct __stop_lk_pdev_init[];

static void pdev_init_driver(uint32_t type, const void* driver_data, uint32_t length, uint level) {
  const struct lk_pdev_init_struct* ptr;
  for (ptr = __start_lk_pdev_init; ptr != __stop_lk_pdev_init; ptr++) {
    if (ptr->type == type && ptr->level == level) {
      ptr->hook(driver_data, length);
      return;
    }
  }
}

static void pdev_run_hooks(uint level) {
  if (!driver_zbi) {
    return;
  }

  const zbi_header_t* item = driver_zbi;
  DEBUG_ASSERT(item->type == ZBI_TYPE_CONTAINER);
  DEBUG_ASSERT(item->extra == ZBI_CONTAINER_MAGIC);

  const uint8_t* start = (uint8_t*)item + sizeof(zbi_header_t);
  const uint8_t* end = start + item->length;

  while (static_cast<size_t>(end - start) > sizeof(zbi_header_t)) {
    item = reinterpret_cast<const zbi_header_t*>(start);
    if (item->type == ZBI_TYPE_KERNEL_DRIVER) {
      // kernel driver type is in boot item extra
      // driver data follows boot item
      pdev_init_driver(item->extra, &item[1], item->length, level);
    }
    start += ZBI_ALIGN((uint32_t)sizeof(zbi_header_t) + item->length);
  }
}

void pdev_init(const zbi_header_t* zbi) {
  ASSERT(zbi);
  driver_zbi = zbi;

  pdev_run_hooks(LK_INIT_LEVEL_PLATFORM_EARLY);
}

static void platform_dev_init(uint level) { pdev_run_hooks(level); }

LK_INIT_HOOK(platform_dev_init, platform_dev_init, LK_INIT_LEVEL_PLATFORM)