Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
3612aad8735400670be8d85426690d3845f5200d
[simgrid.git] / teshsuite / s4u / monkey-semaphore / monkey-semaphore.cpp
1 /* Copyright (c) 2006-2022. 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 // This example implements a simple producer/consumer schema, passing a bunch of items from one to the other,
7 // hopefully implemented in a way that resists resource failures.
8
9 #include <simgrid/s4u.hpp>
10 #include <xbt/config.hpp>
11
12 namespace sg4 = simgrid::s4u;
13
14 XBT_LOG_NEW_DEFAULT_CATEGORY(sem_monkey, "Simple test of the semaphore");
15
16 static simgrid::config::Flag<int> cfg_item_count{"item-count", "Amount of items that must be exchanged to succeed", 2};
17 static simgrid::config::Flag<double> cfg_deadline{"deadline", "When to fail the simulation (infinite loop detection)",
18                                                   120};
19
20 struct SharedBuffer {
21   int value;                                               /* Where the data is exchanged */
22   sg4::SemaphorePtr sem_empty = sg4::Semaphore::create(1); /* indicates whether the buffer is empty */
23   sg4::SemaphorePtr sem_full  = sg4::Semaphore::create(0); /* indicates whether the buffer is full */
24 };
25
26 // A stack to keep track of semaphores.  When destroyed, semaphores remaining on stack are automatically released.
27 class SemStack {
28   std::vector<sg4::Semaphore*> to_release;
29
30 public:
31   SemStack()                = default;
32   SemStack(const SemStack&) = delete;
33   SemStack& operator=(const SemStack&) = delete;
34   ~SemStack()
35   {
36     for (auto* sem : to_release) {
37       sem->release();
38       XBT_INFO("Released a semaphore on exit. It's now %d", sem->get_capacity());
39     }
40   }
41   void push(const sg4::SemaphorePtr& sem) { to_release.push_back(sem.get()); }
42   void pop() { to_release.pop_back(); }
43 };
44
45 static void producer(SharedBuffer& buf)
46 {
47   static bool inited = false;
48   static int todo    = cfg_item_count; // remaining amount of items to exchange
49   SemStack to_release;
50   XBT_INFO("Producer %s", inited ? "rebooting" : "booting");
51
52   if (not inited) {
53     sg4::this_actor::on_exit(
54         [](bool forcefully) { XBT_INFO("Producer dying %s.", forcefully ? "forcefully" : "peacefully"); });
55     inited = true;
56   }
57
58   while (todo > 0) {
59     xbt_assert(sg4::Engine::get_clock() < cfg_deadline,
60                "Failed to exchange all tasks in less than %d seconds. Is this an infinite loop?", (int)cfg_deadline);
61
62     sg4::this_actor::sleep_for(1); // Give a chance to the monkey to kill this actor at this point
63
64     while (buf.sem_empty->acquire_timeout(10))
65       XBT_INFO("Timeouted");
66     to_release.push(buf.sem_empty);
67     XBT_INFO("sem_empty acquired");
68
69     sg4::this_actor::sleep_for(1); // Give a chance to the monkey to kill this actor at this point
70
71     XBT_INFO("Pushing item %d", todo - 1);
72     buf.value = todo - 1;
73     buf.sem_full->release();
74     to_release.pop();
75     XBT_INFO("sem_empty removed from to_release");
76     todo--;
77   }
78 }
79
80 static void consumer(const SharedBuffer& buf)
81 {
82   SemStack to_release;
83
84   static bool inited = false;
85   XBT_INFO("Consumer %s", inited ? "rebooting" : "booting");
86   if (not inited) {
87     sg4::this_actor::on_exit(
88         [](bool forcefully) { XBT_INFO("Consumer dying %s.", forcefully ? "forcefully" : "peacefully"); });
89     inited = true;
90   }
91
92   int item;
93   do {
94     xbt_assert(sg4::Engine::get_clock() < cfg_deadline,
95                "Failed to exchange all tasks in less than %d seconds. Is this an infinite loop?", (int)cfg_deadline);
96
97     sg4::this_actor::sleep_for(0.75); // Give a chance to the monkey to kill this actor at this point
98
99     while (buf.sem_full->acquire_timeout(10))
100       XBT_INFO("Timeouted");
101     to_release.push(buf.sem_full);
102
103     sg4::this_actor::sleep_for(0.75); // Give a chance to the monkey to kill this actor at this point
104
105     item = buf.value;
106     XBT_INFO("Receiving item %d", item);
107     buf.sem_empty->release();
108     to_release.pop();
109   } while (item != 0);
110
111   XBT_INFO("Bye!");
112 }
113
114 int main(int argc, char** argv)
115 {
116   sg4::Engine e(&argc, argv);
117
118   auto* rootzone = sg4::create_full_zone("root");
119   auto* paul     = rootzone->create_host("Paul", 1e9);
120   auto* carol    = rootzone->create_host("Carol", 1e9);
121   sg4::LinkInRoute link(rootzone->create_link("link", "1MBps")->set_latency("24us")->seal());
122   rootzone->add_route(paul->get_netpoint(), carol->get_netpoint(), nullptr, nullptr, {link}, true);
123
124   SharedBuffer buffer;
125   sg4::Actor::create("producer", paul, producer, std::ref(buffer))->set_auto_restart();
126   sg4::Actor::create("consumer", carol, consumer, std::cref(buffer))->set_auto_restart();
127   e.run();
128
129   return 0;
130 }