teshsuite/s4u/storage_client_server/storage_client_server.cpp

   1 /* Copyright (c) 2013-2015, 2017. The    SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "simgrid/s4u.hpp"
   8 #include <string>
   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   xbt_dict_t props = storage->properties();
  15   if (xbt_dict_length(props) > 0) {
  16     XBT_INFO("\tProperties of mounted storage: %s", storage->name());
  17
  18     xbt_dict_cursor_t cursor = NULL;
  19     char* key;
  20     char* data;
  21     xbt_dict_foreach (props, cursor, key, data)
  22       XBT_INFO("\t\t'%s' -> '%s'", key, data);
  23   } else {
  24     XBT_INFO("\tNo property attached.");
  25   }
  26 }
  27
  28 static sg_size_t write_local_file(const char* dest, sg_size_t file_size)
  29 {
  30   simgrid::s4u::File* file = new simgrid::s4u::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()->name().c_str());
  34   delete file;
  35   return written;
  36 }
  37
  38 static sg_size_t read_local_file(const char* src)
  39 {
  40   simgrid::s4u::File* file = new simgrid::s4u::File(src, nullptr);
  41   sg_size_t file_size      = file->size();
  42   sg_size_t read           = file->read(file_size);
  43
  44   XBT_INFO("%s has read %llu on %s", simgrid::s4u::Actor::self()->name().c_str(), read, src);
  45   delete file;
  46
  47   return read;
  48 }
  49
  50 // Read src file on local disk and send a put message to remote host (size of message = size of src file)
  51 static void hsm_put(const char* remote_host, const char* src, const char* dest)
  52 {
  53   // Read local src file, and return the size that was actually read
  54   sg_size_t read_size = read_local_file(src);
  55
  56   // Send file
  57   XBT_INFO("%s sends %llu to %s", simgrid::s4u::this_actor::name().c_str(), read_size, remote_host);
  58   char* payload                    = bprintf("%s %llu", dest, read_size);
  59   simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName(remote_host);
  60   simgrid::s4u::this_actor::send(mailbox, payload, static_cast<double>(read_size));
  61   simgrid::s4u::this_actor::sleep_for(.4);
  62 }
  63
  64 static void display_storage_content(simgrid::s4u::Storage* storage)
  65 {
  66   XBT_INFO("Print the content of the storage element: %s", storage->name());
  67   std::map<std::string, sg_size_t*>* content = storage->content();
  68   if (not content->empty()) {
  69     for (auto entry : *content)
  70       XBT_INFO("\t%s size: %llu bytes", entry.first.c_str(), *entry.second);
  71   } else {
  72     XBT_INFO("\tNo content.");
  73   }
  74 }
  75
  76 static void dump_storage_by_name(char* name)
  77 {
  78   XBT_INFO("*** Dump a storage element ***");
  79   simgrid::s4u::Storage& storage = simgrid::s4u::Storage::byName(name);
  80   display_storage_content(&storage);
  81 }
  82
  83 static void get_set_storage_data(const char* storage_name)
  84 {
  85   XBT_INFO("*** GET/SET DATA for storage element: %s ***", storage_name);
  86   simgrid::s4u::Storage& storage = simgrid::s4u::Storage::byName(storage_name);
  87
  88   char* data = static_cast<char*>(storage.userdata());
  89   XBT_INFO("Get data: '%s'", data);
  90   storage.setUserdata(xbt_strdup("Some data"));
  91   data = static_cast<char*>(storage.userdata());
  92   XBT_INFO("\tSet and get data: '%s'", data);
  93   xbt_free(data);
  94 }
  95
  96 static void dump_platform_storages()
  97 {
  98   std::unordered_map<std::string, simgrid::s4u::Storage*>* storages = simgrid::s4u::Storage().allStorages();
  99
 100   for (auto storage : *storages) {
 101     XBT_INFO("Storage %s is attached to %s", storage.first.c_str(), storage.second->host());
 102     storage.second->setProperty("other usage", xbt_strdup("gpfs"));
 103   }
 104   // Expected output in tesh file that's missing for now
 105   //> [  1.207952] (server@alice) Storage Disk3 is attached to carl
 106   //> [  1.207952] (server@alice) Storage Disk4 is attached to denise
 107 }
 108
 109 static void storage_info(simgrid::s4u::Host* host)
 110 {
 111   XBT_INFO("*** Storage info on %s ***", host->cname());
 112   xbt_dict_cursor_t cursor = NULL;
 113   char* mount_name;
 114   char* storage_name;
 115
 116   xbt_dict_t storage_list = host->mountedStoragesAsDict();
 117   xbt_dict_foreach (storage_list, cursor, mount_name, storage_name) {
 118     XBT_INFO("\tStorage name: %s, mount name: %s", storage_name, mount_name);
 119     simgrid::s4u::Storage& storage = simgrid::s4u::Storage::byName(storage_name);
 120
 121     sg_size_t free_size = storage.sizeFree();
 122     sg_size_t used_size = storage.sizeUsed();
 123
 124     XBT_INFO("\t\tFree size: %llu bytes", free_size);
 125     XBT_INFO("\t\tUsed size: %llu bytes", used_size);
 126
 127     display_storage_properties(&storage);
 128     dump_storage_by_name(storage_name);
 129   }
 130   xbt_dict_free(&storage_list);
 131 }
 132
 133 static void client()
 134 {
 135   hsm_put("alice", "/home/doc/simgrid/examples/msg/icomms/small_platform.xml", "c:\\Windows\\toto.cxx");
 136   hsm_put("alice", "/home/doc/simgrid/examples/msg/parallel_task/test_ptask_deployment.xml", "c:\\Windows\\titi.xml");
 137   hsm_put("alice", "/home/doc/simgrid/examples/msg/alias/masterslave_forwarder_with_alias.c", "c:\\Windows\\tata.c");
 138
 139   simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName("alice");
 140   simgrid::s4u::this_actor::send(mailbox, xbt_strdup("finalize"), 0);
 141
 142   get_set_storage_data("Disk1");
 143 }
 144
 145 static void server()
 146 {
 147   storage_info(simgrid::s4u::this_actor::host());
 148   simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName(simgrid::s4u::this_actor::host()->cname());
 149
 150   XBT_INFO("Server waiting for transfers ...");
 151   while (1) {
 152     char* msg = static_cast<char*>(simgrid::s4u::this_actor::recv(mailbox));
 153     if (not strcmp(msg, "finalize")) { // Shutdown ...
 154       xbt_free(msg);
 155       break;
 156     } else { // Receive file to save
 157       char* saveptr;
 158       char* dest              = strtok_r(msg, " ", &saveptr);
 159       sg_size_t size_to_write = std::stoull(strtok_r(nullptr, " ", &saveptr));
 160       write_local_file(dest, size_to_write);
 161       xbt_free(dest);
 162     }
 163   }
 164
 165   storage_info(simgrid::s4u::this_actor::host());
 166   dump_platform_storages();
 167 }
 168
 169 int main(int argc, char* argv[])
 170 {
 171   simgrid::s4u::Engine* e = new simgrid::s4u::Engine(&argc, argv);
 172   xbt_assert(argc == 2, "Usage: %s platform_file\n", argv[0]);
 173   e->loadPlatform(argv[1]);
 174
 175   simgrid::s4u::Actor::createActor("server", simgrid::s4u::Host::by_name("alice"), server);
 176   simgrid::s4u::Actor::createActor("client", simgrid::s4u::Host::by_name("bob"), client);
 177
 178   e->run();
 179
 180   XBT_INFO("Simulated time: %g", e->getClock());
 181
 182   return 0;
 183 }