fres/cbroker/fcb.c

   1 /**************************************************************************/
   2 /* ---------------------------------------------------------------------- */
   3 /* Copyright (C) 2006 - 2008 FRESCOR consortium partners:                 */
   4 /*                                                                        */
   5 /*   Universidad de Cantabria,              SPAIN                         */
   6 /*   University of York,                    UK                            */
   7 /*   Scuola Superiore Sant'Anna,            ITALY                         */
   8 /*   Kaiserslautern University,             GERMANY                       */
   9 /*   Univ. Politécnica  Valencia,           SPAIN                        */
  10 /*   Czech Technical University in Prague,  CZECH REPUBLIC                */
  11 /*   ENEA                                   SWEDEN                        */
  12 /*   Thales Communication S.A.              FRANCE                        */
  13 /*   Visual Tools S.A.                      SPAIN                         */
  14 /*   Rapita Systems Ltd                     UK                            */
  15 /*   Evidence                               ITALY                         */
  16 /*                                                                        */
  17 /*   See http://www.frescor.org for a link to partners' websites          */
  18 /*                                                                        */
  19 /*          FRESCOR project (FP6/2005/IST/5-034026) is funded             */
  20 /*       in part by the European Union Sixth Framework Programme          */
  21 /*       The European Union is not liable of any use that may be          */
  22 /*       made of this code.                                               */
  23 /*                                                                        */
  24 /*                                                                        */
  25 /*  This file is part of FRSH (FRescor ScHeduler)                         */
  26 /*                                                                        */
  27 /* FRSH is free software; you can redistribute it and/or modify it        */
  28 /* under terms of the GNU General Public License as published by the      */
  29 /* Free Software Foundation; either version 2, or (at your option) any    */
  30 /* later version.  FRSH is distributed in the hope that it will be        */
  31 /* useful, but WITHOUT ANY WARRANTY; without even the implied warranty    */
  32 /* of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU    */
  33 /* General Public License for more details. You should have received a    */
  34 /* copy of the GNU General Public License along with FRSH; see file       */
  35 /* COPYING. If not, write to the Free Software Foundation, 675 Mass Ave,  */
  36 /* Cambridge, MA 02139, USA.                                              */
  37 /*                                                                        */
  38 /* As a special exception, including FRSH header files in a file,         */
  39 /* instantiating FRSH generics or templates, or linking other files       */
  40 /* with FRSH objects to produce an executable application, does not       */
  41 /* by itself cause the resulting executable application to be covered     */
  42 /* by the GNU General Public License. This exception does not             */
  43 /* however invalidate any other reasons why the executable file might be  */
  44 /* covered by the GNU Public License.                                     */
  45 /**************************************************************************/
  46
  47 /**
  48  * @file   fcb.c
  49  * @author Michal Sojka <sojkam1@fel.cvut.cz>
  50  * @date   Mon Oct 20 18:00:18 2008
  51  *
  52  * @brief  FRESCOR Contract Broker
  53  *
  54  *
  55  */
  56 #include <forb.h>
  57 #include <fcb.h>
  58 #include <error.h>
  59 #include <errno.h>
  60 #include <stdio.h>
  61 #include <ul_gavlcust.h>
  62 #include <string.h>
  63 #include <ul_log.h>
  64 #include <forb/server_id.h>
  65 #include <fres_contract.h>
  66
  67 UL_LOG_CUST(ulogd_fcb);
  68 ul_log_domain_t ulogd_fcb = {UL_LOGL_MSG, "fcb"};
  69
  70
  71 /**
  72  * Resource identification
  73  */
  74 struct res_key {
  75         frsh_resource_type_t type;
  76         frsh_resource_id_t id;
  77 };
  78
  79 /**
  80  * Registered resource manager
  81  */
  82 struct res_mng {
  83         gavl_node_t node;
  84         struct res_key resource;
  85         fres_resource_manager rm;
  86         gavl_cust_root_field_t schedulers;      /**< Registered schedulers for this resource (from multiple applications/nodes) */
  87 };
  88
  89 /**
  90  * Registered resource scheduler
  91  */
  92 struct res_sched {
  93         gavl_node_t node;
  94         forb_server_id app;
  95         fres_resource_scheduler rs;
  96 };
  97
  98 struct fcb {
  99         gavl_cust_root_field_t resources;
 100 };
 101
 102 static inline int res_key_cmp(struct res_key *a, struct res_key *b)
 103 {
 104         if (a->type < b->type) {
 105                 return -1;
 106         } else if (a->type > b->type) {
 107                 return +1;
 108         } else if (a->id < b->id) {
 109                 return -1;
 110         } else if (a->id > b->id) {
 111                 return +1;
 112         } else {
 113                 return 0;
 114         }
 115 }
 116
 117 GAVL_CUST_NODE_INT_DEC(fcb_resource   /* cust_prefix */,                \
 118                        struct fcb     /* cust_root_t */,                \
 119                        struct res_mng /* cust_item_t */,                \
 120                        struct res_key /* cust_key_t */,                 \
 121                        resources      /* cust_root_node */,             \
 122                        node           /* cust_item_node */,             \
 123                        resource       /* cust_item_key */,              \
 124                        res_key_cmp    /* cust_cmp_fnc */)
 125
 126 GAVL_CUST_NODE_INT_IMP(fcb_resource   /* cust_prefix */,                \
 127                        struct fcb     /* cust_root_t */,                \
 128                        struct res_mng /* cust_item_t */,                \
 129                        struct res_key /* cust_key_t */,                 \
 130                        resources      /* cust_root_node */,             \
 131                        node           /* cust_item_node */,             \
 132                        resource       /* cust_item_key */,              \
 133                        res_key_cmp    /* cust_cmp_fnc */)
 134
 135 GAVL_CUST_NODE_INT_DEC(fcb_sched        /* cust_prefix */,              \
 136                        struct res_mng   /* cust_root_t */,              \
 137                        struct res_sched /* cust_item_t */,              \
 138                        forb_server_id   /* cust_key_t */,               \
 139                        schedulers       /* cust_root_node */,           \
 140                        node             /* cust_item_node */,           \
 141                        app              /* cust_item_key */,            \
 142                        fres_contract_id_cmp /* cust_cmp_fnc */)
 143
 144 GAVL_CUST_NODE_INT_IMP(fcb_sched        /* cust_prefix */,              \
 145                        struct res_mng   /* cust_root_t */,              \
 146                        struct res_sched /* cust_item_t */,              \
 147                        forb_server_id   /* cust_key_t */,               \
 148                        schedulers       /* cust_root_node */,           \
 149                        node             /* cust_item_node */,           \
 150                        app              /* cust_item_key */,            \
 151                        forb_server_id_cmp /* cust_cmp_fnc */)
 152
 153
 154 #define o2fcb(o) (struct fcb*)forb_instance_data(o)
 155
 156 CORBA_long
 157 negotiate_contract(fres_contract_broker obj,
 158                    const fres_contract_ptr contract,
 159                    fres_contract_id_t* id,
 160                    CORBA_Environment *ev)
 161 {
 162         struct fcb *fcb = o2fcb(obj);
 163         fres_block_resource *res;
 164         struct res_key res_key;
 165         struct res_mng *rm;
 166         struct res_sched *rs;
 167         int ret;
 168         forb_server_id app;
 169         fres_contract_ptr contract_seq_buf;
 170         fres_contract_ptr_seq contracts;
 171         fres_contract_ptr_seq *contracts_sched;
 172         fres_contract_id_seq ids;
 173
 174
 175         res = fres_contract_get_resource(contract);
 176         if (!res) {
 177                 ul_logerr("No resource specified\n");
 178                 ret = -1;
 179                 goto err;
 180         }
 181         res_key.type = res->resource_type;
 182         res_key.id = res->resource_id;
 183
 184         rm = fcb_resource_find(fcb, &res_key);
 185         if (!rm) {
 186                 ul_logerr("No resource manager for %d.%d registered\n",
 187                           res_key.type, res_key.id);
 188                 ret = -1;
 189                 goto err;
 190         }
 191
 192         forb_uuid_generate((forb_uuid_t *)id);
 193         contract->id = *id;
 194
 195         forb_get_req_source(obj, &app);
 196         rs = fcb_sched_find(rm, &app);
 197         if (!rs) {
 198                 char str[60];
 199                 forb_server_id_to_string(str, &app, sizeof(str));
 200                 ul_logerr("No resource scheduler found for %d.%d and %s\n",
 201                           res_key.type, res_key.id, str);
 202                 goto err;
 203         }
 204
 205
 206         /* Reserve contract */
 207         contracts._length = 1;
 208         contract_seq_buf = contract;
 209         contracts._buffer = &contract_seq_buf;
 210         ret = fres_resource_manager_reserve_contracts(rm->rm, &contracts, ev);
 211         if (forb_exception_occured(ev) || ret < 0) {
 212                 goto err;
 213         }
 214         if (ret == 1) {
 215                 ul_logmsg("Contract was not accepted\n");
 216                 goto err;
 217         }
 218
 219         /* Commit contract */
 220         ids._length = 1;
 221         ids._buffer = id;
 222         fres_resource_manager_commit_contracts(rm->rm, &ids, &contracts_sched, ev);
 223         if (forb_exception_occured(ev)) {
 224                 goto err;
 225         }
 226
 227         /* Create VRes */
 228         ret = fres_resource_scheduler_create_vreses(rs->rs, contracts_sched, ev);
 229         if (CORBA_sequence_get_release(contracts_sched)) forb_free(contracts_sched->_buffer);
 230         forb_free(contracts_sched);
 231         if (forb_exception_occured(ev)) {
 232                 goto err;
 233         }
 234         if (ret != 0) {
 235                 ul_logmsg("VRes was not created\n");
 236                 goto err;
 237         }
 238
 239         return 0;
 240 err:
 241         return ret;
 242 }
 243
 244 CORBA_long register_manager(fres_contract_broker obj,
 245                             const frsh_resource_type_t restype,
 246                             const frsh_resource_id_t resid,
 247                             const fres_resource_manager rm_obj,
 248                             CORBA_Environment *ev)
 249 {
 250         struct fcb *fcb = o2fcb(obj);
 251         struct res_mng *rm, *rm2;
 252
 253         rm = malloc(sizeof(*rm));
 254         if (!rm) goto err;
 255         memset(rm, 0, sizeof(*rm));
 256         rm->resource.type = restype;
 257         rm->resource.id = resid;
 258         rm2 = fcb_resource_find(fcb, &rm->resource);
 259         if (rm2) {
 260                 if (forb_object_is_stale(rm2->rm)) {
 261                         ul_logmsg("Removing stale manager for resource %d.%d\n",
 262                                   restype, resid);
 263                         forb_object_release(rm2->rm);
 264                         fcb_resource_delete(fcb, rm2);
 265                         /* TODO: Delete also all schedulers associated
 266                          * with this stale resource manager. */
 267                         free(rm);
 268                         rm = rm2;
 269                 } else {
 270                         ul_logerr("Resource manager %d.%d already registered\n",
 271                                   restype, resid);
 272                         goto free_err;
 273                 }
 274         }
 275         rm->rm = forb_object_duplicate(rm_obj);
 276
 277         fcb_sched_init_root_field(rm);
 278         ul_logmsg("Registering manager for resource %d.%d\n",
 279                   restype, resid);
 280         fcb_resource_insert(fcb, rm);
 281         return 0;
 282 free_err:
 283         free(rm);
 284 err:
 285         return -1;
 286 }
 287
 288
 289 CORBA_long register_scheduler(fres_contract_broker obj,
 290                               const frsh_resource_type_t restype,
 291                               const frsh_resource_id_t resid,
 292                               const fres_resource_scheduler rs_obj,
 293                               CORBA_Environment *ev)
 294 {
 295         struct fcb *fcb = o2fcb(obj);
 296         struct res_mng *rm;
 297         struct res_sched *rs;
 298         struct res_key resource;
 299         forb_server_id server_id;
 300         char server_id_str[40];
 301
 302         forb_get_server_id(rs_obj, &server_id);
 303         forb_server_id_to_string(server_id_str, &server_id, sizeof(server_id_str));
 304         ul_logmsg("Registering scheduler for resource %d.%d in app %s\n",
 305                   restype, resid, server_id_str);
 306
 307         resource.type = restype;
 308         resource.id = resid;
 309         rm = fcb_resource_find(fcb, &resource);
 310         if (!rm) {
 311                 goto err;
 312         }
 313         rs = fcb_sched_find(rm, &server_id);
 314         if (rs) {
 315                 char *str = forb_object_to_string(rs_obj);
 316                 ul_logerr("Scheduler from already registered (%s)\n",
 317                           str);
 318                 forb_free(str);
 319                 goto err;
 320         }
 321         rs = malloc(sizeof(*rs));
 322         if (!rs) {
 323                 goto err;
 324         }
 325         rs->app = server_id;
 326         rs->rs = forb_object_duplicate(rs_obj);
 327         fcb_sched_insert(rm, rs);
 328         return 0;
 329 err:
 330         return -1;
 331 }
 332
 333 struct forb_fres_contract_broker_impl impl = {
 334         .negotiate_contract = negotiate_contract,
 335         .register_manager = register_manager,
 336         .register_scheduler = register_scheduler,
 337 };
 338
 339 int main(int argc, char *argv[])
 340 {
 341         forb_orb orb;
 342         struct fcb fcb_data;
 343         fres_contract_broker fcb;
 344
 345         orb = forb_init(&argc, &argv, "fcb");
 346         if (!orb) error(1, errno, "FORB initialization failed");
 347
 348         fcb_resource_init_root_field(&fcb_data);
 349
 350         fcb = forb_fres_contract_broker_new(orb, &impl, &fcb_data);
 351
 352         if (forb_register_reference(fcb, fres_contract_broker_reg_name) != 0) {
 353                 error(1, errno, "forb_register_reference() failed");
 354         }
 355
 356         forb_execute_object(fcb);
 357
 358         return 0;
 359 }