teshsuite/s4u/storage_client_server/storage_client_server.cpp

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