Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Avoid unsafe things
[simgrid.git] / src / surf / HostImpl.cpp
1 /* Copyright (c) 2013-2016. 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 "src/surf/HostImpl.hpp"
7 #include "src/plugins/vm/VirtualMachineImpl.hpp"
8
9 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_host, surf, "Logging specific to the SURF host module");
10
11 simgrid::surf::HostModel *surf_host_model = nullptr;
12
13 /*************
14  * Callbacks *
15  *************/
16
17 namespace simgrid {
18 namespace surf {
19
20 /*********
21  * Model *
22  *********/
23
24 /* Each VM has a dummy CPU action on the PM layer. This CPU action works as the
25  * constraint (capacity) of the VM in the PM layer. If the VM does not have any
26  * active task, the dummy CPU action must be deactivated, so that the VM does
27  * not get any CPU share in the PM layer. */
28 void HostModel::adjustWeightOfDummyCpuActions()
29 {
30   /* iterate for all virtual machines */
31   for (s4u::VirtualMachine* ws_vm : vm::VirtualMachineImpl::allVms_) {
32
33     Cpu* cpu = ws_vm->pimpl_cpu;
34
35     int is_active = lmm_constraint_used(cpu->model()->getMaxminSystem(), cpu->constraint());
36
37     if (is_active) {
38       /* some tasks exist on this VM */
39       XBT_DEBUG("set the weight of the dummy CPU action on PM to 1");
40
41       /* FIXME: we should use lmm_update_variable_weight() ? */
42       /* FIXME: If we assign 1.05 and 0.05, the system makes apparently wrong values. */
43       ws_vm->pimpl_vm_->action_->setPriority(1);
44
45     } else {
46       /* no task exits on this VM */
47       XBT_DEBUG("set the weight of the dummy CPU action on PM to 0");
48
49       ws_vm->pimpl_vm_->action_->setPriority(0);
50     }
51   }
52 }
53
54 /* Helper function for executeParallelTask */
55 static inline double has_cost(double* array, int pos)
56 {
57   if (array)
58     return array[pos];
59   else
60     return -1.0;
61 }
62 Action* HostModel::executeParallelTask(int host_nb, simgrid::s4u::Host** host_list, double* flops_amount,
63                                        double* bytes_amount, double rate)
64 {
65   Action* action = nullptr;
66   if ((host_nb == 1) && (has_cost(bytes_amount, 0) <= 0)) {
67     action = host_list[0]->pimpl_cpu->execution_start(flops_amount[0]);
68   } else if ((host_nb == 1) && (has_cost(flops_amount, 0) <= 0)) {
69     action = surf_network_model->communicate(host_list[0], host_list[0], bytes_amount[0], rate);
70   } else if ((host_nb == 2) && (has_cost(flops_amount, 0) <= 0) && (has_cost(flops_amount, 1) <= 0)) {
71     int nb = 0;
72     double value = 0.0;
73
74     for (int i = 0; i < host_nb * host_nb; i++) {
75       if (has_cost(bytes_amount, i) > 0.0) {
76         nb++;
77         value = has_cost(bytes_amount, i);
78       }
79     }
80     if (nb == 1) {
81       action = surf_network_model->communicate(host_list[0], host_list[1], value, rate);
82     } else if (nb == 0) {
83       xbt_die("Cannot have a communication with no flop to exchange in this model. You should consider using the "
84               "ptask model");
85     } else {
86       xbt_die("Cannot have a communication that is not a simple point-to-point in this model. You should consider "
87               "using the ptask model");
88     }
89   } else
90     xbt_die(
91         "This model only accepts one of the following. You should consider using the ptask model for the other cases.\n"
92         " - execution with one host only and no communication\n"
93         " - Self-comms with one host only\n"
94         " - Communications with two hosts and no computation");
95   xbt_free(host_list);
96   return action;
97 }
98
99 /************
100  * Resource *
101  ************/
102 HostImpl::HostImpl(s4u::Host* host) : piface_(host)
103 {
104   /* The VM wants to reinstall a new HostImpl, but we don't want to leak the previously existing one */
105   delete piface_->pimpl_;
106   piface_->pimpl_ = this;
107 }
108
109 /** @brief use destroy() instead of this destructor */
110 HostImpl::~HostImpl() = default;
111
112 simgrid::surf::Storage* HostImpl::findStorageOnMountList(const char* mount)
113 {
114   simgrid::surf::Storage* st = nullptr;
115   s_mount_t mnt;
116   unsigned int cursor;
117
118   XBT_DEBUG("Search for storage name '%s' on '%s'", mount, piface_->cname());
119   xbt_dynar_foreach (storage_, cursor, mnt) {
120     XBT_DEBUG("See '%s'", mnt.name);
121     if (!strcmp(mount, mnt.name)) {
122       st = static_cast<simgrid::surf::Storage*>(mnt.storage);
123       break;
124     }
125   }
126   if (!st)
127     xbt_die("Can't find mount '%s' for '%s'", mount, piface_->cname());
128   return st;
129 }
130
131 xbt_dict_t HostImpl::getMountedStorageList()
132 {
133   s_mount_t mnt;
134   unsigned int i;
135   xbt_dict_t storage_list = xbt_dict_new_homogeneous(nullptr);
136   char* storage_name      = nullptr;
137
138   xbt_dynar_foreach (storage_, i, mnt) {
139     storage_name = (char*)static_cast<simgrid::surf::Storage*>(mnt.storage)->cname();
140     xbt_dict_set(storage_list, mnt.name, storage_name, nullptr);
141   }
142   return storage_list;
143 }
144
145 xbt_dynar_t HostImpl::getAttachedStorageList()
146 {
147   xbt_lib_cursor_t cursor;
148   char* key;
149   void** data;
150   xbt_dynar_t result = xbt_dynar_new(sizeof(void*), nullptr);
151   xbt_lib_foreach(storage_lib, cursor, key, data)
152   {
153     if (xbt_lib_get_level(xbt_lib_get_elm_or_null(storage_lib, key), SURF_STORAGE_LEVEL) != nullptr) {
154       simgrid::surf::Storage* storage = static_cast<simgrid::surf::Storage*>(
155           xbt_lib_get_level(xbt_lib_get_elm_or_null(storage_lib, key), SURF_STORAGE_LEVEL));
156       if (!strcmp((const char*)storage->attach_, piface_->cname())) {
157         xbt_dynar_push_as(result, void*, (void*)storage->cname());
158       }
159     }
160   }
161   return result;
162     }
163
164     Action* HostImpl::open(const char* fullpath)
165     {
166
167       simgrid::surf::Storage* st = nullptr;
168       s_mount_t mnt;
169       unsigned int cursor;
170       size_t longest_prefix_length = 0;
171       char* path                   = nullptr;
172       char* file_mount_name        = nullptr;
173       char* mount_name             = nullptr;
174
175       XBT_DEBUG("Search for storage name for '%s' on '%s'", fullpath, piface_->cname());
176       xbt_dynar_foreach (storage_, cursor, mnt) {
177         XBT_DEBUG("See '%s'", mnt.name);
178         file_mount_name = (char*)xbt_malloc((strlen(mnt.name) + 1));
179         strncpy(file_mount_name, fullpath, strlen(mnt.name) + 1);
180         file_mount_name[strlen(mnt.name)] = '\0';
181
182         if (!strcmp(file_mount_name, mnt.name) &&
183             strlen(mnt.name) > longest_prefix_length) { /* The current mount name is found in the full path and is
184                                                            bigger than the previous*/
185           longest_prefix_length = strlen(mnt.name);
186           st                    = static_cast<simgrid::surf::Storage*>(mnt.storage);
187         }
188         free(file_mount_name);
189       }
190       if (longest_prefix_length > 0) { /* Mount point found, split fullpath into mount_name and path+filename*/
191         path       = (char*)xbt_malloc((strlen(fullpath) - longest_prefix_length + 1));
192         mount_name = (char*)xbt_malloc((longest_prefix_length + 1));
193         strncpy(mount_name, fullpath, longest_prefix_length + 1);
194         strncpy(path, fullpath + longest_prefix_length, strlen(fullpath) - longest_prefix_length + 1);
195         path[strlen(fullpath) - longest_prefix_length] = '\0';
196         mount_name[longest_prefix_length]              = '\0';
197       } else
198         xbt_die("Can't find mount point for '%s' on '%s'", fullpath, piface_->cname());
199
200       XBT_DEBUG("OPEN %s on disk '%s'", path, st->cname());
201       Action* action = st->open((const char*)mount_name, (const char*)path);
202       free((char*)path);
203       free((char*)mount_name);
204       return action;
205     }
206
207     Action* HostImpl::close(surf_file_t fd)
208     {
209       simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
210       XBT_DEBUG("CLOSE %s on disk '%s'", fd->name, st->cname());
211       return st->close(fd);
212     }
213
214     Action* HostImpl::read(surf_file_t fd, sg_size_t size)
215     {
216       simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
217       XBT_DEBUG("READ %s on disk '%s'", fd->name, st->cname());
218       return st->read(fd, size);
219     }
220
221     Action* HostImpl::write(surf_file_t fd, sg_size_t size)
222     {
223       simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
224       XBT_DEBUG("WRITE %s on disk '%s'", fd->name, st->cname());
225       return st->write(fd, size);
226     }
227
228     int HostImpl::unlink(surf_file_t fd)
229     {
230       if (!fd) {
231         XBT_WARN("No such file descriptor. Impossible to unlink");
232         return -1;
233       } else {
234
235         simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
236         /* Check if the file is on this storage */
237         if (!xbt_dict_get_or_null(st->content_, fd->name)) {
238           XBT_WARN("File %s is not on disk %s. Impossible to unlink", fd->name, st->cname());
239           return -1;
240         } else {
241           XBT_DEBUG("UNLINK %s on disk '%s'", fd->name, st->cname());
242           st->usedSize_ -= fd->size;
243
244           // Remove the file from storage
245           xbt_dict_remove(st->content_, fd->name);
246
247           xbt_free(fd->name);
248           xbt_free(fd->mount);
249           xbt_free(fd);
250           return 0;
251         }
252       }
253     }
254
255     sg_size_t HostImpl::getSize(surf_file_t fd)
256     {
257       return fd->size;
258     }
259
260     xbt_dynar_t HostImpl::getInfo(surf_file_t fd)
261     {
262       simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
263       sg_size_t* psize           = xbt_new(sg_size_t, 1);
264       *psize                     = fd->size;
265       xbt_dynar_t info           = xbt_dynar_new(sizeof(void*), nullptr);
266       xbt_dynar_push_as(info, sg_size_t*, psize);
267       xbt_dynar_push_as(info, void*, fd->mount);
268       xbt_dynar_push_as(info, void*, (void*)st->cname());
269       xbt_dynar_push_as(info, void*, st->typeId_);
270       xbt_dynar_push_as(info, void*, st->contentType_);
271
272       return info;
273     }
274
275     sg_size_t HostImpl::fileTell(surf_file_t fd)
276     {
277       return fd->current_position;
278     }
279
280     int HostImpl::fileSeek(surf_file_t fd, sg_offset_t offset, int origin)
281     {
282
283       switch (origin) {
284         case SEEK_SET:
285           fd->current_position = offset;
286           return 0;
287         case SEEK_CUR:
288           fd->current_position += offset;
289           return 0;
290         case SEEK_END:
291           fd->current_position = fd->size + offset;
292           return 0;
293         default:
294           return -1;
295       }
296     }
297
298     int HostImpl::fileMove(surf_file_t fd, const char* fullpath)
299     {
300       /* Check if the new full path is on the same mount point */
301       if (!strncmp((const char*)fd->mount, fullpath, strlen(fd->mount))) {
302         sg_size_t* psize = (sg_size_t*)xbt_dict_get_or_null(findStorageOnMountList(fd->mount)->content_, fd->name);
303         if (psize) { // src file exists
304           sg_size_t* new_psize = xbt_new(sg_size_t, 1);
305           *new_psize           = *psize;
306           xbt_dict_remove(findStorageOnMountList(fd->mount)->content_, fd->name);
307           char* path = (char*)xbt_malloc((strlen(fullpath) - strlen(fd->mount) + 1));
308           strncpy(path, fullpath + strlen(fd->mount), strlen(fullpath) - strlen(fd->mount) + 1);
309           xbt_dict_set(findStorageOnMountList(fd->mount)->content_, path, new_psize, nullptr);
310           XBT_DEBUG("Move file from %s to %s, size '%llu'", fd->name, fullpath, *psize);
311           free(path);
312           return 0;
313         } else {
314           XBT_WARN("File %s doesn't exist", fd->name);
315           return -1;
316         }
317       } else {
318         XBT_WARN("New full path %s is not on the same mount point: %s. Action has been canceled.", fullpath, fd->mount);
319         return -1;
320       }
321     }
322
323     }
324     }