Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Replace linear search by a faster one.
[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/plugins/vm/VirtualMachineImpl.hpp"
7 #include <string>
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()
111 {
112   for (auto mnt : storage_)
113     xbt_free(mnt.name);
114 }
115
116 simgrid::surf::Storage* HostImpl::findStorageOnMountList(const char* mount)
117 {
118   simgrid::surf::Storage* st = nullptr;
119
120   XBT_DEBUG("Search for storage name '%s' on '%s'", mount, piface_->cname());
121   for (auto mnt : storage_) {
122     XBT_DEBUG("See '%s'", mnt.name);
123     if (!strcmp(mount, mnt.name)) {
124       st = static_cast<simgrid::surf::Storage*>(mnt.storage);
125       break;
126     }
127   }
128   if (!st)
129     xbt_die("Can't find mount '%s' for '%s'", mount, piface_->cname());
130   return st;
131 }
132
133 xbt_dict_t HostImpl::getMountedStorageList()
134 {
135   xbt_dict_t storage_list = xbt_dict_new_homogeneous(nullptr);
136   char* storage_name      = nullptr;
137
138   for (auto mnt : storage_) {
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 void HostImpl::getAttachedStorageList(std::vector<const char*>* storages)
146 {
147   xbt_lib_cursor_t cursor;
148   char* key;
149   void** data;
150   xbt_lib_foreach(storage_lib, cursor, key, data)
151   {
152     if (xbt_lib_get_level(xbt_lib_get_elm_or_null(storage_lib, key), SURF_STORAGE_LEVEL) != nullptr) {
153       simgrid::surf::Storage* storage = static_cast<simgrid::surf::Storage*>(
154           xbt_lib_get_level(xbt_lib_get_elm_or_null(storage_lib, key), SURF_STORAGE_LEVEL));
155       if (!strcmp(static_cast<const char*>(storage->attach_), piface_->cname())) {
156         storages->push_back(storage->cname());
157       }
158     }
159   }
160 }
161
162 Action* HostImpl::open(const char* fullpath)
163 {
164
165   simgrid::surf::Storage* st = nullptr;
166   size_t longest_prefix_length = 0;
167   std::string path;
168   std::string mount_name;
169
170   XBT_DEBUG("Search for storage name for '%s' on '%s'", fullpath, piface_->cname());
171   for (auto mnt : storage_) {
172     XBT_DEBUG("See '%s'", mnt.name);
173     std::string file_mount_name = std::string(fullpath).substr(0, strlen(mnt.name));
174
175     if (!strcmp(file_mount_name.c_str(), mnt.name) &&
176         strlen(mnt.name) > longest_prefix_length) { /* The current mount name is found in the full path and is
177                                                            bigger than the previous*/
178       longest_prefix_length = strlen(mnt.name);
179       st                    = static_cast<simgrid::surf::Storage*>(mnt.storage);
180     }
181   }
182   if (longest_prefix_length > 0) { /* Mount point found, split fullpath into mount_name and path+filename*/
183     mount_name = std::string(fullpath).substr(0, longest_prefix_length);
184     path       = std::string(fullpath).substr(longest_prefix_length, strlen(fullpath));
185   } else
186     xbt_die("Can't find mount point for '%s' on '%s'", fullpath, piface_->cname());
187
188   XBT_DEBUG("OPEN %s on disk '%s'", path.c_str(), st->cname());
189   Action* action = st->open(mount_name.c_str(), path.c_str());
190   return action;
191 }
192
193 Action* HostImpl::close(surf_file_t fd)
194 {
195   simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
196   XBT_DEBUG("CLOSE %s on disk '%s'", fd->name, st->cname());
197   return st->close(fd);
198 }
199
200 Action* HostImpl::read(surf_file_t fd, sg_size_t size)
201 {
202   simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
203   XBT_DEBUG("READ %s on disk '%s'", fd->name, st->cname());
204   return st->read(fd, size);
205 }
206
207 Action* HostImpl::write(surf_file_t fd, sg_size_t size)
208 {
209   simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
210   XBT_DEBUG("WRITE %s on disk '%s'", fd->name, st->cname());
211   return st->write(fd, size);
212 }
213
214 int HostImpl::unlink(surf_file_t fd)
215 {
216   if (!fd) {
217     XBT_WARN("No such file descriptor. Impossible to unlink");
218     return -1;
219   } else {
220
221     simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
222     /* Check if the file is on this storage */
223     if (st->content_->find(fd->name) == st->content_->end()) {
224       XBT_WARN("File %s is not on disk %s. Impossible to unlink", fd->name, st->cname());
225       return -1;
226     } else {
227       XBT_DEBUG("UNLINK %s on disk '%s'", fd->name, st->cname());
228       st->usedSize_ -= fd->size;
229
230       // Remove the file from storage
231       sg_size_t* psize = st->content_->at(fd->name);
232       delete psize;
233       st->content_->erase(fd->name);
234
235       xbt_free(fd->name);
236       xbt_free(fd->mount);
237       xbt_free(fd);
238       return 0;
239     }
240   }
241 }
242
243 sg_size_t HostImpl::getSize(surf_file_t fd)
244 {
245   return fd->size;
246 }
247
248 xbt_dynar_t HostImpl::getInfo(surf_file_t fd)
249 {
250   simgrid::surf::Storage* st = findStorageOnMountList(fd->mount);
251   sg_size_t* psize           = xbt_new(sg_size_t, 1);
252   *psize                     = fd->size;
253   xbt_dynar_t info           = xbt_dynar_new(sizeof(void*), nullptr);
254   xbt_dynar_push_as(info, sg_size_t*, psize);
255   xbt_dynar_push_as(info, void*, fd->mount);
256   xbt_dynar_push_as(info, void*, (void*)st->cname());
257   xbt_dynar_push_as(info, void*, st->typeId_);
258   xbt_dynar_push_as(info, void*, st->contentType_);
259
260   return info;
261 }
262
263 sg_size_t HostImpl::fileTell(surf_file_t fd)
264 {
265   return fd->current_position;
266 }
267
268 int HostImpl::fileSeek(surf_file_t fd, sg_offset_t offset, int origin)
269 {
270   switch (origin) {
271   case SEEK_SET:
272     fd->current_position = offset;
273     return 0;
274   case SEEK_CUR:
275     fd->current_position += offset;
276     return 0;
277   case SEEK_END:
278     fd->current_position = fd->size + offset;
279     return 0;
280   default:
281     return -1;
282   }
283 }
284
285 int HostImpl::fileMove(surf_file_t fd, const char* fullpath)
286 {
287   /* Check if the new full path is on the same mount point */
288   if (!strncmp((const char*)fd->mount, fullpath, strlen(fd->mount))) {
289     std::map<std::string, sg_size_t*>* content = findStorageOnMountList(fd->mount)->content_;
290     if (content->find(fd->name) != content->end()) { // src file exists
291       sg_size_t* psize     = content->at(std::string(fd->name));
292       sg_size_t* new_psize = new sg_size_t;
293       *new_psize           = *psize;
294       delete psize;
295       content->erase(fd->name);
296       std::string path = std::string(fullpath).substr(strlen(fd->mount), strlen(fullpath));
297       content->insert({path.c_str(), new_psize});
298       XBT_DEBUG("Move file from %s to %s, size '%llu'", fd->name, fullpath, *psize);
299       return 0;
300     } else {
301       XBT_WARN("File %s doesn't exist", fd->name);
302       return -1;
303     }
304   } else {
305     XBT_WARN("New full path %s is not on the same mount point: %s. Action has been canceled.", fullpath, fd->mount);
306     return -1;
307   }
308 }
309
310 }
311 }