examples/s4u/io/s4u_io_test.cpp

   1 /* Copyright (c) 2006-2015. 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 <map>
   7 #include <unordered_map>
   8 #include <vector>
   9
  10 #include "simgrid/s4u.h"
  11
  12 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "a sample log category");
  13
  14 class myHost : simgrid::s4u::Actor {
  15 public:
  16         myHost(const char*procname, simgrid::s4u::Host *host,int argc, char **argv)
  17 : simgrid::s4u::Actor(procname,host,argc,argv){}
  18
  19         void show_info(boost::unordered_map <std::string, simgrid::s4u::Storage*> &mounts) {
  20                 XBT_INFO("Storage info on %s:",
  21                         simgrid::s4u::Host::current()->name().c_str());
  22
  23                 for (const auto&kv : mounts) {
  24                         const char* mountpoint = kv.first.c_str();
  25                         simgrid::s4u::Storage &storage = *kv.second;
  26
  27                         // Retrieve disk's information
  28                         sg_size_t free_size = storage.size_free();
  29                         sg_size_t used_size = storage.size_used();
  30                         sg_size_t size = storage.size();
  31
  32                         XBT_INFO("    %s (%s) Used: %llu; Free: %llu; Total: %llu.",
  33                                         storage.name(), mountpoint, used_size, free_size, size);
  34                 }
  35         }
  36
  37         int main(int argc, char **argv) {
  38                 boost::unordered_map <std::string, simgrid::s4u::Storage *>& mounts =
  39                         simgrid::s4u::Host::current()->mountedStorages();
  40
  41                 show_info(mounts);
  42
  43                 // Open an non-existing file to create it
  44                 const char *filename = "/home/tmp/data.txt";
  45                 simgrid::s4u::File *file = new simgrid::s4u::File(filename, NULL);
  46                 sg_size_t write, read, file_size;
  47
  48                 write = file->write(200000);  // Write 200,000 bytes
  49                 XBT_INFO("Create a %llu bytes file named '%s' on /sd1", write, filename);
  50
  51                 // check that sizes have changed
  52                 show_info(mounts);
  53
  54                 // Now retrieve the size of created file and read it completely
  55                 file_size = file->size();
  56                 file->seek(0);
  57                 read = file->read(file_size);
  58                 XBT_INFO("Read %llu bytes on %s", read, filename);
  59
  60                 // Now write 100,000 bytes in tmp/data.txt
  61                 write = file->write(100000);  // Write 100,000 bytes
  62                 XBT_INFO("Write %llu bytes on %s", write, filename);
  63
  64                 simgrid::s4u::Storage &storage = simgrid::s4u::Storage::byName("Disk4");
  65
  66                 // Now rename file from ./tmp/data.txt to ./tmp/simgrid.readme
  67                 const char *newpath = "/home/tmp/simgrid.readme";
  68                 XBT_INFO("Move '%s' to '%s'", file->path(), newpath);
  69                 file->move(newpath);
  70
  71                 // Test attaching some user data to the file
  72                 file->set_userdata(xbt_strdup("777"));
  73                 XBT_INFO("User data attached to the file: %s", (char*)file->userdata());
  74
  75                 // Close the file
  76                 delete file;
  77
  78                 // Now attach some user data to disk1
  79                 XBT_INFO("Get/set data for storage element: %s",storage.name());
  80                 XBT_INFO("    Uninitialized storage data: '%s'", (char*)storage.userdata());
  81
  82                 storage.set_userdata(xbt_strdup("Some user data"));
  83                 XBT_INFO("    Set and get data: '%s'", (char*)storage.userdata());
  84
  85                 /*
  86                   // Dump disks contents
  87                   XBT_INFO("*** Dump content of %s ***",Host::current()->name());
  88                   xbt_dict_t contents = NULL;
  89                   contents = MSG_host_get_storage_content(MSG_host_self()); // contents is a dict of dicts
  90                   xbt_dict_cursor_t curs, curs2 = NULL;
  91                   char* mountname;
  92                   xbt_dict_t content;
  93                   char* path;
  94                   sg_size_t *size;
  95                   xbt_dict_foreach(contents, curs, mountname, content){
  96                     XBT_INFO("Print the content of mount point: %s",mountname);
  97                     xbt_dict_foreach(content,curs2,path,size){
  98                        XBT_INFO("%s size: %llu bytes", path,*((sg_size_t*)size));
  99                     }
 100                   xbt_dict_free(&content);
 101                   }
 102                   xbt_dict_free(&contents);
 103                  */
 104                 return 0;
 105         }
 106 };
 107
 108 int main(int argc, char **argv) {
 109         simgrid::s4u::Engine *e = new simgrid::s4u::Engine(&argc,argv);
 110         e->loadPlatform("../../platforms/storage/storage.xml");
 111
 112         new myHost("host", simgrid::s4u::Host::byName("denise"), 0, NULL);
 113         e->run();
 114         return 0;
 115 }