openmpi/orte/mca/snapc/base/base.h

/*
 * Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * Copyright (c) 2007      Evergrid, Inc. All rights reserved.
 *
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */
#ifndef ORTE_SNAPC_BASE_H
#define ORTE_SNAPC_BASE_H

#include "orte_config.h"

#include "orte/mca/rml/rml.h"
#include "orte/mca/gpr/gpr_types.h"
#include "orte/dss/dss.h"

#include "orte/mca/snapc/snapc.h"

/*
 * Global functions for MCA overall SNAPC
 */

#if defined(c_plusplus) || defined(__cplusplus)
extern "C" {
#endif

    /**
     * Global Snapshot Coordinator RML tags
     */
#define ORTE_SNAPC_GLOBAL_INIT_CMD    0x01
#define ORTE_SNAPC_GLOBAL_TERM_CMD    0x02
    
    /**
     * Global Snapshot Object Maintenance functions
     */
    void orte_snapc_base_snapshot_construct(orte_snapc_base_snapshot_t *obj);
    void orte_snapc_base_snapshot_destruct( orte_snapc_base_snapshot_t *obj);

    void orte_snapc_base_global_snapshot_construct(orte_snapc_base_global_snapshot_t *obj);
    void orte_snapc_base_global_snapshot_destruct( orte_snapc_base_global_snapshot_t *obj);

    /**
     * Initialize the SNAPC MCA framework
     *
     * @retval ORTE_SUCCESS Upon success
     * @retval ORTE_ERROR   Upon failures
     * 
     * This function is invoked during orte_init();
     */
    ORTE_DECLSPEC int orte_snapc_base_open(void);
    
    /**
     * Select an available component.
     *
     * @retval ORTE_SUCCESS Upon Success
     * @retval ORTE_NOT_FOUND If no component can be selected
     * @retval ORTE_ERROR Upon other failure
     *
     */
    ORTE_DECLSPEC int orte_snapc_base_select(bool seed, bool app);
    
    /**
     * Finalize the SNAPC MCA framework
     *
     * @retval ORTE_SUCCESS Upon success
     * @retval ORTE_ERROR   Upon failures
     * 
     * This function is invoked during orte_finalize();
     */
    ORTE_DECLSPEC int orte_snapc_base_close(void);

    /**
     * 'None' component functions
     * These are to be used when no component is selected.
     * They just return success, and empty strings as necessary.
     */
    int orte_snapc_base_none_open(void);
    int orte_snapc_base_none_close(void);

    int orte_snapc_base_module_init(bool seed, bool app);
    int orte_snapc_base_module_finalize(void);
    int orte_snapc_base_none_setup_job(orte_jobid_t jobid);
    int orte_snapc_base_none_release_job(orte_jobid_t jobid);

    ORTE_DECLSPEC extern int  orte_snapc_base_output;
    ORTE_DECLSPEC extern opal_list_t orte_snapc_base_components_available;
    ORTE_DECLSPEC extern orte_snapc_base_component_t orte_snapc_base_selected_component;
    ORTE_DECLSPEC extern orte_snapc_base_module_t orte_snapc;

    /**
     * Globals
     */
#define orte_snapc_base_metadata_filename (strdup("global_snapshot_meta.data"))

    ORTE_DECLSPEC extern char * orte_snapc_base_global_snapshot_dir;
    ORTE_DECLSPEC extern char * orte_snapc_base_global_snapshot_ref;
    ORTE_DECLSPEC extern char * orte_snapc_base_global_snapshot_loc;
    ORTE_DECLSPEC extern bool   orte_snapc_base_store_in_place;
    ORTE_DECLSPEC extern bool   orte_snapc_base_store_only_one_seq;
    ORTE_DECLSPEC extern bool   orte_snapc_base_establish_gloabl_snapshot_dir;
    ORTE_DECLSPEC extern size_t orte_snapc_base_snapshot_seq_number;


    /**
     * Some utility functions
     */
    ORTE_DECLSPEC char * orte_snapc_ckpt_state_str(size_t state);

    ORTE_DECLSPEC char * orte_snapc_base_unique_global_snapshot_name(pid_t pid);
    ORTE_DECLSPEC char * orte_snapc_base_get_global_snapshot_metadata_file(char *uniq_snapshot_name);
    ORTE_DECLSPEC char * orte_snapc_base_get_global_snapshot_directory(char *uniq_global_snapshot_name);
    ORTE_DECLSPEC int    orte_snapc_base_init_global_snapshot_directory(char *uniq_global_snapshot_name,
                                                                        bool empty_metadata);
    ORTE_DECLSPEC int    orte_snapc_base_get_job_ckpt_info( orte_jobid_t jobid,
                                                            size_t *ckpt_state,
                                                            char **ckpt_snapshot_ref,
                                                            char **ckpt_snapshot_loc);
    ORTE_DECLSPEC int    orte_snapc_base_set_job_ckpt_info( orte_jobid_t jobid,
                                                            size_t ckpt_state,
                                                            char  *ckpt_snapshot_ref,
                                                            char  *ckpt_snapshot_loc);
    ORTE_DECLSPEC int    orte_snapc_base_get_vpid_ckpt_info( orte_process_name_t proc,
                                                             size_t *ckpt_state,
                                                             char **ckpt_ref,
                                                             char **ckpt_loc);
    ORTE_DECLSPEC int    orte_snapc_base_set_vpid_ckpt_info( orte_process_name_t proc,
                                                             size_t ckpt_state,
                                                             char *ckpt_ref,
                                                             char *ckpt_loc);
    ORTE_DECLSPEC int orte_snapc_base_add_timestamp(char * global_snapshot_ref);
    ORTE_DECLSPEC int orte_snapc_base_add_vpid_metadata(orte_process_name_t *proc,
                                                        char * global_snapshot_ref,
                                                        char *snapshot_ref,
                                                        char *snapshot_location);
    ORTE_DECLSPEC int orte_snapc_base_finalize_metadata(char * global_snapshot_ref);
    ORTE_DECLSPEC int orte_snapc_base_extract_metadata(orte_snapc_base_global_snapshot_t *snapshot);

    /*******************************
     * Global Coordinator functions
     *******************************/
    /* Initial handshake with the orte_checkpoint command */
    ORTE_DECLSPEC int orte_snapc_base_global_coord_ckpt_init_cmd(orte_process_name_t* peer, bool *term, orte_jobid_t *jobid);
    ORTE_DECLSPEC int orte_snapc_base_global_coord_ckpt_update_cmd(orte_process_name_t* peer, char *global_snapshot_handle, int seq_num, int ckpt_status);

    ORTE_DECLSPEC int orte_snapc_base_global_coord_recv_ack(orte_process_name_t* peer, bool *ack);
    ORTE_DECLSPEC int orte_snapc_base_global_coord_send_ack(orte_process_name_t* peer, bool  ack);

    ORTE_DECLSPEC int orte_snapc_base_global_init_request(orte_jobid_t jobid,
                                                          orte_rml_buffer_callback_fn_t rml_cbfunc, void* rml_cbdata,
                                                          orte_gpr_notify_cb_fn_t       gpr_cbfunc, void* gpr_cbdata);

    ORTE_DECLSPEC int orte_snapc_base_extract_gpr_vpid_ckpt_info(orte_gpr_notify_data_t *data,
                                                                 orte_process_name_t **proc,
                                                                 size_t *ckpt_state,
                                                                 char **ckpt_snapshot_ref,
                                                                 char **ckpt_snapshot_loc);
    ORTE_DECLSPEC int orte_snapc_base_extract_gpr_job_ckpt_info(orte_gpr_notify_data_t *data,
                                                                size_t *ckpt_state,
                                                                char **ckpt_snapshot_ref,
                                                                char **ckpt_snapshot_loc);

#if defined(c_plusplus) || defined(__cplusplus)
}
#endif

#endif /* ORTE_SNAPC_BASE_H */