/** * make a bucket * * @param ctx * @param bucket * @param location_constraint * @param sysmd * * @return */ dpl_status_t dpl_s3_make_bucket(dpl_ctx_t *ctx, const char *bucket, const dpl_sysmd_t *sysmd, char **locationp) { int ret, ret2; dpl_conn_t *conn = NULL; char header[dpl_header_size]; u_int header_len; struct iovec iov[10]; int n_iov = 0; int connection_close = 0; char *location_constraint_str; char data_str[1024]; u_int data_len; dpl_dict_t *headers_request = NULL; dpl_dict_t *headers_reply = NULL; dpl_req_t *req = NULL; dpl_s3_req_mask_t req_mask = 0u; DPL_TRACE(ctx, DPL_TRACE_BACKEND, ""); req = dpl_req_new(ctx); if (NULL == req) { ret = DPL_ENOMEM; goto end; } dpl_req_set_method(req, DPL_METHOD_PUT); if (NULL == bucket) { ret = DPL_EINVAL; goto end; } ret2 = dpl_req_set_bucket(req, bucket); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } ret2 = dpl_req_set_resource(req, "/"); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } if (sysmd) { if (sysmd->mask & DPL_SYSMD_MASK_LOCATION_CONSTRAINT) { if (DPL_LOCATION_CONSTRAINT_US_STANDARD == sysmd->location_constraint) { data_str[0] = 0; data_len = 0; } else { location_constraint_str = dpl_location_constraint_str(sysmd->location_constraint); if (NULL == location_constraint_str) { ret = DPL_ENOMEM; goto end; } snprintf(data_str, sizeof (data_str), "<CreateBucketConfiguration xmlns=\"http://s3.amazonaws.com/doc/2006-03-01/\">\n" "<LocationConstraint>%s</LocationConstraint>\n" "</CreateBucketConfiguration>\n", location_constraint_str); data_len = strlen(data_str); } } else { data_str[0] = 0; data_len = 0; } } else { data_str[0] = 0; data_len = 0; } dpl_req_set_data(req, data_str, data_len); if (sysmd) { if (sysmd->mask & DPL_SYSMD_MASK_CANNED_ACL) dpl_req_set_canned_acl(req, sysmd->canned_acl); if (sysmd->mask & DPL_SYSMD_MASK_STORAGE_CLASS) dpl_req_set_storage_class(req, sysmd->storage_class); } //build request ret2 = dpl_s3_req_build(req, req_mask, &headers_request); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } ret2 = dpl_try_connect(ctx, req, &conn); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } ret2 = dpl_add_host_to_headers(req, headers_request); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } ret2 = dpl_req_gen_http_request(ctx, req, headers_request, NULL, header, sizeof (header), &header_len); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } iov[n_iov].iov_base = header; iov[n_iov].iov_len = header_len; n_iov++; //final crlf iov[n_iov].iov_base = "\r\n"; iov[n_iov].iov_len = 2; n_iov++; //buffer iov[n_iov].iov_base = data_str; iov[n_iov].iov_len = data_len; n_iov++; ret2 = dpl_conn_writev_all(conn, iov, n_iov, conn->ctx->write_timeout); if (DPL_SUCCESS != ret2) { DPL_TRACE(conn->ctx, DPL_TRACE_ERR, "writev failed"); connection_close = 1; ret = ret2; goto end; } ret2 = dpl_read_http_reply(conn, 1, NULL, NULL, &headers_reply, &connection_close); if (DPL_SUCCESS != ret2) { ret = ret2; goto end; } ret = DPL_SUCCESS; end: if (NULL != conn) { if (1 == connection_close) dpl_conn_terminate(conn); else dpl_conn_release(conn); } if (NULL != headers_reply) dpl_dict_free(headers_reply); if (NULL != headers_request) dpl_dict_free(headers_request); if (NULL != req) dpl_req_free(req); DPL_TRACE(ctx, DPL_TRACE_BACKEND, "ret=%d", ret); return ret; }
int cmd_mb(int argc, char **argv) { int ret; char opt; char *bucket; char *resource; dpl_canned_acl_t canned_acl = DPL_CANNED_ACL_PRIVATE; int Aflag = 0; int i; dpl_location_constraint_t location_constraint = DPL_LOCATION_CONSTRAINT_US_STANDARD; int Lflag = 0; var_set("status", "1", VAR_CMD_SET, NULL); optind = 0; while ((opt = getopt(argc, argv, usage_getoptstr(mb_usage))) != -1) switch (opt) { case 'l': location_constraint = dpl_location_constraint(optarg); if (-1 == location_constraint) { fprintf(stderr, "bad location constraint '%s'\n", optarg); return SHELL_CONT; } break ; case 'L': Lflag = 1; break ; case 'a': canned_acl = dpl_canned_acl(optarg); if (-1 == canned_acl) { fprintf(stderr, "bad canned acl '%s'\n", optarg); return SHELL_CONT; } break ; case 'A': Aflag = 1; break ; case '?': default: usage_help(&mb_cmd); return SHELL_CONT; } argc -= optind; argv += optind; if (1 == Aflag) { for (i = 0;i < DPL_N_CANNED_ACL;i++) printf("%s\n", dpl_canned_acl_str(i)); return SHELL_CONT; } if (1 == Lflag) { for (i = 0;i < DPL_N_LOCATION_CONSTRAINT;i++) printf("%s\n", dpl_location_constraint_str(i)); return SHELL_CONT; } if (1 != argc) { usage_help(&mb_cmd); return SHELL_CONT; } bucket = argv[0]; resource = "/"; ret = dpl_make_bucket(ctx, bucket, location_constraint, canned_acl); if (DPL_SUCCESS != ret) { fprintf(stderr, "status: %s (%d)\n", dpl_status_str(ret), ret); return SHELL_CONT; } var_set("status", "0", VAR_CMD_SET, NULL); return SHELL_CONT; }