diff options
Diffstat (limited to 'src/H5VLdaosm.c')
-rw-r--r-- | src/H5VLdaosm.c | 911 |
1 files changed, 900 insertions, 11 deletions
diff --git a/src/H5VLdaosm.c b/src/H5VLdaosm.c index d3e4589..0e883be 100644 --- a/src/H5VLdaosm.c +++ b/src/H5VLdaosm.c @@ -30,6 +30,7 @@ #include "H5FDprivate.h" /* File drivers */ #include "H5FFprivate.h" /* Fast Forward */ #include "H5Iprivate.h" /* IDs */ +#include "H5Mprivate.h" /* Maps */ #include "H5MMprivate.h" /* Memory management */ #include "H5Opkg.h" /* Objects */ #include "H5Pprivate.h" /* Property lists */ @@ -53,6 +54,10 @@ hid_t H5VL_DAOSM_g = -1; #define H5VL_DAOSM_ATTR_KEY "/Attribute" #define H5VL_DAOSM_CHUNK_KEY 0u +#define H5VL_DAOSM_KTYPE_KEY "Key Datatype" +#define H5VL_DAOSM_VTYPE_KEY "Value Datatype" +#define H5VL_DAOSM_MAP_KEY "MAP_AKEY" + /* Stack allocation sizes */ #define H5VL_DAOSM_GH_BUF_SIZE 1024 #define H5VL_DAOSM_FOI_BUF_SIZE 1024 @@ -69,6 +74,7 @@ hid_t H5VL_DAOSM_g = -1; #define H5VL_DAOSM_TYPE_GRP 0x0000000000000000ull #define H5VL_DAOSM_TYPE_DSET 0x4000000000000000ull #define H5VL_DAOSM_TYPE_DTYPE 0x8000000000000000ull +#define H5VL_DAOSM_TYPE_MAP 0xc000000000000000ull /* DAOSM-specific file access properties */ typedef struct H5VL_daosm_fapl_t { @@ -212,6 +218,7 @@ static herr_t H5VL_daosm_object_close(void *_obj, hid_t dxpl_id, void **req); H5FL_DEFINE(H5VL_daosm_file_t); H5FL_DEFINE(H5VL_daosm_group_t); H5FL_DEFINE(H5VL_daosm_dset_t); +H5FL_DEFINE(H5VL_daosm_map_t); H5FL_DEFINE(H5VL_daosm_attr_t); /* The DAOS-M VOL plugin struct */ @@ -511,6 +518,8 @@ H5VL_daosm_init(void) * initialize that type) */ if(H5G_init() < 0) HGOTO_ERROR(H5E_FUNC, H5E_CANTINIT, FAIL, "unable to initialize group interface") + if(H5M_init() < 0) + HGOTO_ERROR(H5E_FUNC, H5E_CANTINIT, FAIL, "unable to initialize map interface") if(H5D_init() < 0) HGOTO_ERROR(H5E_FUNC, H5E_CANTINIT, FAIL, "unable to initialize dataset interface") if(H5T_init() < 0) @@ -840,9 +849,11 @@ H5VL_daosm_oid_encode(daos_obj_id_t *oid, uint64_t idx, H5I_type_t obj_type) type_bits = H5VL_DAOSM_TYPE_GRP; else if(obj_type == H5I_DATASET) type_bits = H5VL_DAOSM_TYPE_DSET; - else { - HDassert(obj_type == H5I_DATATYPE); + else if(obj_type == H5I_DATATYPE) type_bits = H5VL_DAOSM_TYPE_DTYPE; + else { + HDassert(obj_type == H5I_MAP); + type_bits = H5VL_DAOSM_TYPE_MAP; } /* end else */ /* Encode type and address and generate oid */ @@ -866,6 +877,8 @@ H5VL_daosm_addr_to_type(uint64_t addr) return(H5I_DATASET); else if(type_bits == H5VL_DAOSM_TYPE_DTYPE) return(H5I_DATATYPE); + else if(type_bits == H5VL_DAOSM_TYPE_MAP) + return(H5I_MAP); else return(H5I_BADID); } /* end H5VL_daosm_oid_to_type() */ @@ -3470,7 +3483,7 @@ H5VL_daosm_dataset_create(void *_item, /* Check for collective access, if not already set by the file */ if(!collective) if(H5Pget_all_coll_metadata_ops(dapl_id, &collective) < 0) - HGOTO_ERROR(H5E_SYM, H5E_CANTGET, NULL, "can't get collective access property") + HGOTO_ERROR(H5E_DATASET, H5E_CANTGET, NULL, "can't get collective access property") /* Get the dcpl plist structure */ if(NULL == (plist = (H5P_genplist_t *)H5I_object(dcpl_id))) @@ -3738,8 +3751,8 @@ H5VL_daosm_dataset_open(void *_item, if(0 != (ret = daos_obj_open(item->file->coh, dset->obj.oid, item->file->epoch, item->file->flags & H5F_ACC_RDWR ? DAOS_COO_RW : DAOS_COO_RO, &dset->obj.obj_oh, NULL /*event*/))) HGOTO_ERROR(H5E_DATASET, H5E_CANTOPENOBJ, NULL, "can't open dataset: %d", ret) - /* Set up operation to read datatype, dataspace, and DCPL sizes from dataset - */ + /* Set up operation to read datatype, dataspace, and DCPL sizes from + * dataset */ /* Set up dkey */ daos_iov_set(&dkey, int_md_key, (daos_size_t)(sizeof(int_md_key) - 1)); @@ -4772,13 +4785,19 @@ H5VL_daosm_object_open(void *_item, H5VL_loc_params_t loc_params, ? loc_params.loc_data.loc_by_name.lapl_id : H5P_DATASET_ACCESS_DEFAULT, dxpl_id, req))) HGOTO_ERROR(H5E_OHDR, H5E_CANTOPENOBJ, NULL, "can't open dataset") } /* end if */ - else { - HDassert(obj_type == H5I_DATATYPE); + else if(obj_type == H5I_DATATYPE) { if(NULL == (obj = (H5VL_daosm_obj_t *)H5VL_daosm_datatype_open(item, sub_loc_params, NULL, ((H5VL_OBJECT_BY_NAME == loc_params.type) && (loc_params.loc_data.loc_by_name.lapl_id != H5P_DEFAULT)) ? loc_params.loc_data.loc_by_name.lapl_id : H5P_DATATYPE_ACCESS_DEFAULT, dxpl_id, req))) HGOTO_ERROR(H5E_OHDR, H5E_CANTOPENOBJ, NULL, "can't open datatype") } /* end if */ + else { + HDassert(obj_type == H5I_MAP); + if(NULL == (obj = (H5VL_daosm_obj_t *)H5VL_daosm_map_open(item, sub_loc_params, NULL, + ((H5VL_OBJECT_BY_NAME == loc_params.type) && (loc_params.loc_data.loc_by_name.lapl_id != H5P_DEFAULT)) + ? loc_params.loc_data.loc_by_name.lapl_id : H5P_MAP_ACCESS_DEFAULT, dxpl_id, req))) + HGOTO_ERROR(H5E_OHDR, H5E_CANTOPENOBJ, NULL, "can't open map") + } /* end if */ /* Set return value */ if(opened_type) @@ -4880,9 +4899,11 @@ H5VL_daosm_object_optional(void *_item, hid_t dxpl_id, void **req, obj_info->type = H5O_TYPE_GROUP; else if(target_obj->item.type == H5I_DATASET) obj_info->type = H5O_TYPE_DATASET; - else { - HDassert(target_obj->item.type == H5I_DATATYPE); + else if(target_obj->item.type == H5I_DATATYPE) obj_info->type = H5O_TYPE_NAMED_DATATYPE; + else { + HDassert(target_obj->item.type == H5I_MAP); + obj_info->type = H5O_TYPE_MAP; } /* end else */ /* Reference count is always 1 - change this when @@ -4935,7 +4956,7 @@ H5VL_daosm_object_close(void *_obj, hid_t dxpl_id, void **req) HDassert(obj); HDassert(obj->item.type == H5I_GROUP || obj->item.type == H5I_DATASET - || obj->item.type == H5I_DATATYPE); + || obj->item.type == H5I_DATATYPE || obj->item.type == H5I_MAP); /* Call type's close function */ if(obj->item.type == H5I_GROUP) { @@ -4950,6 +4971,10 @@ H5VL_daosm_object_close(void *_obj, hid_t dxpl_id, void **req) if(H5VL_daosm_datatype_close(obj, dxpl_id, req)) HGOTO_ERROR(H5E_DATATYPE, H5E_CLOSEERROR, FAIL, "can't close datatype") } /* end if */ + else if(obj->item.type == H5I_MAP) { + if(H5VL_daosm_map_close(obj, dxpl_id, req)) + HGOTO_ERROR(H5E_MAP, H5E_CLOSEERROR, FAIL, "can't close map") + } /* end if */ else HDassert(0 && "Invalid object type"); @@ -5589,7 +5614,7 @@ H5VL_daosm_attribute_get(void *_item, H5VL_attr_get_t get_type, H5VL_loc_params_t loc_params = va_arg(arguments, H5VL_loc_params_t); size_t buf_size = va_arg(arguments, size_t); char *buf = va_arg(arguments, char *); - ssize_t *ret_val = va_arg(arguments, ssize_t *); + ssize_t *ret_val = va_arg(arguments, ssize_t *); H5VL_daosm_attr_t *attr = (H5VL_daosm_attr_t *)_item; if(H5VL_OBJECT_BY_SELF == loc_params.type) { @@ -5897,3 +5922,867 @@ H5VL_daosm_attribute_close(void *_attr, hid_t dxpl_id, void **req) FUNC_LEAVE_NOAPI(ret_value) } /* end H5VL_daosm_attribute_close() */ + +/*------------------------------------------------------------------------- + * Function: H5VL_daosm_map_create + * + * Purpose: Sends a request to DAOS-M to create a map + * + * Return: Success: map object. + * Failure: NULL + * + *------------------------------------------------------------------------- + */ +void * +H5VL_daosm_map_create(void *_item, H5VL_loc_params_t H5_ATTR_UNUSED loc_params, + const char *name, hid_t ktype_id, hid_t vtype_id, + hid_t H5_ATTR_UNUSED mcpl_id, hid_t mapl_id, hid_t dxpl_id, void **req) +{ + H5VL_daosm_item_t *item = (H5VL_daosm_item_t *)_item; + H5VL_daosm_map_t *map = NULL; + H5VL_daosm_group_t *target_grp = NULL; + void *ktype_buf = NULL; + void *vtype_buf = NULL; + hbool_t collective = item->file->collective; + int ret; + void *ret_value = NULL; + + FUNC_ENTER_NOAPI_NOINIT + + /* Check for write access */ + if(!(item->file->flags & H5F_ACC_RDWR)) + HGOTO_ERROR(H5E_FILE, H5E_BADVALUE, NULL, "no write intent on file") + + /* Check for collective access, if not already set by the file */ + if(!collective) + if(H5Pget_all_coll_metadata_ops(mapl_id, &collective) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, NULL, "can't get collective access property") + + /* Allocate the map object that is returned to the user */ + if(NULL == (map = H5FL_CALLOC(H5VL_daosm_map_t))) + HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, NULL, "can't allocate DAOS-M map struct") + map->obj.item.type = H5I_MAP; + map->obj.item.file = item->file; + map->obj.item.rc = 1; + map->obj.obj_oh = DAOS_HDL_INVAL; + map->ktype_id = FAIL; + map->vtype_id = FAIL; + + /* Generate map oid */ + H5VL_daosm_oid_encode(&map->obj.oid, item->file->max_oid + (uint64_t)1, H5I_MAP); + + /* Create map and write metadata if this process should */ + if(!collective || (item->file->my_rank == 0)) { + const char *target_name = NULL; + H5VL_daosm_link_val_t link_val; + daos_key_t dkey; + daos_iod_t iod[2]; + daos_sg_list_t sgl[2]; + daos_iov_t sg_iov[2]; + size_t ktype_size = 0; + size_t vtype_size = 0; + char int_md_key[] = H5VL_DAOSM_INT_MD_KEY; + char ktype_key[] = H5VL_DAOSM_KTYPE_KEY; + char vtype_key[] = H5VL_DAOSM_VTYPE_KEY; + + /* Traverse the path */ + if(NULL == (target_grp = H5VL_daosm_group_traverse(item, name, dxpl_id, + req, &target_name, NULL, NULL))) + HGOTO_ERROR(H5E_MAP, H5E_BADITER, NULL, "can't traverse path") + + /* Create map */ + /* Update max_oid */ + item->file->max_oid = H5VL_daosm_oid_to_idx(map->obj.oid); + + /* Write max OID */ + if(H5VL_daosm_write_max_oid(item->file) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTINIT, NULL, "can't write max OID") + + /* Open map */ + if(0 != (ret = daos_obj_open(item->file->coh, map->obj.oid, item->file->epoch, DAOS_OO_RW, &map->obj.obj_oh, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTOPENOBJ, NULL, "can't open map: %d", ret) + + /* Encode datatypes */ + if(H5Tencode(ktype_id, NULL, &ktype_size) < 0) + HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, NULL, "can't determine serialized length of datatype") + if(NULL == (ktype_buf = H5MM_malloc(ktype_size))) + HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, NULL, "can't allocate buffer for serialized datatype") + if(H5Tencode(ktype_id, ktype_buf, &ktype_size) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTENCODE, NULL, "can't serialize datatype") + + if(H5Tencode(vtype_id, NULL, &vtype_size) < 0) + HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, NULL, "can't determine serialized length of datatype") + if(NULL == (vtype_buf = H5MM_malloc(vtype_size))) + HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, NULL, "can't allocate buffer for serialized datatype") + if(H5Tencode(vtype_id, vtype_buf, &vtype_size) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTENCODE, NULL, "can't serialize datatype") + + /* Eventually we will want to store the MCPL in the file, and hold + * copies of the MCP and MAPL in memory. To do this look at the dataset + * and group code for examples. -NAF */ + + /* Set up operation to write datatypes to map */ + /* Set up dkey */ + daos_iov_set(&dkey, int_md_key, (daos_size_t)(sizeof(int_md_key) - 1)); + + /* Set up iod */ + HDmemset(iod, 0, sizeof(iod)); + daos_iov_set(&iod[0].iod_name, (void *)ktype_key, (daos_size_t)(sizeof(ktype_key) - 1)); + daos_csum_set(&iod[0].iod_kcsum, NULL, 0); + iod[0].iod_nr = 1u; + iod[0].iod_size = (uint64_t)ktype_size; + iod[0].iod_type = DAOS_IOD_SINGLE; + + daos_iov_set(&iod[1].iod_name, (void *)vtype_key, (daos_size_t)(sizeof(vtype_key) - 1)); + daos_csum_set(&iod[1].iod_kcsum, NULL, 0); + iod[1].iod_nr = 1u; + iod[1].iod_size = (uint64_t)vtype_size; + iod[1].iod_type = DAOS_IOD_SINGLE; + + /* Set up sgl */ + daos_iov_set(&sg_iov[0], ktype_buf, (daos_size_t)ktype_size); + sgl[0].sg_nr.num = 1; + sgl[0].sg_iovs = &sg_iov[0]; + daos_iov_set(&sg_iov[1], vtype_buf, (daos_size_t)vtype_size); + sgl[1].sg_nr.num = 1; + sgl[1].sg_iovs = &sg_iov[1]; + + /* Write internal metadata to map */ + if(0 != (ret = daos_obj_update(map->obj.obj_oh, item->file->epoch, &dkey, 2, iod, sgl, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTINIT, NULL, "can't write metadata to map: %d", ret) + + /* Create link to map */ + link_val.type = H5L_TYPE_HARD; + link_val.target.hard = map->obj.oid; + if(H5VL_daosm_link_write(target_grp, target_name, HDstrlen(target_name), &link_val) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTINIT, NULL, "can't create link to map") + } /* end if */ + else { + /* Update max_oid */ + item->file->max_oid = map->obj.oid.lo; + + /* Open map */ + if(0 != (ret = daos_obj_open(item->file->coh, map->obj.oid, item->file->epoch, DAOS_OO_RW, &map->obj.obj_oh, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTOPENOBJ, NULL, "can't open map: %d", ret) + } /* end else */ + + /* Finish setting up map struct */ + if((map->ktype_id = H5Tcopy(ktype_id)) < 0) + HGOTO_ERROR(H5E_SYM, H5E_CANTCOPY, NULL, "failed to copy datatype") + if((map->vtype_id = H5Tcopy(vtype_id)) < 0) + HGOTO_ERROR(H5E_SYM, H5E_CANTCOPY, NULL, "failed to copy datatype") + + /* Set return value */ + ret_value = (void *)map; + +done: + /* Close target group */ + if(target_grp && H5VL_daosm_group_close(target_grp, dxpl_id, req) < 0) + HDONE_ERROR(H5E_MAP, H5E_CLOSEERROR, NULL, "can't close group") + + /* Cleanup on failure */ + /* Destroy DAOS object if created before failure DSMINC */ + if(NULL == ret_value) + /* Close map */ + if(map && H5VL_daosm_map_close(map, dxpl_id, req) < 0) + HDONE_ERROR(H5E_MAP, H5E_CLOSEERROR, NULL, "can't close map"); + + /* Free memory */ + ktype_buf = H5MM_xfree(ktype_buf); + vtype_buf = H5MM_xfree(vtype_buf); + + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_create() */ + + +/*------------------------------------------------------------------------- + * Function: H5VL_daosm_map_open + * + * Purpose: Sends a request to DAOS-M to open a map + * + * Return: Success: map object. + * Failure: NULL + * + *------------------------------------------------------------------------- + */ +void * +H5VL_daosm_map_open(void *_item, H5VL_loc_params_t loc_params, const char *name, + hid_t mapl_id, hid_t dxpl_id, void **req) +{ + H5VL_daosm_item_t *item = (H5VL_daosm_item_t *)_item; + H5VL_daosm_map_t *map = NULL; + H5VL_daosm_group_t *target_grp = NULL; + const char *target_name = NULL; + daos_key_t dkey; + daos_iod_t iod[2]; + daos_sg_list_t sgl[2]; + daos_iov_t sg_iov[2]; + uint64_t ktype_len = 0; + uint64_t vtype_len = 0; + uint64_t tot_len; + uint8_t minfo_buf_static[H5VL_DAOSM_DINFO_BUF_SIZE]; + uint8_t *minfo_buf_dyn = NULL; + uint8_t *minfo_buf = minfo_buf_static; + char int_md_key[] = H5VL_DAOSM_INT_MD_KEY; + char ktype_key[] = H5VL_DAOSM_KTYPE_KEY; + char vtype_key[] = H5VL_DAOSM_VTYPE_KEY; + uint8_t *p; + hbool_t collective = item->file->collective; + hbool_t must_bcast = FALSE; + int ret; + void *ret_value = NULL; + + FUNC_ENTER_NOAPI_NOINIT + + /* Check for collective access, if not already set by the file */ + if(!collective) + if(H5Pget_all_coll_metadata_ops(mapl_id, &collective) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, NULL, "can't get collective access property") + + /* Allocate the map object that is returned to the user */ + if(NULL == (map = H5FL_CALLOC(H5VL_daosm_map_t))) + HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, NULL, "can't allocate DAOS-M map struct") + map->obj.item.type = H5I_MAP; + map->obj.item.file = item->file; + map->obj.item.rc = 1; + map->obj.obj_oh = DAOS_HDL_INVAL; + map->ktype_id = FAIL; + map->vtype_id = FAIL; + + /* Check if we're actually opening the group or just receiving the map + * info from the leader */ + if(!collective || (item->file->my_rank == 0)) { + if(collective && (item->file->num_procs > 1)) + must_bcast = TRUE; + + /* Check for open by address */ + if(H5VL_OBJECT_BY_ADDR == loc_params.type) { + /* Generate oid from address */ + H5VL_daosm_oid_generate(&map->obj.oid, (uint64_t)loc_params.loc_data.loc_by_addr.addr, H5I_MAP); + } /* end if */ + else { + /* Open using name parameter */ + /* Traverse the path */ + if(NULL == (target_grp = H5VL_daosm_group_traverse(item, name, dxpl_id, req, &target_name, NULL, NULL))) + HGOTO_ERROR(H5E_MAP, H5E_BADITER, NULL, "can't traverse path") + + /* Follow link to map */ + if(H5VL_daosm_link_follow(target_grp, target_name, HDstrlen(target_name), dxpl_id, req, &map->obj.oid) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTINIT, NULL, "can't follow link to map") + } /* end else */ + + /* Open map */ + if(0 != (ret = daos_obj_open(item->file->coh, map->obj.oid, item->file->epoch, item->file->flags & H5F_ACC_RDWR ? DAOS_COO_RW : DAOS_COO_RO, &map->obj.obj_oh, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTOPENOBJ, NULL, "can't open map: %d", ret) + + /* Set up operation to read datatype sizes from map */ + /* Set up dkey */ + daos_iov_set(&dkey, int_md_key, (daos_size_t)(sizeof(int_md_key) - 1)); + + /* Set up iod */ + HDmemset(iod, 0, sizeof(iod)); + daos_iov_set(&iod[0].iod_name, (void *)ktype_key, (daos_size_t)(sizeof(ktype_key) - 1)); + daos_csum_set(&iod[0].iod_kcsum, NULL, 0); + iod[0].iod_nr = 1u; + iod[0].iod_size = DAOS_REC_ANY; + iod[0].iod_type = DAOS_IOD_SINGLE; + + daos_iov_set(&iod[1].iod_name, (void *)vtype_key, (daos_size_t)(sizeof(vtype_key) - 1)); + daos_csum_set(&iod[1].iod_kcsum, NULL, 0); + iod[1].iod_nr = 1u; + iod[1].iod_size = DAOS_REC_ANY; + iod[1].iod_type = DAOS_IOD_SINGLE; + + /* Read internal metadata sizes from map */ + if(0 != (ret = daos_obj_fetch(map->obj.obj_oh, item->file->epoch, &dkey, 2, iod, NULL, + NULL /*maps*/, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTDECODE, NULL, "can't read metadata sizes from map: %d", ret) + + /* Check for metadata not found */ + if((iod[0].iod_size == (uint64_t)0) || (iod[1].iod_size == (uint64_t)0)) + HGOTO_ERROR(H5E_MAP, H5E_NOTFOUND, NULL, "internal metadata not found"); + + /* Compute map info buffer size */ + ktype_len = iod[0].iod_size; + vtype_len = iod[1].iod_size; + tot_len = ktype_len + vtype_len; + + /* Allocate map info buffer if necessary */ + if(tot_len > sizeof(minfo_buf_static)) { + if(NULL == (minfo_buf_dyn = (uint8_t *)H5MM_malloc(tot_len + (5 * sizeof(uint64_t))))) + HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, NULL, "can't allocate map info buffer") + minfo_buf = minfo_buf_dyn; + } /* end if */ + + /* Set up sgl */ + p = minfo_buf + (5 * sizeof(uint64_t)); + daos_iov_set(&sg_iov[0], p, (daos_size_t)ktype_len); + sgl[0].sg_nr.num = 1; + sgl[0].sg_iovs = &sg_iov[0]; + p += ktype_len; + daos_iov_set(&sg_iov[1], p, (daos_size_t)vtype_len); + sgl[1].sg_nr.num = 1; + sgl[1].sg_iovs = &sg_iov[1]; + + /* Read internal metadata from map */ + if(0 != (ret = daos_obj_fetch(map->obj.obj_oh, item->file->epoch, &dkey, 2, iod, sgl, NULL /*maps*/, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTDECODE, NULL, "can't read metadata from map: %d", ret) + + /* Broadcast map info if there are other processes that need it */ + if(collective && (item->file->num_procs > 1)) { + HDassert(minfo_buf); + HDassert(sizeof(minfo_buf_static) >= 5 * sizeof(uint64_t)); + + /* Encode oid */ + p = minfo_buf; + UINT64ENCODE(p, map->obj.oid.lo) + UINT64ENCODE(p, map->obj.oid.mid) + UINT64ENCODE(p, map->obj.oid.hi) + + /* Encode serialized info lengths */ + UINT64ENCODE(p, ktype_len) + UINT64ENCODE(p, vtype_len) + + /* MPI_Bcast minfo_buf */ + if(MPI_SUCCESS != MPI_Bcast((char *)minfo_buf, sizeof(minfo_buf_static), MPI_BYTE, 0, item->file->comm)) + HGOTO_ERROR(H5E_MAP, H5E_MPI, NULL, "can't bcast map info"); + + /* Need a second bcast if it did not fit in the receivers' static + * buffer */ + if(tot_len + (5 * sizeof(uint64_t)) > sizeof(minfo_buf_static)) + if(MPI_SUCCESS != MPI_Bcast((char *)p, (int)tot_len, MPI_BYTE, 0, item->file->comm)) + HGOTO_ERROR(H5E_MAP, H5E_MPI, NULL, "can't bcast map info (second bcast)") + } /* end if */ + else + p = minfo_buf + (5 * sizeof(uint64_t)); + } /* end if */ + else { + /* Receive map info */ + if(MPI_SUCCESS != MPI_Bcast((char *)minfo_buf, sizeof(minfo_buf_static), MPI_BYTE, 0, item->file->comm)) + HGOTO_ERROR(H5E_MAP, H5E_MPI, NULL, "can't bcast map info") + + /* Decode oid */ + p = minfo_buf_static; + UINT64DECODE(p, map->obj.oid.lo) + UINT64DECODE(p, map->obj.oid.mid) + UINT64DECODE(p, map->obj.oid.hi) + + /* Decode serialized info lengths */ + UINT64DECODE(p, ktype_len) + UINT64DECODE(p, vtype_len) + tot_len = ktype_len + vtype_len; + + /* Check for type_len set to 0 - indicates failure */ + if(ktype_len == 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTINIT, NULL, "lead process failed to open map") + + /* Check if we need to perform another bcast */ + if(tot_len + (5 * sizeof(uint64_t)) > sizeof(minfo_buf_static)) { + /* Allocate a dynamic buffer if necessary */ + if(tot_len > sizeof(minfo_buf_static)) { + if(NULL == (minfo_buf_dyn = (uint8_t *)H5MM_malloc(tot_len))) + HGOTO_ERROR(H5E_RESOURCE, H5E_CANTALLOC, NULL, "can't allocate space for map info") + minfo_buf = minfo_buf_dyn; + } /* end if */ + + /* Receive map info */ + if(MPI_SUCCESS != MPI_Bcast((char *)minfo_buf, (int)tot_len, MPI_BYTE, 0, item->file->comm)) + HGOTO_ERROR(H5E_MAP, H5E_MPI, NULL, "can't bcast map info (second bcast)") + + p = minfo_buf; + } /* end if */ + + /* Open map */ + if(0 != (ret = daos_obj_open(item->file->coh, map->obj.oid, item->file->epoch, item->file->flags & H5F_ACC_RDWR ? DAOS_COO_RW : DAOS_COO_RO, &map->obj.obj_oh, NULL /*event*/))) + HGOTO_ERROR(H5E_MAP, H5E_CANTOPENOBJ, NULL, "can't open map: %d", ret) + } /* end else */ + + /* Decode datatype, dataspace, and DCPL */ + if((map->ktype_id = H5Tdecode(p)) < 0) + HGOTO_ERROR(H5E_ARGS, H5E_CANTDECODE, NULL, "can't deserialize datatype") + p += ktype_len; + if((map->vtype_id = H5Tdecode(p)) < 0) + HGOTO_ERROR(H5E_ARGS, H5E_CANTDECODE, NULL, "can't deserialize datatype") + + /* Set return value */ + ret_value = (void *)map; + +done: + /* Cleanup on failure */ + if(NULL == ret_value) { + /* Bcast minfo_buf as '0' if necessary - this will trigger failures in + * in other processes so we do not need to do the second bcast. */ + if(must_bcast) { + HDmemset(minfo_buf_static, 0, sizeof(minfo_buf_static)); + if(MPI_SUCCESS != MPI_Bcast(minfo_buf_static, sizeof(minfo_buf_static), MPI_BYTE, 0, item->file->comm)) + HDONE_ERROR(H5E_MAP, H5E_MPI, NULL, "can't bcast empty map info") + } /* end if */ + + /* Close map */ + if(map && H5VL_daosm_map_close(map, dxpl_id, req) < 0) + HDONE_ERROR(H5E_MAP, H5E_CLOSEERROR, NULL, "can't close map") + } /* end if */ + + /* Close target group */ + if(target_grp && H5VL_daosm_group_close(target_grp, dxpl_id, req) < 0) + HDONE_ERROR(H5E_MAP, H5E_CLOSEERROR, NULL, "can't close group") + + /* Free memory */ + minfo_buf_dyn = (uint8_t *)H5MM_xfree(minfo_buf_dyn); + + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_open() */ + + +/*------------------------------------------------------------------------- + * Function: H5VL_daos_map_get_size + * + * Purpose: Retrieves the size of a Key or Value binary + * buffer given its datatype and buffer contents. + * + * Return: Success: SUCCEED + * Failure: Negative + * + *------------------------------------------------------------------------- + */ +static herr_t +H5VL_daos_map_get_size(hid_t type_id, const void *buf, + /*out*/uint64_t H5_ATTR_UNUSED *checksum, /*out*/size_t *size, + /*out*/H5T_class_t *ret_class) +{ + size_t buf_size = 0; + H5T_t *dt = NULL; + H5T_class_t dt_class; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT + + if(NULL == (dt = (H5T_t *)H5I_object_verify(type_id, H5I_DATATYPE))) + HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, H5T_NO_CLASS, "not a datatype") + + dt_class = H5T_get_class(dt, FALSE); + + switch(dt_class) { + case H5T_STRING: + /* If this is a variable length string, get the size using strlen(). */ + if(H5T_is_variable_str(dt)) { + buf_size = HDstrlen((const char*)buf) + 1; + + break; + } + case H5T_INTEGER: + case H5T_FLOAT: + case H5T_TIME: + case H5T_BITFIELD: + case H5T_OPAQUE: + case H5T_ENUM: + case H5T_ARRAY: + case H5T_NO_CLASS: + case H5T_REFERENCE: + case H5T_NCLASSES: + case H5T_COMPOUND: + /* Data is not variable length, so use H5Tget_size() */ + /* MSC - This is not correct. Compound/Array can contian + VL datatypes, but for now we don't support that. Need + to check for that too */ + buf_size = H5T_get_size(dt); + + break; + + /* If this is a variable length datatype, iterate over it */ + case H5T_VLEN: + { + H5T_t *super = NULL; + const hvl_t *vl; + + vl = (const hvl_t *)buf; + + if(NULL == (super = H5T_get_super(dt))) + HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, FAIL, "invalid super type of VL type"); + + buf_size = H5T_get_size(super) * vl->len; + H5T_close(super); + break; + } /* end block */ + default: + HGOTO_ERROR(H5E_ARGS, H5E_CANTINIT, FAIL, "unsupported datatype"); + } /* end switch */ + + *size = buf_size; + if(ret_class) + *ret_class = dt_class; + +done: + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daos_map_get_size */ + + +/*------------------------------------------------------------------------- + * Function: H5VL_daos_map_dtype_info + * + * Purpose: Retrieves information about the datatype of Map Key or + * value datatype, whether it's VL or not. If it is not VL + * return the size. + * + * Return: Success: SUCCEED + * Failure: Negative + * + *------------------------------------------------------------------------- + */ +static herr_t +H5VL_daos_map_dtype_info(hid_t type_id, hbool_t *is_vl, size_t *size, + H5T_class_t *cls) +{ + size_t buf_size = 0; + H5T_t *dt = NULL; + H5T_class_t dt_class; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT + + if(NULL == (dt = (H5T_t *)H5I_object_verify(type_id, H5I_DATATYPE))) + HGOTO_ERROR(H5E_ARGS, H5E_BADTYPE, H5T_NO_CLASS, "not a datatype") + + dt_class = H5T_get_class(dt, FALSE); + + switch(dt_class) { + case H5T_STRING: + /* If this is a variable length string, get the size using strlen(). */ + if(H5T_is_variable_str(dt)) { + *is_vl = TRUE; + break; + } + case H5T_INTEGER: + case H5T_FLOAT: + case H5T_TIME: + case H5T_BITFIELD: + case H5T_OPAQUE: + case H5T_ENUM: + case H5T_ARRAY: + case H5T_NO_CLASS: + case H5T_REFERENCE: + case H5T_NCLASSES: + case H5T_COMPOUND: + /* Data is not variable length, so use H5Tget_size() */ + /* MSC - This is not correct. Compound/Array can contian + VL datatypes, but for now we don't support that. Need + to check for that too */ + buf_size = H5T_get_size(dt); + *is_vl = FALSE; + break; + + /* If this is a variable length datatype, iterate over it */ + case H5T_VLEN: + *is_vl = TRUE; + break; + default: + HGOTO_ERROR(H5E_ARGS, H5E_CANTINIT, FAIL, "unsupported datatype"); + } + + if(size) + *size = buf_size; + if(cls) + *cls = dt_class; +done: + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daos_map_dtype_info */ + + +herr_t +H5VL_daosm_map_set(void *_map, hid_t key_mem_type_id, const void *key, + hid_t val_mem_type_id, const void *value, hid_t H5_ATTR_UNUSED dxpl_id, + void H5_ATTR_UNUSED **req) +{ + H5VL_daosm_map_t *map = (H5VL_daosm_map_t *)_map; + size_t key_size, val_size; + char const_akey[] = H5VL_DAOSM_MAP_KEY; + daos_key_t dkey; + daos_iod_t iod; + daos_sg_list_t sgl; + daos_iov_t sg_iov; + H5T_class_t cls; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT + + /* get the key size and checksum from the provdied key datatype & buffer */ + if(H5VL_daos_map_get_size(key_mem_type_id, key, NULL, &key_size, NULL) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "can't get key size"); + + /* get the val size and checksum from the provdied val datatype & buffer */ + if(H5VL_daos_map_get_size(val_mem_type_id, value, NULL, &val_size, &cls) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "can't get val size"); + + /* Set up dkey */ + daos_iov_set(&dkey, (void *)key, (daos_size_t)key_size); + + /* Set up iod */ + HDmemset(&iod, 0, sizeof(iod)); + daos_iov_set(&iod.iod_name, (void *)const_akey, (daos_size_t)(sizeof(const_akey) - 1)); + daos_csum_set(&iod.iod_kcsum, NULL, 0); + iod.iod_nr = 1u; + iod.iod_size = (daos_size_t)val_size; + iod.iod_type = DAOS_IOD_SINGLE; + + /* Set up sgl */ + if (H5T_VLEN == cls) { + hvl_t *vl_buf = (hvl_t *)value; + + daos_iov_set(&sg_iov, (void *)vl_buf->p, (daos_size_t)val_size); + } + else { + daos_iov_set(&sg_iov, (void *)value, (daos_size_t)val_size); + } + + sgl.sg_nr.num = 1; + sgl.sg_iovs = &sg_iov; + + if(0 != (ret_value = daos_obj_update(map->obj.obj_oh, + map->obj.item.file->epoch, &dkey, + 1, &iod, &sgl, NULL))) + HGOTO_ERROR(H5E_MAP, H5E_CANTSET, FAIL, "Map set failed: %d", ret_value); + +done: + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_set() */ + + +herr_t +H5VL_daosm_map_get(void *_map, hid_t key_mem_type_id, const void *key, + hid_t val_mem_type_id, void *value, hid_t H5_ATTR_UNUSED dxpl_id, + void H5_ATTR_UNUSED **req) +{ + H5VL_daosm_map_t *map = (H5VL_daosm_map_t *)_map; + size_t key_size, val_size; + hbool_t val_is_vl; + char const_akey[] = H5VL_DAOSM_MAP_KEY; + daos_key_t dkey; + daos_iod_t iod; + daos_sg_list_t sgl; + daos_iov_t sg_iov; + H5T_class_t cls; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT; + + /* get the key size and checksum from the provdied key datatype & buffer */ + if(H5VL_daos_map_get_size(key_mem_type_id, key, NULL, &key_size, NULL) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "can't get key size"); + + /* get information about the datatype of the value. Get the values + size if it is not VL. val_size will be 0 if it is VL */ + if(H5VL_daos_map_dtype_info(val_mem_type_id, &val_is_vl, &val_size, &cls) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "can't get key size"); + + /* Set up dkey */ + daos_iov_set(&dkey, (void *)key, (daos_size_t)key_size); + /* Set up iod */ + HDmemset(&iod, 0, sizeof(iod)); + daos_iov_set(&iod.iod_name, const_akey, (daos_size_t)(sizeof(const_akey) - 1)); + daos_csum_set(&iod.iod_kcsum, NULL, 0); + iod.iod_nr = 1u; + iod.iod_type = DAOS_IOD_SINGLE; + + if (!val_is_vl) { + iod.iod_size = (daos_size_t)val_size; + + /* Set up sgl */ + daos_iov_set(&sg_iov, value, (daos_size_t)val_size); + sgl.sg_nr.num = 1; + sgl.sg_iovs = &sg_iov; + + if(0 != (ret_value = daos_obj_fetch(map->obj.obj_oh, + map->obj.item.file->epoch, &dkey, + 1, &iod, &sgl, NULL , NULL))) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "MAP get failed: %d", ret_value); + } + else { + iod.iod_size = DAOS_REC_ANY; + if(0 != (ret_value = daos_obj_fetch(map->obj.obj_oh, + map->obj.item.file->epoch, &dkey, + 1, &iod, NULL, NULL , NULL))) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "MAP get failed: %d", ret_value); + + val_size = iod.iod_size; + + if(cls == H5T_STRING) { + char *val; + + val = (char *)malloc(val_size); + daos_iov_set(&sg_iov, val, (daos_size_t)val_size); + (*(void **) value) = val; + } + else { + hvl_t *vl_buf = (hvl_t *)value; + + HDassert(H5T_VLEN == cls); + + vl_buf->len = val_size; + vl_buf->p = malloc(val_size); + daos_iov_set(&sg_iov, vl_buf->p, (daos_size_t)val_size); + } + + sgl.sg_nr.num = 1; + sgl.sg_iovs = &sg_iov; + + if(0 != (ret_value = daos_obj_fetch(map->obj.obj_oh, + map->obj.item.file->epoch, &dkey, + 1, &iod, &sgl, NULL , NULL))) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "MAP get failed: %d", ret_value); + } + +done: + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_get() */ + + +herr_t +H5VL_daosm_map_get_types(void *_map, hid_t *key_type_id, hid_t *val_type_id, + void H5_ATTR_UNUSED **req) +{ + H5VL_daosm_map_t *map = (H5VL_daosm_map_t *)_map; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT; + + if((*key_type_id = H5Tcopy(map->ktype_id)) < 0) + HGOTO_ERROR(H5E_ARGS, H5E_CANTGET, FAIL, "can't get datatype ID of map key"); + + if((*val_type_id = H5Tcopy(map->vtype_id)) < 0) + HGOTO_ERROR(H5E_ARGS, H5E_CANTGET, FAIL, "can't get datatype ID of map val"); + +done: + FUNC_LEAVE_NOAPI(ret_value); +} /* end H5VL_daosm_map_get_types() */ + + +#define ENUM_DESC_BUF 512 +#define ENUM_DESC_NR 5 + +herr_t +H5VL_daosm_map_get_count(void *_map, hsize_t *count, void H5_ATTR_UNUSED **req) +{ + H5VL_daosm_map_t *map = (H5VL_daosm_map_t *)_map; + char *buf; + daos_key_desc_t kds[ENUM_DESC_NR]; + daos_hash_out_t anchor; + uint32_t number; + hsize_t key_nr; + daos_sg_list_t sgl; + daos_iov_t sg_iov; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT; + + memset(&anchor, 0, sizeof(anchor)); + buf = (char *)malloc(ENUM_DESC_BUF); + + daos_iov_set(&sg_iov, buf, ENUM_DESC_BUF); + sgl.sg_nr.num = 1; + sgl.sg_iovs = &sg_iov; + + for (number = ENUM_DESC_NR, key_nr = 0; !daos_hash_is_eof(&anchor); + number = ENUM_DESC_NR) { + HDmemset(buf, 0, ENUM_DESC_BUF); + + ret_value = daos_obj_list_dkey(map->obj.obj_oh, + map->obj.item.file->epoch, + &number, kds, &sgl, &anchor, NULL); + if(ret_value != 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "Map List failed: %d", ret_value); + if (number == 0) + continue; /* loop should break for EOF */ + + key_nr += (hsize_t)number; + } + + /* -1 for MD dkey */ + *count = (hsize_t)(key_nr - 1); + +done: + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_get_count() */ + + +herr_t +H5VL_daosm_map_exists(void *_map, hid_t key_mem_type_id, const void *key, + hbool_t *exists, void H5_ATTR_UNUSED **req) +{ + H5VL_daosm_map_t *map = (H5VL_daosm_map_t *)_map; + size_t key_size; + char const_akey[] = H5VL_DAOSM_MAP_KEY; + daos_key_t dkey; + daos_iod_t iod; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT; + + /* get the key size and checksum from the provdied key datatype & buffer */ + if(H5VL_daos_map_get_size(key_mem_type_id, key, NULL, &key_size, NULL) < 0) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "can't get key size"); + + /* Set up dkey */ + daos_iov_set(&dkey, (void *)key, (daos_size_t)key_size); + /* Set up iod */ + HDmemset(&iod, 0, sizeof(iod)); + daos_iov_set(&iod.iod_name, (void *)const_akey, (daos_size_t)(sizeof(const_akey) - 1)); + daos_csum_set(&iod.iod_kcsum, NULL, 0); + iod.iod_nr = 1u; + iod.iod_type = DAOS_IOD_SINGLE; + iod.iod_size = DAOS_REC_ANY; + + if(0 != (ret_value = daos_obj_fetch(map->obj.obj_oh, + map->obj.item.file->epoch, &dkey, + 1, &iod, NULL, NULL , NULL))) + HGOTO_ERROR(H5E_MAP, H5E_CANTGET, FAIL, "MAP get failed: %d", ret_value); + + if(iod.iod_size != 0) + *exists = TRUE; + else + *exists = FALSE; + +done: + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_exists() */ + + +/*------------------------------------------------------------------------- + * Function: H5VL_daosm_map_close + * + * Purpose: Closes a daos-m HDF5 map. + * + * Return: Success: 0 + * Failure: -1 + * + * Programmer: Neil Fortner + * November, 2016 + * + *------------------------------------------------------------------------- + */ +herr_t +H5VL_daosm_map_close(void *_map, hid_t H5_ATTR_UNUSED dxpl_id, + void H5_ATTR_UNUSED **req) +{ + H5VL_daosm_map_t *map = (H5VL_daosm_map_t *)_map; + int ret; + herr_t ret_value = SUCCEED; + + FUNC_ENTER_NOAPI_NOINIT + + HDassert(map); + + if(--map->obj.item.rc == 0) { + /* Free map data structures */ + if(!daos_handle_is_inval(map->obj.obj_oh)) + if(0 != (ret = daos_obj_close(map->obj.obj_oh, NULL /*event*/))) + HDONE_ERROR(H5E_MAP, H5E_CANTCLOSEOBJ, FAIL, "can't close map DAOS object: %d", ret) + if(map->ktype_id != FAIL && H5I_dec_app_ref(map->ktype_id) < 0) + HDONE_ERROR(H5E_MAP, H5E_CANTDEC, FAIL, "failed to close datatype") + if(map->vtype_id != FAIL && H5I_dec_app_ref(map->vtype_id) < 0) + HDONE_ERROR(H5E_MAP, H5E_CANTDEC, FAIL, "failed to close datatype") + map = H5FL_FREE(H5VL_daosm_map_t, map); + } /* end if */ + + FUNC_LEAVE_NOAPI(ret_value) +} /* end H5VL_daosm_map_close() */ + |