partition.h

/*
 * Copyright (C) 2018, 2020-2022 Kernkonzept GmbH.
 * Author(s): Sarah Hoffmann <sarah.hoffmann@kernkonzept.com>
 *
 * This file is distributed under the terms of the GNU General Public
 * License, version 2.  Please see the COPYING-GPL-2 file for details.
 */
#pragma once
 
#include <cstring>
#include <string>
#include <cassert>
 
#include <l4/cxx/ref_ptr>
 
#include <l4/l4virtio/virtio_block.h>
 
#include <l4/libblock-device/debug.h>
#include <l4/libblock-device/errand.h>
#include <l4/libblock-device/inout_memory.h>
#include <l4/libblock-device/gpt.h>
 
#include <l4/sys/cache.h>
 
namespace Block_device {
 
struct Partition_info
{
  char           guid[37];  
  std::u16string name;      
  l4_uint64_t    first;     
  l4_uint64_t    last;      
  l4_uint64_t    flags;     
};
 
 
template <typename DEV>
class Partition_reader : public cxx::Ref_obj
{
  enum
  {
    Max_partitions = 1024  
  };
 
public:
  using Device_type = DEV;
 
  Partition_reader(Device_type *dev)
  : _num_partitions(0),
    _dev(dev),
    _header(2, dev, L4Re::Dma_space::Direction::From_device)
  {}
 
  void read(Errand::Callback const &callback)
  {
    _num_partitions = 0;
    _callback = callback;
 
    // preparation: read the first two sectors
    _db = _header.inout_block();
    read_sectors(0, &Partition_reader::get_gpt);
  }
 
  l4_size_t table_size() const
  { return _num_partitions; }
 
  int get_partition(l4_size_t idx, Partition_info *inf) const
  {
    if (idx == 0 || idx > _num_partitions)
      return -L4_ERANGE;
 
    unsigned secsz = _dev->sector_size();
    auto *header = _header.template get<Gpt::Header const>(secsz);
 
    Gpt::Entry *e = _parray.template get<Gpt::Entry>((idx - 1) * header->entry_size);
 
    if (*((l4_uint64_t *) &e->partition_guid) == 0ULL)
      return -L4_ENODEV;
 
    render_guid(e->partition_guid, inf->guid);
 
    auto name =
      std::u16string((char16_t *)e->name, sizeof(e->name) / sizeof(e->name[0]));
    inf->name = name.substr(0, name.find((char16_t) 0));
 
    inf->first = e->first;
    inf->last = e->last;
    inf->flags = e->flags;
 
    auto info = Dbg::info();
    if (info.is_active())
      {
        info.printf("%3zu: %10lld %10lld  %5gMiB [%.37s]\n",
                    idx, e->first, e->last,
                    (e->last - e->first + 1.0) * secsz / (1 << 20),
                    inf->guid);
 
        char buf[37];
        info.printf("   : Type: %s\n", render_guid(e->type_guid, buf));
      }
 
    auto warn = Dbg::warn();
    if (inf->last < inf->first)
      {
        warn.printf(
          "Invalid settings of %3zu. Last lba before first lba. Will ignore.\n",
          idx);
        // Errors in the GPT shall not crash any service -- just ignore the
        // corresponding partition.
        return -L4_ENODEV;
      }
 
    return L4_EOK;
  }
 
private:
  void invoke_callback()
  {
    assert(_callback);
    _callback();
    // Reset the callback to drop potential transitive self-references
    _callback = nullptr;
  }
 
  void get_gpt(int error, l4_size_t)
  {
    _header.unmap();
 
    if (error < 0)
      {
        // can't read from device, we are done
        invoke_callback();
        return;
      }
 
    // prepare reading of the table from disk
    unsigned secsz = _dev->sector_size();
    auto *header = _header.template get<Gpt::Header const>(secsz);
 
    auto info = Dbg::info();
    auto trace = Dbg::trace();
 
    if (strncmp(header->signature, "EFI PART", 8) != 0)
      {
        info.printf("No GUID partition header found.\n");
        invoke_callback();
        return;
      }
 
    // XXX check CRC32 of header
 
    info.printf("GUID partition header found with up to %d partitions.\n",
                header->partition_array_size);
    char buf[37];
    info.printf("GUID: %s\n", render_guid(header->disk_guid, buf));
    trace.printf("Header positions: %llx (Backup: %llx)\n",
                 header->current_lba, header->backup_lba);
    trace.printf("First + last: %llx and %llx\n",
                 header->first_lba, header->last_lba);
    trace.printf("Partition table at %llx\n",
                 header->partition_array_lba);
    trace.printf("Size of a partition entry: %d\n",
                 header->entry_size);
 
    info.printf("GUID partition header found with %d partitions.\n",
                header->partition_array_size);
 
    _num_partitions = cxx::min<l4_uint32_t>(header->partition_array_size,
                                            Max_partitions);
 
    l4_size_t arraysz = _num_partitions * header->entry_size;
    l4_size_t numsec = (arraysz - 1 + secsz) / secsz;
 
    _parray = Inout_memory<Device_type>(numsec, _dev, L4Re::Dma_space::Direction::From_device);
    trace.printf("Reading GPT table @ 0x%p\n", _parray.template get<void>(0));
 
    _db = _parray.inout_block();
    read_sectors(header->partition_array_lba, &Partition_reader::done_gpt);
  }
 
 
  void done_gpt(int error, l4_size_t)
  {
    _parray.unmap();
 
    // XXX check CRC32 of table
 
    if (error < 0)
      _num_partitions = 0;
 
    invoke_callback();
  }
 
  void read_sectors(l4_uint64_t sector,
                    void (Partition_reader::*func)(int, l4_size_t))
  {
    using namespace std::placeholders;
    auto next = std::bind(func, this, _1, _2);
 
    l4_addr_t vstart = (l4_addr_t)_db.virt_addr;
    l4_addr_t vend   = vstart + _db.num_sectors * _dev->sector_size();
    l4_cache_inv_data(vstart, vend);
 
    Errand::poll(10, 10000,
                 [=]()
                   {
                     int ret = _dev->inout_data(
                                 sector, _db,
                                 [next, vstart, vend](int error, l4_size_t size)
                                   {
                                     l4_cache_inv_data(vstart, vend);
                                     next(error, size);
                                   },
                                   L4Re::Dma_space::Direction::From_device);
                     if (ret < 0 && ret != -L4_EBUSY)
                       invoke_callback();
                     return ret != -L4_EBUSY;
                   },
                 [=](bool ret) { if (!ret) invoke_callback(); }
                );
  }
 
  static char const *render_guid(void const *guid_p, char buf[])
  {
    auto *p = static_cast<unsigned char const *>(guid_p);
    snprintf(buf, 37,
             "%02X%02X%02X%02X-%02X%02X-%02X%02X-%02X%02X-%02X%02X%02X%02X%02X%02X",
             p[3],  p[2], p[1],  p[0], p[5],  p[4], p[7],  p[6],
             p[8],  p[9], p[10], p[11], p[12], p[13], p[14], p[15]);
 
    return buf;
  }
 
  l4_size_t _num_partitions;
  Inout_block _db;
  Device_type *_dev;
  Inout_memory<Device_type> _header;
  Inout_memory<Device_type> _parray;
  Errand::Callback _callback;
};
 
 
 
}