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 %s\n",
 201                           str);
 202         }
 203
 204
 205         /* Reserve contract */
 206         contracts._length = 1;
 207         contract_seq_buf = contract;
 208         contracts._buffer = &contract_seq_buf;
 209         ret = fres_resource_manager_reserve_contracts(rm->rm, &contracts, ev);
 210         if (forb_exception_occured(ev)) {
 211                 goto err;
 212         }
 213         if (ret != 0) {
 214                 ul_logmsg("Contract was not accepted\n");
 215                 goto err;
 216         }
 217
 218         /* Commit contract */
 219         ids._length = 1;
 220         ids._buffer = id;
 221         fres_resource_manager_commit_contracts(rm->rm, &ids, &contracts_sched, ev);
 222         if (forb_exception_occured(ev)) {
 223                 goto err;
 224         }
 225
 226         /* Create VRes */
 227         ret = fres_resource_scheduler_create_vreses(rs->rs, contracts_sched, ev);
 228         if (CORBA_sequence_get_release(contracts_sched)) forb_free(contracts_sched->_buffer);
 229         forb_free(contracts_sched);
 230         if (forb_exception_occured(ev)) {
 231                 goto err;
 232         }
 233         if (ret != 0) {
 234                 ul_logmsg("VRes was not created\n");
 235                 goto err;
 236         }
 237
 238         return 0;
 239 err:
 240         return ret;
 241 }
 242
 243 CORBA_long register_manager(fres_contract_broker obj,
 244                             const frsh_resource_type_t restype,
 245                             const frsh_resource_id_t resid,
 246                             const fres_resource_manager rm_obj,
 247                             CORBA_Environment *ev)
 248 {
 249         struct fcb *fcb = o2fcb(obj);
 250         struct res_mng *rm, *rm2;
 251         rm = malloc(sizeof(*rm));
 252         if (!rm) goto err;
 253         memset(rm, 0, sizeof(*rm));
 254         rm->resource.type = restype;
 255         rm->resource.id = resid;
 256         rm2 = fcb_resource_find(fcb, &rm->resource);
 257         if (rm2) {
 258                 ul_logerr("Resource manager %d.%d already regsitered\n",
 259                           restype, resid);
 260                 goto free_err;
 261         }
 262         rm->rm = forb_object_duplicate(rm_obj);
 263         fcb_resource_insert(fcb, rm);
 264         return 0;
 265 free_err:
 266         free(rm);
 267 err:
 268         return -1;
 269 }
 270
 271
 272 CORBA_long register_scheduler(fres_contract_broker obj,
 273                               const frsh_resource_type_t restype,
 274                               const frsh_resource_id_t resid,
 275                               const fres_resource_scheduler rs_obj,
 276                               CORBA_Environment *ev)
 277 {
 278         struct fcb *fcb = o2fcb(obj);
 279         struct res_mng *rm;
 280         struct res_sched *rs;
 281         struct res_key resource;
 282         forb_server_id server_id;
 283
 284
 285         resource.type = restype;
 286         resource.id = resid;
 287         rm = fcb_resource_find(fcb, &resource);
 288         if (!rm) {
 289                 goto err;
 290         }
 291         forb_get_server_id(rs_obj, &server_id);
 292         rs = fcb_sched_find(rm, &server_id);
 293         if (rs) {
 294                 char *str = forb_object_to_string(rs_obj);
 295                 ul_logerr("Scheduler from already registered (%s)\n",
 296                           str);
 297                 forb_free(str);
 298                 goto err;
 299         }
 300         rs = malloc(sizeof(*rs));
 301         if (!rs) {
 302                 goto err;
 303         }
 304         rs->app = server_id;
 305         rs->rs = forb_object_duplicate(rs_obj);
 306         fcb_sched_insert(rm, rs);
 307         return 0;
 308 err:
 309         return -1;
 310 }
 311
 312 struct forb_fres_contract_broker_impl impl = {
 313         .negotiate_contract = negotiate_contract,
 314         .register_manager = register_manager,
 315         .register_scheduler = register_scheduler,
 316 };
 317
 318 int main(int argc, char *argv[])
 319 {
 320         forb_orb orb;
 321         struct fcb fcb_data;
 322         fres_contract_broker fcb;
 323
 324         orb = forb_init(&argc, &argv, "fcb");
 325         if (!orb) error(1, errno, "FORB initialization failed");
 326
 327
 328
 329         fcb = forb_fres_contract_broker_new(orb, &impl, &fcb_data);
 330
 331         if (forb_register_reference(fcb, fres_contract_broker_reg_name) != 0) {
 332                 error(1, errno, "forb_register_reference() failed");
 333         }
 334
 335         forb_execute_object(fcb);
 336
 337         return 0;
 338 }