main.c

Go to the documentation of this file.

/*
START OF LICENSE STUB
    DeDOS: Declarative Dispersion-Oriented Software
    Copyright (C) 2017 University of Pennsylvania, Georgetown University

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
END OF LICENSE STUB
*/
#include "worker_thread.h"
#include "local_files.h"
#include "rt_stats.h"
#include "logging.h"
#include "runtime_dfg.h"
#include "controller_communication.h"
#include "output_thread.h"
#include "socket_monitor.h"
#include "profiler.h"
#include "runtime_dfg.h"

#include <stdlib.h>
#include <unistd.h>
#include <stdbool.h>
#include <getopt.h>
#include <libgen.h>
#include <stdio.h>

#define USAGE_ARGS \
    " -j dfg.json -i runtime_id [-l statlog] [-p profiling_prob]"

int main(int argc, char **argv) {

    char *dfg_json = NULL;
    char *statlog = NULL;
    int runtime_id = -1;
    float prof_prob = 0;

    struct option long_options[] = {
        {NULL, 0, 0, 0}
    };

    while (1) {
        int option_index = 0;
        int c = getopt_long(argc, argv, "j:i:l:p:", long_options, &option_index);

        if (c == -1) {
            break;
        }
        switch (c) {
            case 'j':
                dfg_json = optarg;
                break;
            case 'i':
                runtime_id = atoi(optarg);
                break;
            case 'l':
                statlog = optarg;
                break;
            case 'p':
                prof_prob = atof(optarg);
                break;
            default:
                printf("Unknown option provided: %c. Exiting\n", c);
                exit(-1);
        }
    }

    if (dfg_json == NULL || runtime_id == -1) {
        printf("%s " USAGE_ARGS "\n", argv[0]);
        exit(-1);
    }
    set_local_directory(dirname(argv[0]));

    if (init_statistics() != 0) {
        log_warn("Error initializing runtime statistics");
    }
    INIT_PROFILER(prof_prob);

    if (init_runtime_dfg(dfg_json, runtime_id) != 0) {
        log_critical("Error initializing runtime %d  from json file %s. Exiting",
                  runtime_id, dfg_json);
        exit(-1);
    }

    struct sockaddr_in ctrl_addr;
    int rtn = controller_address(&ctrl_addr);
    if (rtn < 0) {
        log_critical("Could not get controller address from DFG");
        exit(-1);
    }

    struct dedos_thread *output_thread = start_output_monitor_thread();
    if (output_thread == NULL) {
        log_critical("Error starting output thread!");
        exit(-1);
    }

    // This will block until the socket monitor exits
    rtn = run_socket_monitor(local_runtime_port(), &ctrl_addr);
    if (rtn < 0) {
        log_critical("Error running socket monitor. Runtime will exit");
    }

    stop_output_monitor();
    stop_all_worker_threads();
    join_output_thread();

    finalize_statistics(statlog);
    free_runtime_dfg();
    log_info("Exiting runtime...");
    return 0;
}