/* * Copyright (c) 2004-2010 The Trustees of Indiana University and Indiana * University Research and Technology * Corporation. All rights reserved. * Copyright (c) 2004-201 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) 2010-201 Oak Ridge National Labs. All rights reserved. * Copyright (c) 201-2014 Cisco Systems, Inc. All rights reserved. * Copyright (c) 201-2013 Los Alamos National Security, LLC. All rights * reserved. * Copyright (c) 2013-2020 Intel, Inc. All rights reserved. * Copyright (c) 2019 Mellanox Technologies, Inc. * All rights reserved. * Copyright (c) 2021-2022 Nanook Consulting. All rights reserved. * $COPYRIGHT$ * * Additional copyrights may follow * * $HEADER$ * */ #include "src/include/pmix_config.h" #include "bfrop_v12.h" #include "internal.h" #include "src/mca/bfrops/base/base.h" static pmix_status_t init(void); static void finalize(void); static const char *data_type_string(pmix_data_type_t type); pmix_bfrops_module_t pmix_bfrops_pmix12_module = { .version = "v12", .init = init, .finalize = finalize, .pack = pmix12_bfrop_pack, .unpack = pmix12_bfrop_unpack, .copy = pmix12_bfrop_copy, .print = pmix12_bfrop_print, .copy_payload = pmix12_bfrop_copy_payload, .value_xfer = pmix12_bfrop_value_xfer, .value_load = pmix12_bfrop_value_load, .value_unload = pmix12_bfrop_value_unload, .value_cmp = pmix12_bfrop_value_cmp, .data_type_string = data_type_string }; static pmix_status_t init(void) { /* some standard types don't require anything special */ PMIX_REGISTER_TYPE("PMIX_BOOL", PMIX_BOOL, pmix12_bfrop_pack_bool, pmix12_bfrop_unpack_bool, pmix12_bfrop_std_copy, pmix12_bfrop_print_bool, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_BYTE", PMIX_BYTE, pmix12_bfrop_pack_byte, pmix12_bfrop_unpack_byte, pmix12_bfrop_std_copy, pmix12_bfrop_print_byte, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_STRING", PMIX_STRING, pmix12_bfrop_pack_string, pmix12_bfrop_unpack_string, pmix12_bfrop_copy_string, pmix12_bfrop_print_string, &pmix_mca_bfrops_v12_component.types); /* Register the rest of the standard generic types to point to internal functions */ PMIX_REGISTER_TYPE("PMIX_SIZE", PMIX_SIZE, pmix12_bfrop_pack_sizet, pmix12_bfrop_unpack_sizet, pmix12_bfrop_std_copy, pmix12_bfrop_print_size, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PID", PMIX_PID, pmix12_bfrop_pack_pid, pmix12_bfrop_unpack_pid, pmix12_bfrop_std_copy, pmix12_bfrop_print_pid, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_INT", PMIX_INT, pmix12_bfrop_pack_int, pmix12_bfrop_unpack_int, pmix12_bfrop_std_copy, pmix12_bfrop_print_int, &pmix_mca_bfrops_v12_component.types); /* Register all the standard fixed types to point to base functions */ PMIX_REGISTER_TYPE("PMIX_INT8", PMIX_INT8, pmix12_bfrop_pack_byte, pmix12_bfrop_unpack_byte, pmix12_bfrop_std_copy, pmix12_bfrop_print_int8, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_INT16", PMIX_INT16, pmix12_bfrop_pack_int16, pmix12_bfrop_unpack_int16, pmix12_bfrop_std_copy, pmix12_bfrop_print_int16, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_INT32", PMIX_INT32, pmix12_bfrop_pack_int32, pmix12_bfrop_unpack_int32, pmix12_bfrop_std_copy, pmix12_bfrop_print_int32, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_INT64", PMIX_INT64, pmix12_bfrop_pack_int64, pmix12_bfrop_unpack_int64, pmix12_bfrop_std_copy, pmix12_bfrop_print_int64, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_UINT", PMIX_UINT, pmix12_bfrop_pack_int, pmix12_bfrop_unpack_int, pmix12_bfrop_std_copy, pmix12_bfrop_print_uint, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_UINT8", PMIX_UINT8, pmix12_bfrop_pack_byte, pmix12_bfrop_unpack_byte, pmix12_bfrop_std_copy, pmix12_bfrop_print_uint8, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_UINT16", PMIX_UINT16, pmix12_bfrop_pack_int16, pmix12_bfrop_unpack_int16, pmix12_bfrop_std_copy, pmix12_bfrop_print_uint16, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_UINT32", PMIX_UINT32, pmix12_bfrop_pack_int32, pmix12_bfrop_unpack_int32, pmix12_bfrop_std_copy, pmix12_bfrop_print_uint32, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_UINT64", PMIX_UINT64, pmix12_bfrop_pack_int64, pmix12_bfrop_unpack_int64, pmix12_bfrop_std_copy, pmix12_bfrop_print_uint64, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_FLOAT", PMIX_FLOAT, pmix12_bfrop_pack_float, pmix12_bfrop_unpack_float, pmix12_bfrop_std_copy, pmix12_bfrop_print_float, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_DOUBLE", PMIX_DOUBLE, pmix12_bfrop_pack_double, pmix12_bfrop_unpack_double, pmix12_bfrop_std_copy, pmix12_bfrop_print_double, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_TIMEVAL", PMIX_TIMEVAL, pmix12_bfrop_pack_timeval, pmix12_bfrop_unpack_timeval, pmix12_bfrop_std_copy, pmix12_bfrop_print_timeval, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_TIME", PMIX_TIME, pmix12_bfrop_pack_time, pmix12_bfrop_unpack_time, pmix12_bfrop_std_copy, pmix12_bfrop_print_time, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_STATUS", PMIX_STATUS, pmix12_bfrop_pack_status, pmix12_bfrop_unpack_status, pmix12_bfrop_std_copy, pmix12_bfrop_print_status, &pmix_mca_bfrops_v12_component.types); /* structured values need to be done here as they might * callback into standard and/or derived values */ PMIX_REGISTER_TYPE("PMIX_VALUE", PMIX_VALUE, pmix12_bfrop_pack_value, pmix12_bfrop_unpack_value, pmix12_bfrop_copy_value, pmix12_bfrop_print_value, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PROC", PMIX_PROC, pmix12_bfrop_pack_proc, pmix12_bfrop_unpack_proc, pmix12_bfrop_copy_proc, pmix12_bfrop_print_proc, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_APP", PMIX_APP, pmix12_bfrop_pack_app, pmix12_bfrop_unpack_app, pmix12_bfrop_copy_app, pmix12_bfrop_print_app, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_INFO", PMIX_INFO, pmix12_bfrop_pack_info, pmix12_bfrop_unpack_info, pmix12_bfrop_copy_info, pmix12_bfrop_print_info, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PDATA", PMIX_PDATA, pmix12_bfrop_pack_pdata, pmix12_bfrop_unpack_pdata, pmix12_bfrop_copy_pdata, pmix12_bfrop_print_pdata, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_BUFFER", PMIX_BUFFER, pmix12_bfrop_pack_buf, pmix12_bfrop_unpack_buf, pmix12_bfrop_copy_buf, pmix12_bfrop_print_buf, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_BYTE_OBJECT", PMIX_BYTE_OBJECT, pmix12_bfrop_pack_bo, pmix12_bfrop_unpack_bo, pmix12_bfrop_copy_bo, pmix12_bfrop_print_bo, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_KVAL", PMIX_KVAL, pmix12_bfrop_pack_kval, pmix12_bfrop_unpack_kval, pmix12_bfrop_copy_kval, pmix12_bfrop_print_kval, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_MODEX", PMIX_MODEX, pmix12_bfrop_pack_modex, pmix12_bfrop_unpack_modex, pmix12_bfrop_copy_modex, pmix12_bfrop_print_modex, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PERSIST", PMIX_PERSIST, pmix12_bfrop_pack_persist, pmix12_bfrop_unpack_persist, pmix12_bfrop_std_copy, pmix12_bfrop_print_persist, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_POINTER", PMIX_POINTER, pmix12_bfrop_pack_ptr, pmix12_bfrop_unpack_ptr, pmix12_bfrop_std_copy, pmix12_bfrop_print_ptr, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_SCOPE", PMIX_SCOPE, pmix12_bfrop_pack_scope, pmix12_bfrop_unpack_scope, pmix12_bfrop_std_copy, pmix12_bfrop_print_scope, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_DATA_RANGE", PMIX_DATA_RANGE, pmix12_bfrop_pack_range, pmix12_bfrop_unpack_range, pmix12_bfrop_std_copy, pmix12_bfrop_print_ptr, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_COMMAND", PMIX_COMMAND, pmix12_bfrop_pack_cmd, pmix12_bfrop_unpack_cmd, pmix12_bfrop_std_copy, pmix12_bfrop_print_cmd, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_INFO_DIRECTIVES", PMIX_INFO_DIRECTIVES, pmix12_bfrop_pack_info_directives, pmix12_bfrop_unpack_info_directives, pmix12_bfrop_std_copy, pmix12_bfrop_print_info_directives, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_DATA_TYPE", PMIX_DATA_TYPE, pmix12_bfrop_pack_datatype, pmix12_bfrop_unpack_datatype, pmix12_bfrop_std_copy, pmix12_bfrop_print_datatype, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PROC_STATE", PMIX_PROC_STATE, pmix12_bfrop_pack_proc_state, pmix12_bfrop_unpack_proc_state, pmix12_bfrop_std_copy, pmix12_bfrop_print_proc_state, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PROC_INFO", PMIX_PROC_INFO, pmix12_bfrop_pack_proc_info, pmix12_bfrop_unpack_proc_info, pmix12_bfrop_copy_proc_info, pmix12_bfrop_print_proc_info, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_DATA_ARRAY", PMIX_DATA_ARRAY, pmix12_bfrop_pack_darray, pmix12_bfrop_unpack_darray, pmix12_bfrop_copy_darray, pmix12_bfrop_print_darray, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_PROC_RANK", PMIX_PROC_RANK, pmix12_bfrop_pack_rank, pmix12_bfrop_unpack_rank, pmix12_bfrop_std_copy, pmix12_bfrop_print_rank, &pmix_mca_bfrops_v12_component.types); PMIX_REGISTER_TYPE("PMIX_QUERY", PMIX_QUERY, pmix12_bfrop_pack_query, pmix12_bfrop_unpack_query, pmix12_bfrop_copy_query, pmix12_bfrop_print_query, &pmix_mca_bfrops_v12_component.types); /**** DEPRECATED ****/ PMIX_REGISTER_TYPE("PMIX_INFO_ARRAY", PMIX_INFO_ARRAY, pmix12_bfrop_pack_array, pmix12_bfrop_unpack_array, pmix12_bfrop_copy_array, pmix12_bfrop_print_array, &pmix_mca_bfrops_v12_component.types); /********************/ return PMIX_SUCCESS; } static void finalize(void) { int n; pmix_bfrop_type_info_t *info; for (n = 0; n < pmix_mca_bfrops_v12_component.types.size; n++) { if (NULL != (info = (pmix_bfrop_type_info_t *) pmix_pointer_array_get_item(&pmix_mca_bfrops_v12_component.types, n))) { PMIX_RELEASE(info); pmix_pointer_array_set_item(&pmix_mca_bfrops_v12_component.types, n, NULL); } } } static const char *data_type_string(pmix_data_type_t type) { pmix_bfrop_type_info_t *info; if (NULL == (info = (pmix_bfrop_type_info_t *) pmix_pointer_array_get_item(&pmix_mca_bfrops_v12_component.types, type))) { return NULL; } return info->odti_name; } int pmix12_v2_to_v1_datatype(pmix_data_type_t v2type) { int v1type; /* I'm passing the data type to * a PMIx v1 compatible peer. The data type was redefined * in v2, and so we have to do some conversions here */ switch (v2type) { case 20: /* the client thinks this is simply an int */ v1type = 6; break; case 44: /* the client thinks this is PMIX_INFO_ARRAY */ v1type = 22; break; case 40: /* proc rank is just an int in v1 */ v1type = 6; break; case 39: /* data arrays must be converted to info arrays */ v1type = 22; break; case 23: case 24: case 25: case 26: case 27: case 28: case 29: case 30: /* shift up one */ v1type = v2type + 1; break; default: v1type = v2type; } return v1type; } pmix_status_t pmix12_bfrop_store_data_type(pmix_pointer_array_t *regtypes, pmix_buffer_t *buffer, pmix_data_type_t type) { int v1type; v1type = pmix12_v2_to_v1_datatype(type); return pmix12_bfrop_pack_datatype(regtypes, buffer, &v1type, 1, PMIX_INT); } pmix_data_type_t pmix12_v1_to_v2_datatype(int v1type) { pmix_data_type_t v2type; /* I'm getting the data type that was given to * me by a PMIx v1 compatible peer. The data type was redefined * in v2, and so we have to do some conversions here */ switch (v1type) { case 20: /* the peer thinks this is PMIX_HWLOC_TOPO, which we don't support */ v2type = PMIX_UNDEF; break; case 22: /* the peer thinks this is PMIX_INFO_ARRAY */ v2type = PMIX_INFO_ARRAY; break; case 23: case 24: case 25: case 26: case 27: case 28: case 29: case 30: case 31: /* shift down one */ v2type = v1type - 1; break; default: v2type = v1type; } return v2type; } pmix_status_t pmix12_bfrop_get_data_type(pmix_pointer_array_t *regtypes, pmix_buffer_t *buffer, pmix_data_type_t *type) { int32_t n = 1; int v1type; pmix_status_t rc; rc = pmix12_bfrop_unpack_datatype(regtypes, buffer, &v1type, &n, PMIX_INT); if (UINT16_MAX < v1type) { *type = 0; return PMIX_ERR_UNKNOWN_DATA_TYPE; } if (PMIX_SUCCESS == rc) { *type = pmix12_v1_to_v2_datatype(v1type); } return rc; } void pmix12_bfrop_value_load(pmix_value_t *v, const void *data, pmix_data_type_t type) { pmix_byte_object_t *bo; v->type = type; if (NULL == data) { /* just set the fields to zero */ memset(&v->data, 0, sizeof(v->data)); } else { switch (type) { case PMIX_UNDEF: break; case PMIX_BOOL: memcpy(&(v->data.flag), data, 1); break; case PMIX_BYTE: memcpy(&(v->data.byte), data, 1); break; case PMIX_STRING: v->data.string = strdup(data); break; case PMIX_SIZE: memcpy(&(v->data.size), data, sizeof(size_t)); break; case PMIX_PID: memcpy(&(v->data.pid), data, sizeof(pid_t)); break; case PMIX_INT: memcpy(&(v->data.integer), data, sizeof(int)); break; case PMIX_INT8: memcpy(&(v->data.int8), data, 1); break; case PMIX_INT16: memcpy(&(v->data.int16), data, 2); break; case PMIX_INT32: memcpy(&(v->data.int32), data, 4); break; case PMIX_INT64: memcpy(&(v->data.int64), data, 8); break; case PMIX_UINT: memcpy(&(v->data.uint), data, sizeof(int)); break; case PMIX_UINT8: memcpy(&(v->data.uint8), data, 1); break; case PMIX_UINT16: memcpy(&(v->data.uint16), data, 2); break; case PMIX_UINT32: memcpy(&(v->data.uint32), data, 4); break; case PMIX_UINT64: memcpy(&(v->data.uint64), data, 8); break; case PMIX_FLOAT: memcpy(&(v->data.fval), data, sizeof(float)); break; case PMIX_DOUBLE: memcpy(&(v->data.dval), data, sizeof(double)); break; case PMIX_TIMEVAL: memcpy(&(v->data.tv), data, sizeof(struct timeval)); break; case PMIX_BYTE_OBJECT: bo = (pmix_byte_object_t *) data; v->data.bo.bytes = bo->bytes; memcpy(&(v->data.bo.size), &bo->size, sizeof(size_t)); break; case PMIX_TIME: case PMIX_VALUE: case PMIX_INFO_ARRAY: case PMIX_APP: case PMIX_INFO: case PMIX_PDATA: case PMIX_BUFFER: case PMIX_KVAL: case PMIX_MODEX: case PMIX_PERSIST: case PMIX_PROC: /* silence warnings */ break; } } } pmix_status_t pmix12_bfrop_value_unload(pmix_value_t *kv, void **data, size_t *sz) { pmix_status_t rc; rc = PMIX_SUCCESS; if (NULL == data || (NULL == *data && PMIX_STRING != kv->type && PMIX_BYTE_OBJECT != kv->type)) { rc = PMIX_ERR_BAD_PARAM; } else { switch (kv->type) { case PMIX_UNDEF: rc = PMIX_ERR_UNKNOWN_DATA_TYPE; break; case PMIX_BOOL: memcpy(*data, &(kv->data.flag), 1); *sz = 1; break; case PMIX_BYTE: memcpy(*data, &(kv->data.byte), 1); *sz = 1; break; case PMIX_STRING: if (NULL != kv->data.string) { *data = strdup(kv->data.string); *sz = strlen(kv->data.string); } break; case PMIX_SIZE: memcpy(*data, &(kv->data.size), sizeof(size_t)); *sz = sizeof(size_t); break; case PMIX_PID: memcpy(*data, &(kv->data.pid), sizeof(pid_t)); *sz = sizeof(pid_t); break; case PMIX_INT: memcpy(*data, &(kv->data.integer), sizeof(int)); *sz = sizeof(int); break; case PMIX_INT8: memcpy(*data, &(kv->data.int8), 1); *sz = 1; break; case PMIX_INT16: memcpy(*data, &(kv->data.int16), 2); *sz = 2; break; case PMIX_INT32: memcpy(*data, &(kv->data.int32), 4); *sz = 4; break; case PMIX_INT64: memcpy(*data, &(kv->data.int64), 8); *sz = 8; break; case PMIX_UINT: memcpy(*data, &(kv->data.uint), sizeof(int)); *sz = sizeof(int); break; case PMIX_UINT8: memcpy(*data, &(kv->data.uint8), 1); *sz = 1; break; case PMIX_UINT16: memcpy(*data, &(kv->data.uint16), 2); *sz = 2; break; case PMIX_UINT32: memcpy(*data, &(kv->data.uint32), 4); *sz = 4; break; case PMIX_UINT64: memcpy(*data, &(kv->data.uint64), 8); *sz = 8; break; case PMIX_FLOAT: memcpy(*data, &(kv->data.fval), sizeof(float)); *sz = sizeof(float); break; case PMIX_DOUBLE: memcpy(*data, &(kv->data.dval), sizeof(double)); *sz = sizeof(double); break; case PMIX_TIMEVAL: memcpy(*data, &(kv->data.tv), sizeof(struct timeval)); *sz = sizeof(struct timeval); break; case PMIX_BYTE_OBJECT: if (NULL != kv->data.bo.bytes && 0 < kv->data.bo.size) { *data = kv->data.bo.bytes; *sz = kv->data.bo.size; } else { *data = NULL; *sz = 0; } break; case PMIX_TIME: case PMIX_VALUE: case PMIX_INFO_ARRAY: case PMIX_APP: case PMIX_INFO: case PMIX_PDATA: case PMIX_BUFFER: case PMIX_KVAL: case PMIX_MODEX: case PMIX_PERSIST: case PMIX_PROC: /* silence warnings */ rc = PMIX_ERROR; break; } } return rc; }