teshsuite/s4u/storage_client_server/storage_client_server.cpp

   1 /* Copyright (c) 2013-2018. 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_storage_properties(simgrid::s4u::Storage* storage)
  15 {
  16   std::map<std::string, std::string>* props = storage->getProperties();
  17   if (not props->empty()) {
  18     XBT_INFO("\tProperties of mounted storage: %s", storage->get_cname());
  19
  20     for (auto const& elm : *props) {
  21       XBT_INFO("    %s->%s", elm.first.c_str(), elm.second.c_str());
  22     }
  23   } else {
  24     XBT_INFO("\tNo property attached.");
  25   }
  26 }
  27
  28 static sg_size_t write_local_file(const std::string& dest, sg_size_t file_size)
  29 {
  30   simgrid::s4u::File file(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   return written;
  35 }
  36
  37 static sg_size_t read_local_file(const std::string& src)
  38 {
  39   simgrid::s4u::File file(src, nullptr);
  40   sg_size_t file_size = file.size();
  41   sg_size_t read      = file.read(file_size);
  42   XBT_INFO("%s has read %llu on %s", simgrid::s4u::Actor::self()->get_cname(), read, src.c_str());
  43   return read;
  44 }
  45
  46 // Read src file on local disk and send a put message to remote host (size of message = size of src file)
  47 static void hsm_put(const std::string& remote_host, const std::string& src, const std::string& dest)
  48 {
  49   // Read local src file, and return the size that was actually read
  50   sg_size_t read_size = read_local_file(src);
  51
  52   // Send file
  53   XBT_INFO("%s sends %llu to %s", simgrid::s4u::this_actor::get_cname(), read_size, remote_host.c_str());
  54   std::string* payload             = new std::string(simgrid::xbt::string_printf("%s %llu", dest.c_str(), read_size));
  55   simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName(remote_host);
  56   mailbox->put(payload, static_cast<double>(read_size));
  57   simgrid::s4u::this_actor::sleep_for(.4);
  58 }
  59
  60 static void display_storage_content(simgrid::s4u::Storage* storage)
  61 {
  62   XBT_INFO("Print the content of the storage element: %s", storage->get_cname());
  63   std::map<std::string, sg_size_t>* content = storage->extension<simgrid::s4u::FileSystemStorageExt>()->getContent();
  64   if (not content->empty()) {
  65     for (auto const& entry : *content)
  66       XBT_INFO("\t%s size: %llu bytes", entry.first.c_str(), entry.second);
  67   } else {
  68     XBT_INFO("\tNo content.");
  69   }
  70 }
  71
  72 static void dump_storage_by_name(const std::string& name)
  73 {
  74   XBT_INFO("*** Dump a storage element ***");
  75   simgrid::s4u::Storage* storage = simgrid::s4u::Storage::byName(name);
  76   display_storage_content(storage);
  77 }
  78
  79 static void get_set_storage_data(const std::string& storage_name)
  80 {
  81   XBT_INFO("*** GET/SET DATA for storage element: %s ***", storage_name.c_str());
  82   simgrid::s4u::Storage* storage = simgrid::s4u::Storage::byName(storage_name);
  83
  84   std::string* data = static_cast<std::string*>(storage->getUserdata());
  85   XBT_INFO("Get data: '%s'", data ? data->c_str() : "No User Data");
  86   storage->setUserdata(new std::string("Some data"));
  87   data = static_cast<std::string*>(storage->getUserdata());
  88   XBT_INFO("\tSet and get data: '%s'", data->c_str());
  89   delete data;
  90 }
  91
  92 static void dump_platform_storages()
  93 {
  94   std::vector<simgrid::s4u::Storage*> storages = simgrid::s4u::Engine::getInstance()->getAllStorages();
  95
  96   for (auto const& s : storages) {
  97     XBT_INFO("Storage %s is attached to %s", s->get_cname(), s->getHost()->get_cname());
  98     s->setProperty("other usage", "gpfs");
  99   }
 100 }
 101
 102 static void storage_info(simgrid::s4u::Host* host)
 103 {
 104   XBT_INFO("*** Storage info on %s ***", host->get_cname());
 105
 106   for (auto const& elm : host->getMountedStorages()) {
 107     const std::string& mount_name  = elm.first;
 108     simgrid::s4u::Storage* storage = elm.second;
 109     XBT_INFO("\tStorage name: %s, mount name: %s", storage->get_cname(), mount_name.c_str());
 110
 111     XBT_INFO("\t\tFree size: %llu bytes", sg_storage_get_size_free(storage));
 112     XBT_INFO("\t\tUsed size: %llu bytes", sg_storage_get_size_used(storage));
 113
 114     display_storage_properties(storage);
 115     dump_storage_by_name(storage->get_cname());
 116   }
 117 }
 118
 119 static void client()
 120 {
 121   hsm_put("alice", "/home/doc/simgrid/examples/msg/icomms/small_platform.xml", "c:\\Windows\\toto.cxx");
 122   hsm_put("alice", "/home/doc/simgrid/examples/msg/parallel_task/test_ptask_deployment.xml", "c:\\Windows\\titi.xml");
 123   hsm_put("alice", "/home/doc/simgrid/examples/msg/alias/masterslave_forwarder_with_alias.c", "c:\\Windows\\tata.c");
 124
 125   simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName("alice");
 126   mailbox->put(new std::string("finalize"), 0);
 127
 128   get_set_storage_data("Disk1");
 129 }
 130
 131 static void server()
 132 {
 133   storage_info(simgrid::s4u::this_actor::get_host());
 134   simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName(simgrid::s4u::this_actor::get_host()->get_cname());
 135
 136   XBT_INFO("Server waiting for transfers ...");
 137   while (1) {
 138     std::string* msg = static_cast<std::string*>(mailbox->get());
 139     if (*msg == "finalize") { // Shutdown ...
 140       delete msg;
 141       break;
 142     } else { // Receive file to save
 143       size_t pos              = msg->find(' ');
 144       std::string dest        = msg->substr(0, pos);
 145       sg_size_t size_to_write = std::stoull(msg->substr(pos + 1));
 146       write_local_file(dest, size_to_write);
 147       delete msg;
 148     }
 149   }
 150
 151   storage_info(simgrid::s4u::this_actor::get_host());
 152   dump_platform_storages();
 153 }
 154
 155 int main(int argc, char* argv[])
 156 {
 157   simgrid::s4u::Engine e(&argc, argv);
 158   sg_storage_file_system_init();
 159   xbt_assert(argc == 2, "Usage: %s platform_file\n", argv[0]);
 160   e.load_platform(argv[1]);
 161
 162   simgrid::s4u::Actor::create("server", simgrid::s4u::Host::by_name("alice"), server);
 163   simgrid::s4u::Actor::create("client", simgrid::s4u::Host::by_name("bob"), client);
 164
 165   e.run();
 166
 167   XBT_INFO("Simulated time: %g", e.getClock());
 168   return 0;
 169 }