1 /* Copyright (c) 2013-2022. The SimGrid Team. All rights reserved. */
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. */
6 #include <simgrid/plugins/file_system.h>
7 #include <simgrid/s4u.hpp>
8 #include <xbt/string.hpp>
12 XBT_LOG_NEW_DEFAULT_CATEGORY(storage, "Messages specific for this simulation");
14 static void display_disk_properties(const simgrid::s4u::Disk* disk)
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());
20 for (auto const& elm : *props) {
21 XBT_INFO(" %s->%s", elm.first.c_str(), elm.second.c_str());
24 XBT_INFO(" No property attached.");
28 static sg_size_t write_local_file(const std::string& dest, sg_size_t file_size)
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());
37 static sg_size_t read_local_file(const std::string& src)
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());
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)
49 // Read local src file, and return the size that was actually read
50 sg_size_t read_size = read_local_file(src);
53 XBT_INFO("%s sends %llu to %s", simgrid::s4u::this_actor::get_cname(), read_size, remote_host.c_str());
54 auto* payload = new std::string(simgrid::xbt::string_printf("%s %llu", dest.c_str(), read_size));
55 auto* mailbox = simgrid::s4u::Mailbox::by_name(remote_host);
56 mailbox->put(payload, read_size);
57 simgrid::s4u::this_actor::sleep_for(.4);
60 static void display_disk_content(const simgrid::s4u::Disk* disk)
62 XBT_INFO("*** Dump a disk ***");
63 XBT_INFO("Print the content of the disk: %s", disk->get_cname());
64 const auto* 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);
69 XBT_INFO(" No content.");
73 static void get_set_disk_data(simgrid::s4u::Disk* disk)
75 XBT_INFO("*** GET/SET DATA for disk: %s ***", disk->get_cname());
77 const std::string* data = disk->get_data<std::string>();
78 XBT_INFO("Get data: '%s'", data ? data->c_str() : "No User Data");
79 disk->set_data(new std::string("Some data"));
80 data = disk->get_data<std::string>();
81 XBT_INFO(" Set and get data: '%s'", data->c_str());
85 static void dump_platform_disks()
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");
95 static void disk_info(const simgrid::s4u::Host* host)
97 XBT_INFO("*** Disk info on %s ***", host->get_cname());
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);
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));
106 display_disk_properties(disk);
107 display_disk_content(disk);
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");
117 simgrid::s4u::Mailbox* mailbox = simgrid::s4u::Mailbox::by_name("alice");
118 mailbox->put(new std::string("finalize"), 0);
120 get_set_disk_data(simgrid::s4u::Host::current()->get_disks().front()); // Disk1
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());
128 XBT_INFO("Server waiting for transfers ...");
130 auto msg = mailbox->get_unique<std::string>();
131 if (*msg == "finalize") { // Shutdown ...
133 } else { // Receive file to save
134 size_t pos = msg->find(' ');
135 std::string dest = msg->substr(0, pos);
136 sg_size_t size_to_write = std::stoull(msg->substr(pos + 1));
137 write_local_file(dest, size_to_write);
141 disk_info(simgrid::s4u::this_actor::get_host());
142 dump_platform_disks();
145 int main(int argc, char* argv[])
147 simgrid::s4u::Engine e(&argc, argv);
148 sg_storage_file_system_init();
149 xbt_assert(argc == 2, "Usage: %s platform_file\n", argv[0]);
150 e.load_platform(argv[1]);
152 simgrid::s4u::Actor::create("server", e.host_by_name("alice"), server);
153 simgrid::s4u::Actor::create("client", e.host_by_name("bob"), client);
157 XBT_INFO("Simulated time: %g", simgrid::s4u::Engine::get_clock());