// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <ddk/binding.h>
#include <ddk/device.h>
#include <ddk/driver.h>
#include <ddk/protocol/test.h>
#include <ddktl/device.h>
#include <lib/zx/socket.h>

#include <limits.h>
#include <memory>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unittest/unittest.h>

extern test_case_element* test_case_ddktl_device;
extern test_case_element* test_case_ddktl_ethernet_device;

namespace {

void ddktl_test_output_func(const char* line, int len, void* arg) {
  zx_handle_t h = *static_cast<zx_handle_t*>(arg);
  zx::unowned_socket s(h);
  // len is not actually the number of bytes to output
  s->write(0u, line, strlen(line), nullptr);
}

static void inline update_test_report(bool success, test_report_t* report) {
  report->n_tests++;
  if (success) {
    report->n_success++;
  } else {
    report->n_failed++;
  }
}

zx_status_t ddktl_test_func(void* cookie, test_report_t* report) {
  auto dev = static_cast<zx_device_t*>(cookie);

  test_protocol_t proto;
  auto status = device_get_protocol(dev, ZX_PROTOCOL_TEST, reinterpret_cast<void*>(&proto));
  if (status != ZX_OK) {
    return status;
  }

  zx_handle_t output;
  proto.ops->get_output_socket(proto.ctx, &output);
  if (output != ZX_HANDLE_INVALID) {
    unittest_set_output_function(ddktl_test_output_func, &output);
  }

  memset(report, 0, sizeof(*report));
  update_test_report(unittest_run_one_test(test_case_ddktl_device, TEST_ALL), report);
  update_test_report(unittest_run_one_test(test_case_ddktl_ethernet_device, TEST_ALL), report);
  unittest_restore_output_function();
  zx_handle_close(output);
  return report->n_failed == 0 ? ZX_OK : ZX_ERR_INTERNAL;
}

class ChildDevice;
using ChildDeviceType = ddk::Device<ChildDevice, ddk::Unbindable>;

class ChildDevice : public ChildDeviceType {
 public:
  ChildDevice(zx_device_t* parent) : ChildDeviceType(parent) {}
  void DdkUnbind() { DdkRemove(); }
  void DdkRelease() { delete this; }
};

}  // namespace

extern "C" zx_status_t ddktl_test_bind(void* ctx, zx_device_t* parent) {
  auto device = std::make_unique<ChildDevice>(parent);
  zx_status_t status = device->DdkAdd("child", DEVICE_ADD_NON_BINDABLE);
  if (status != ZX_OK) {
    return status;
  }
  // devmgr now owns this
  __UNUSED auto ptr = device.release();

  test_protocol_t proto;
  status = device_get_protocol(parent, ZX_PROTOCOL_TEST, &proto);
  if (status != ZX_OK) {
    return status;
  }

  const test_func_t test = {ddktl_test_func, parent};
  proto.ops->set_test_func(proto.ctx, &test);
  return ZX_OK;
}