Logo AND Algorithmique Numérique Distribuée

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