teshsuite/s4u/storage_client_server/storage_client_server.cpp

   1 /* Copyright (c) 2013-2023. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include <simgrid/plugins/file_system.h>
   7 #include <simgrid/s4u.hpp>
   8 #include <xbt/string.hpp>
   9
  10 #include <string>
  11
  12 XBT_LOG_NEW_DEFAULT_CATEGORY(storage, "Messages specific for this simulation");
  13
  14 static void display_disk_properties(const simgrid::s4u::Disk* disk)
  15 {
  16   const std::unordered_map<std::string, std::string>* props = disk->get_properties();
  17   if (not props->empty()) {
  18     XBT_INFO("  Properties of disk: %s", disk->get_cname());
  19
  20     for (auto const& [key, value] : *props) {
  21       XBT_INFO("    %s->%s", key.c_str(), value.c_str());
  22     }
  23   } else {
  24     XBT_INFO("  No property attached.");
  25   }
  26 }
  27
  28 static sg_size_t write_local_file(const std::string& dest, sg_size_t file_size)
  29 {
  30   auto* file        = simgrid::s4u::File::open(dest, nullptr);
  31   sg_size_t written = file->write(file_size);
  32   XBT_INFO("%llu bytes on %llu bytes have been written by %s on /sd1", written, file_size,
  33            simgrid::s4u::Actor::self()->get_cname());
  34   file->close();
  35   return written;
  36 }
  37
  38 static sg_size_t read_local_file(const std::string& src)
  39 {
  40   auto* file          = simgrid::s4u::File::open(src, nullptr);
  41   sg_size_t file_size = file->size();
  42   sg_size_t read      = file->read(file_size);
  43   XBT_INFO("%s has read %llu on %s", simgrid::s4u::Actor::self()->get_cname(), read, src.c_str());
  44   file->close();
  45   return read;
  46 }
  47
  48 // Read src file on local disk and send a put message to remote host (size of message = size of src file)
  49 static void hsm_put(const std::string& remote_host, const std::string& src, const std::string& dest)
  50 {
  51   // Read local src file, and return the size that was actually read
  52   sg_size_t read_size = read_local_file(src);
  53
  54   // Send file
  55   XBT_INFO("%s sends %llu to %s", simgrid::s4u::this_actor::get_cname(), read_size, remote_host.c_str());
  56   auto* payload = new std::string(simgrid::xbt::string_printf("%s %llu", dest.c_str(), read_size));
  57   auto* mailbox = simgrid::s4u::Mailbox::by_name(remote_host);
  58   mailbox->put(payload, read_size);
  59   simgrid::s4u::this_actor::sleep_for(.4);
  60 }
  61
  62 static void display_disk_content(const simgrid::s4u::Disk* disk)
  63 {
  64   XBT_INFO("*** Dump a disk ***");
  65   XBT_INFO("Print the content of the disk: %s", disk->get_cname());
  66   const auto* content = disk->extension<simgrid::s4u::FileSystemDiskExt>()->get_content();
  67   if (not content->empty()) {
  68     for (auto const& [name, size] : *content)
  69       XBT_INFO("  %s size: %llu bytes", name.c_str(), size);
  70   } else {
  71     XBT_INFO("  No content.");
  72   }
  73 }
  74
  75 static void get_set_disk_data(simgrid::s4u::Disk* disk)
  76 {
  77   XBT_INFO("*** GET/SET DATA for disk: %s ***", disk->get_cname());
  78
  79   const std::string* data = disk->get_data<std::string>();
  80   XBT_INFO("Get data: '%s'", data ? data->c_str() : "No User Data");
  81   disk->set_data(new std::string("Some data"));
  82   data = disk->get_data<std::string>();
  83   XBT_INFO("  Set and get data: '%s'", data->c_str());
  84   delete data;
  85 }
  86
  87 static void dump_platform_disks()
  88 {
  89   for (auto const& h : simgrid::s4u::Engine::get_instance()->get_all_hosts())
  90     for (auto const& d : h->get_disks()) {
  91       if (h == d->get_host())
  92         XBT_INFO("%s is attached to %s", d->get_cname(), d->get_host()->get_cname());
  93       d->set_property("other usage", "gpfs");
  94     }
  95 }
  96
  97 static void disk_info(const simgrid::s4u::Host* host)
  98 {
  99   XBT_INFO("*** Disk info on %s ***", host->get_cname());
 100
 101   for (auto const& disk : host->get_disks()) {
 102     const char* mount_name = sg_disk_get_mount_point(disk);
 103     XBT_INFO("  Disk name: %s, mount name: %s", disk->get_cname(), mount_name);
 104
 105     XBT_INFO("    Free size: %llu bytes", sg_disk_get_size_free(disk));
 106     XBT_INFO("    Used size: %llu bytes", sg_disk_get_size_used(disk));
 107
 108     display_disk_properties(disk);
 109     display_disk_content(disk);
 110   }
 111 }
 112
 113 static void client()
 114 {
 115   hsm_put("alice", "/scratch/doc/simgrid/examples/msg/icomms/small_platform.xml", "/tmp/toto.xml");
 116   hsm_put("alice", "/scratch/doc/simgrid/examples/msg/parallel_task/test_ptask_deployment.xml", "/tmp/titi.xml");
 117   hsm_put("alice", "/scratch/doc/simgrid/examples/msg/alias/masterslave_forwarder_with_alias.c", "/tmp/tata.c");
 118
 119   simgrid::s4u::Mailbox* mailbox = simgrid::s4u::Mailbox::by_name("alice");
 120   mailbox->put(new std::string("finalize"), 0);
 121
 122   get_set_disk_data(simgrid::s4u::Host::current()->get_disks().front()); // Disk1
 123 }
 124
 125 static void server()
 126 {
 127   disk_info(simgrid::s4u::this_actor::get_host());
 128   simgrid::s4u::Mailbox* mailbox = simgrid::s4u::Mailbox::by_name(simgrid::s4u::this_actor::get_host()->get_cname());
 129
 130   XBT_INFO("Server waiting for transfers ...");
 131   while (true) {
 132     auto msg = mailbox->get_unique<std::string>();
 133     if (*msg == "finalize") { // Shutdown ...
 134       break;
 135     } else { // Receive file to save
 136       size_t pos              = msg->find(' ');
 137       std::string dest        = msg->substr(0, pos);
 138       sg_size_t size_to_write = std::stoull(msg->substr(pos + 1));
 139       write_local_file(dest, size_to_write);
 140     }
 141   }
 142
 143   disk_info(simgrid::s4u::this_actor::get_host());
 144   dump_platform_disks();
 145 }
 146
 147 int main(int argc, char* argv[])
 148 {
 149   simgrid::s4u::Engine e(&argc, argv);
 150   sg_storage_file_system_init();
 151   xbt_assert(argc == 2, "Usage: %s platform_file\n", argv[0]);
 152   e.load_platform(argv[1]);
 153
 154   simgrid::s4u::Actor::create("server", e.host_by_name("alice"), server);
 155   simgrid::s4u::Actor::create("client", e.host_by_name("bob"), client);
 156
 157   e.run();
 158
 159   XBT_INFO("Simulated time: %g", simgrid::s4u::Engine::get_clock());
 160   return 0;
 161 }