Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
[sonar] Constify pointer and reference local variables in teshsuite/.
[simgrid.git] / teshsuite / s4u / storage_client_server / storage_client_server.cpp
1 /* Copyright (c) 2013-2019. 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_disk_properties(simgrid::s4u::Disk* disk)
15 {
16   const std::unordered_map<std::string, std::string>* props = disk->get_properties();
17   if (not props->empty()) {
18     XBT_INFO("  Properties of disk: %s", disk->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("  No 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::Mailbox* mailbox   = simgrid::s4u::Mailbox::by_name(remote_host);
56   mailbox->put(payload, read_size);
57   simgrid::s4u::this_actor::sleep_for(.4);
58 }
59
60 static void display_disk_content(simgrid::s4u::Disk* disk)
61 {
62   XBT_INFO("*** Dump a disk ***");
63   XBT_INFO("Print the content of the disk: %s", disk->get_cname());
64   const std::map<std::string, sg_size_t>* content = disk->extension<simgrid::s4u::FileSystemDiskExt>()->get_content();
65   if (not content->empty()) {
66     for (auto const& entry : *content)
67       XBT_INFO("  %s size: %llu bytes", entry.first.c_str(), entry.second);
68   } else {
69     XBT_INFO("  No content.");
70   }
71 }
72
73 static void get_set_disk_data(simgrid::s4u::Disk* disk)
74 {
75   XBT_INFO("*** GET/SET DATA for disk: %s ***", disk->get_cname());
76
77   const std::string* data = static_cast<std::string*>(disk->get_data());
78   XBT_INFO("Get data: '%s'", data ? data->c_str() : "No User Data");
79   disk->set_data(new std::string("Some data"));
80   data = static_cast<std::string*>(disk->get_data());
81   XBT_INFO("  Set and get data: '%s'", data->c_str());
82   delete data;
83 }
84
85 static void dump_platform_disks()
86 {
87   for (auto const& h : simgrid::s4u::Engine::get_instance()->get_all_hosts())
88     for (auto const& d : h->get_disks()) {
89       if (h == d->get_host())
90         XBT_INFO("%s is attached to %s", d->get_cname(), d->get_host()->get_cname());
91       d->set_property("other usage", "gpfs");
92     }
93 }
94
95 static void disk_info(simgrid::s4u::Host* host)
96 {
97   XBT_INFO("*** Disk info on %s ***", host->get_cname());
98
99   for (auto const& disk : host->get_disks()) {
100     const char* mount_name = sg_disk_get_mount_point(disk);
101     XBT_INFO("  Disk name: %s, mount name: %s", disk->get_cname(), mount_name);
102
103     XBT_INFO("    Free size: %llu bytes", sg_disk_get_size_free(disk));
104     XBT_INFO("    Used size: %llu bytes", sg_disk_get_size_used(disk));
105
106     display_disk_properties(disk);
107     display_disk_content(disk);
108   }
109 }
110
111 static void client()
112 {
113   hsm_put("alice", "/scratch/doc/simgrid/examples/msg/icomms/small_platform.xml", "/tmp/toto.xml");
114   hsm_put("alice", "/scratch/doc/simgrid/examples/msg/parallel_task/test_ptask_deployment.xml", "/tmp/titi.xml");
115   hsm_put("alice", "/scratch/doc/simgrid/examples/msg/alias/masterslave_forwarder_with_alias.c", "/tmp/tata.c");
116
117   simgrid::s4u::Mailbox* mailbox = simgrid::s4u::Mailbox::by_name("alice");
118   mailbox->put(new std::string("finalize"), 0);
119
120   get_set_disk_data(simgrid::s4u::Host::current()->get_disks().front()); // Disk1
121 }
122
123 static void server()
124 {
125   disk_info(simgrid::s4u::this_actor::get_host());
126   simgrid::s4u::Mailbox* mailbox = simgrid::s4u::Mailbox::by_name(simgrid::s4u::this_actor::get_host()->get_cname());
127
128   XBT_INFO("Server waiting for transfers ...");
129   while (1) {
130     const std::string* msg = static_cast<std::string*>(mailbox->get());
131     if (*msg == "finalize") { // Shutdown ...
132       delete msg;
133       break;
134     } else { // Receive file to save
135       size_t pos              = msg->find(' ');
136       std::string dest        = msg->substr(0, pos);
137       sg_size_t size_to_write = std::stoull(msg->substr(pos + 1));
138       write_local_file(dest, size_to_write);
139       delete msg;
140     }
141   }
142
143   disk_info(simgrid::s4u::this_actor::get_host());
144   dump_platform_disks();
145 }
146
147 int main(int argc, char* argv[])
148 {
149   simgrid::s4u::Engine e(&argc, argv);
150   sg_storage_file_system_init();
151   xbt_assert(argc == 2, "Usage: %s platform_file\n", argv[0]);
152   e.load_platform(argv[1]);
153
154   simgrid::s4u::Actor::create("server", simgrid::s4u::Host::by_name("alice"), server);
155   simgrid::s4u::Actor::create("client", simgrid::s4u::Host::by_name("bob"), client);
156
157   e.run();
158
159   XBT_INFO("Simulated time: %g", e.get_clock());
160   return 0;
161 }