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