/*
* OSD object identification sense data descriptor. Always need one
* of these, then perhaps some others. This goes just beyond the 8
* byte sense header above. The length of this is 32. osd2r01 Sec 4.14.2.1
*/
static int sense_info_build(uint8_t *data, int len, uint32_t not_init_funcs,
uint32_t completed_funcs, uint64_t pid, uint64_t oid)
{
if (len < 32)
return 0;
data[0] = 0x6;
data[1] = 0x1e;
set_htonl(&data[8], not_init_funcs);
set_htonl(&data[12], completed_funcs);
set_htonll(&data[16], pid);
set_htonll(&data[24], oid);
osd_warning(" identification pid=%llx oid=%llx",pid, oid);
return 32;
}
/*
* SPC-3 command-specific information sense data descriptor. p. 33.
*/
static int sense_csi_build(uint8_t *data, int len, uint64_t csi)
{
if (len < 12)
return 0;
data[0] = 0x1;
data[1] = 0xa;
set_htonll(&data[4], csi);
osd_warning(" command-specific information=%llx",csi);
return 12;
}
/*
* this function executes id retrieval statement. Only the functions
* retireiving a list of oids, cids or pids may use this function
*
* returns:
* OSD_ERROR: in case of any error
* OSD_OK: on success
* OSD_REPEAT: in case of sqlite_schema error, statements are prepared again,
* hence values need to be bound again.
*/
int db_exec_id_rtrvl_stmt(struct db_context *dbc, sqlite3_stmt *stmt,
int ret, const char *func, uint64_t alloc_len,
uint8_t *outdata, uint64_t *used_outlen,
uint64_t *add_len, uint64_t *cont_id)
{
uint64_t len = 0;
int bound = (ret == SQLITE_OK);
if (!bound) {
error_sql(dbc->db, "%s: bind failed", func);
goto out_reset;
}
len = 0;
*add_len = 0;
*cont_id = 0;
*used_outlen = 0;
while (1) {
ret = sqlite3_step(stmt);
if (ret == SQLITE_ROW) {
if ((alloc_len - len) >= 8) {
set_htonll(outdata,
sqlite3_column_int64(stmt, 0));
outdata += 8;
len += 8;
} else if (*cont_id == 0) {
*cont_id = sqlite3_column_int64(stmt, 0);
}
/* handle overflow: osd2r01 Sec 6.14.2 */
if (*add_len + 8 > *add_len) {
*add_len += 8;
} else {
/* terminate since add_len overflew */
*add_len = (uint64_t) -1;
break;
}
} else if (ret == SQLITE_BUSY) {
continue;
} else {
break;
}
}
out_reset:
ret = db_reset_stmt(dbc, stmt, bound, func);
if (ret == OSD_OK)
*used_outlen = len;
return ret;
}
static void query_speed(struct osd_device *osd, int numiter, int numobj,
int numobj_in_coll, int nummatch, int numattr,
int numcriteria)
{
struct osd_command c;
int i;
uint64_t start, end;
double *v;
uint64_t pid = PARTITION_PID_LB;
uint64_t cid = COLLECTION_OID_LB;
uint8_t *results, *query;
double mu, sd;
struct attribute_list *attr;
uint8_t *attr_val_lo, *attr_val_match, *attr_val_nomatch, *attr_val_hi;
uint64_t alloc_len, query_len, criterion_len;
uint8_t cidn[8], *cp;
if (numobj_in_coll > numobj)
return;
if (nummatch > numobj_in_coll)
return;
if (numcriteria > numattr)
return;
v = Malloc(numiter * sizeof(*v));
if (!v)
return;
alloc_len = 12 + numobj * 8;
results = Malloc(alloc_len);
if (!results)
return;
attr = Malloc((1 + numattr) * sizeof(*attr));
if (!attr)
return;
attr_val_lo = Malloc(4 * ATTR_LEN);
if (!attr_val_lo)
return;
attr_val_match = attr_val_lo + ATTR_LEN;
attr_val_hi = attr_val_lo + 2 * ATTR_LEN;
attr_val_nomatch = attr_val_lo + 3 * ATTR_LEN;
memset(attr_val_lo, 0x4b, ATTR_LEN);
memset(attr_val_match, 0x5a, ATTR_LEN);
memset(attr_val_hi, 0x69, ATTR_LEN);
memset(attr_val_nomatch, 0x78, ATTR_LEN);
osd_command_set_create_partition(&c, pid);
run(osd, &c);
osd_command_set_create_collection(&c, pid, cid);
run(osd, &c);
for (i=0; i<numattr; i++) {
attr[i].type = ATTR_SET;
attr[i].page = LUN_PG_LB;
attr[i].number = 1 + i; /* skip directory */
attr[i].val = attr_val_nomatch;
attr[i].len = ATTR_LEN;
}
/* one extra attr to stick it in the collection, for some of them */
set_htonll(cidn, cid);
attr[numattr].type = ATTR_SET;
attr[numattr].page = USER_COLL_PG;
attr[numattr].number = 1;
attr[numattr].len = 8;
attr[numattr].val = cidn;
for (i=0; i<numobj; i++) {
int j, this_numattr;
osd_command_set_create(&c, pid, 0, 1);
this_numattr = numattr;
if (i < numobj_in_coll)
++this_numattr; /* stick in coll */
/* some of the ones in the collection will match */
for (j=0; j<numattr; j++)
attr[j].val = (i < nummatch ? attr_val_match
: attr_val_nomatch);
osd_command_attr_build(&c, attr, this_numattr);
run(osd, &c);
osd_command_attr_free(&c);
}
criterion_len = 12 + 2 + ATTR_LEN + 2 + ATTR_LEN;
if (numcriteria == 0)
query_len = 8; /* special case, just one empty criterion */
else
query_len = 4 + numcriteria * criterion_len;
query = Malloc(query_len);
if (!query)
return;
memset(query, 0, query_len);
query[0] = 1; /* intersection */
cp = query + 4;
for (i=0; i<numcriteria; i++) {
set_htons(cp + 2, criterion_len);
set_htonl(cp + 4, LUN_PG_LB);
set_htonl(cp + 8, 1 + i);
set_htons(cp + 12, ATTR_LEN);
memcpy(cp + 12 + 2, attr_val_lo, ATTR_LEN);
set_htons(cp + 12 + 2 + ATTR_LEN, ATTR_LEN);
memcpy(cp + 12 + 2 + ATTR_LEN + 2, attr_val_hi, ATTR_LEN);
cp += criterion_len;
}
osd_command_set_query(&c, pid, cid, query_len, alloc_len);
c.outdata = query;
c.outlen = query_len;
for (i=0; i<5; i++)
run(osd, &c);
for (i=0; i<numiter; i++) {
rdtsc(start);
run(osd, &c);
rdtsc(end);
v[i] = (double)(end - start) / mhz;
}
mu = mean(v, numiter);
sd = stddev(v, mu, numiter);
printf("query numiter %d numobj %d numobj_in_coll %d nummatch %d"
" numattr %d numcriteria %d avg %lf +/- %lf us\n",
numiter, numobj, numobj_in_coll, nummatch, numattr,
numcriteria, mu, sd);
free(query);
free(attr_val_lo);
free(attr);
free(results);
free(v);
}
/*
* return values:
* -EINVAL: invalid argument
* -EIO: prepare or some other sqlite function failed
* OSD_ERROR: some other error
* OSD_OK: success
*/
int mtq_run_query(struct db_context *dbc, uint64_t pid, uint64_t cid,
struct query_criteria *qc, void *outdata,
uint32_t alloc_len, uint64_t *used_outlen,
uint64_t matches_cid)
{
int ret = 0;
int pos = 0;
char *cp = NULL;
char *SQL = NULL;
uint8_t *p = NULL;
uint32_t i = 0;
uint32_t sqlen = 0;
uint32_t number;
uint32_t factor = 2; /* this query fills space quickly */
uint64_t len = 0;
const char *op = NULL;
sqlite3_stmt *stmt = NULL;
char select_stmt[MAXSQLEN];
const char *coll = coll_getname(dbc);
const char *attr = attr_getname(dbc);
assert(dbc && dbc->db && qc && outdata && used_outlen && coll
&& attr);
if (qc->query_type == 0) {
op = " UNION ";
} else if (qc->query_type == 1) {
op = " INTERSECT ";
} else {
ret = -EINVAL;
goto out;
}
SQL = Malloc(MAXSQLEN*factor);
if (SQL == NULL) {
ret = -ENOMEM;
goto out;
}
cp = SQL;
sqlen = 0;
/*
* XXX:SD the spec does not mention whether min or max values have to
* in tested with '<' or '<='. We assume the tests are inclusive of
* boundaries, i.e. use '<=' for comparison.
*/
/* build the SQL statment */
if (matches_cid != 0) {
ret = coll_max_pointer(dbc, pid, cid, &number);
if (ret != SQLITE_OK) {
goto out;
}
sprintf(select_stmt, "SELECT coll.pid, %llu, attr.oid, %d "
"FROM %s AS coll, %s AS attr WHERE "
"coll.pid=coll.pid AND coll.oid=attr.oid "
"AND coll.pid = %llu AND coll.cid = %llu ",
llu(matches_cid), number, coll, attr, llu(pid),
llu(cid));
} else {
sprintf(select_stmt, "SELECT attr.oid FROM %s AS coll, "
"%s AS attr WHERE coll.pid = coll.pid AND "
"coll.oid = attr.oid AND coll.pid = %llu "
"AND coll.cid = %llu ",
coll, attr, llu(pid), llu(cid));
}
strncpy(cp, select_stmt, MAXSQLEN*factor);
sqlen += strlen(cp);
cp += sqlen;
for (i = 0; i < qc->qc_cnt; i++) {
sprintf(cp, " AND attr.page = %u AND attr.number = %u ",
qc->page[i], qc->number[i]);
if (qc->min_len[i] > 0)
cp = strcat(cp, " AND ? <= attr.value ");
if (qc->max_len[i] > 0)
cp = strcat(cp, " AND attr.value <= ? ");
if ((i+1) < qc->qc_cnt) {
cp = strcat(cp, op);
cp = strcat(cp, select_stmt);
}
sqlen += strlen(cp);
if (sqlen >= (MAXSQLEN*factor - 400)) {
factor *= 2;
SQL = realloc(SQL, MAXSQLEN*factor);
if (!SQL) {
ret = -ENOMEM;
goto out;
}
}
cp = SQL + sqlen;
}
cp = strcat(cp, " GROUP BY attr.oid ORDER BY 1;");
if (matches_cid != 0) {
char *SQL2 = Malloc(strlen(SQL) + 100);
if (SQL2 == NULL) {
ret = -ENOMEM;
goto out;
}
sprintf(SQL2, "INSERT INTO %s %s", coll, SQL);
free(SQL);
SQL = SQL2;
}
ret = sqlite3_prepare(dbc->db, SQL, strlen(SQL)+1, &stmt, NULL);
if (ret != SQLITE_OK) {
error_sql(dbc->db, "%s: sqlite3_prepare", __func__);
ret = -EIO;
goto out;
}
repeat:
/* bind the values */
pos = 1;
for (i = 0; i < qc->qc_cnt; i++) {
if (qc->min_len[i] > 0) {
ret = sqlite3_bind_blob(stmt, pos, qc->min_val[i],
qc->min_len[i],
SQLITE_TRANSIENT);
if (ret != SQLITE_OK) {
ret = -EIO;
error_sql(dbc->db, "%s: bind min_val @ %d",
__func__, pos);
goto out_finalize;
}
pos++;
}
if (qc->max_len[i] > 0) {
ret = sqlite3_bind_blob(stmt, pos, qc->max_val[i],
qc->max_len[i],
SQLITE_TRANSIENT);
if (ret != SQLITE_OK) {
ret = -EIO;
error_sql(dbc->db, "%s: bind max_val @ %d",
__func__, pos);
goto out_finalize;
}
pos++;
}
}
if (matches_cid != 0) {
ret = db_exec_dms(dbc, stmt, ret, __func__);
if (ret == OSD_REPEAT)
goto repeat;
goto out_finalize;
}
/* execute the query */
p = outdata;
p += ML_ODL_OFF;
len = ML_ODL_OFF - 8; /* subtract len of addition_len */
*used_outlen = ML_ODL_OFF;
while ((ret = sqlite3_step(stmt)) == SQLITE_ROW) {
if ((alloc_len - len) > 8) {
/*
* TODO: query is a multi-object command, so delete
* the objects from the collection, once they are
* selected
*/
set_htonll(p, sqlite3_column_int64(stmt, 0));
*used_outlen += 8;
}
p += 8;
/* handle overflow: osd2r01 Sec 6.18.3 */
if (len != (uint64_t) -1 && (len + 8) > len) {
len += 8;
} else {
len = (uint64_t) -1;
}
}
if (ret != SQLITE_DONE) {
error_sql(dbc->db, "%s: sqlite3_step", __func__);
ret = -EIO;
goto out_finalize;
}
set_htonll(outdata, len);
ret = OSD_OK;
out_finalize:
if (sqlite3_finalize(stmt) != SQLITE_OK)
error_sql(dbc->db, "%s: finalize", __func__);
out:
free(SQL);
return ret;
}
/*
* returns list of objects along with requested attributes
*
* return values:
* -EINVAL: invalid argument
* -EIO: prepare or some other sqlite function failed
* OSD_ERROR: some other error
* OSD_OK: success
*/
int mtq_list_oids_attr(struct db_context *dbc, uint64_t pid,
uint64_t initial_oid, struct getattr_list *get_attr,
uint64_t alloc_len, void *outdata,
uint64_t *used_outlen, uint64_t *add_len,
uint64_t *cont_id)
{
int ret = 0;
char *cp = NULL;
char *SQL = NULL;
uint32_t i = 0;
uint32_t factor = 2; /* this query uses space fast */
uint32_t attr_list_len = 0; /*XXX:SD see below */
uint32_t sqlen = 0;
uint64_t oid = 0;
uint32_t page;
uint32_t number;
uint16_t len;
const void *val = NULL;
sqlite3_stmt *stmt = NULL;
uint8_t *head = NULL, *tail = NULL;
const char *select_stmt = NULL;
const char *obj = obj_getname(dbc);
const char *attr = attr_getname(dbc);
assert(dbc && dbc->db && get_attr && outdata && used_outlen
&& add_len && obj && attr);
if (get_attr->sz == 0) {
ret = -EINVAL;
goto out;
}
SQL = Malloc(MAXSQLEN*factor);
if (!SQL) {
ret = -ENOMEM;
goto out;
}
/*
* For each attribute requested, create a select statement,
* which will try to index into the attr table with a full key rather
* than just (pid, oid) prefix key. Analogous to loop unrolling, we
* unroll each requested attribute into its own select statement.
*/
select_stmt = "SELECT obj.oid as myoid, attr.page, "
" attr.number, attr.value FROM %s as obj, %s as attr "
" WHERE obj.pid = attr.pid AND obj.oid = attr.oid AND "
" obj.pid = %llu AND obj.type = %u AND "
" attr.page = %u AND attr.number = %u AND obj.oid >= %llu ";
cp = SQL;
sqlen = 0;
sprintf(SQL, select_stmt, obj, attr, llu(pid), USEROBJECT,
USER_TMSTMP_PG, 0, llu(initial_oid));
SQL = strcat(SQL, " UNION ALL ");
sqlen += strlen(SQL);
cp += sqlen;
for (i = 0; i < get_attr->sz; i++) {
sprintf(cp, select_stmt, obj, attr, llu(pid), USEROBJECT,
get_attr->le[i].page, get_attr->le[i].number,
llu(initial_oid));
if (i < (get_attr->sz - 1))
cp = strcat(cp, " UNION ALL ");
sqlen += strlen(cp);
if (sqlen > (MAXSQLEN*factor - 400)) {
factor *= 2;
SQL = realloc(SQL, MAXSQLEN*factor);
if (!SQL) {
ret = -ENOMEM;
goto out;
}
}
cp = SQL + sqlen;
}
sprintf(cp, " ORDER BY myoid; ");
ret = sqlite3_prepare(dbc->db, SQL, strlen(SQL)+1, &stmt, NULL);
if (ret != SQLITE_OK) {
error_sql(dbc->db, "%s: sqlite3_prepare", __func__);
goto out;
}
/* execute the statement */
head = tail = outdata;
attr_list_len = 0;
while(1) {
ret = sqlite3_step(stmt);
if (ret == SQLITE_BUSY) {
continue;
} else if (ret != SQLITE_ROW) {
break;
}
/* for the rest of loop body ret == SQLITE_ROW */
/*
* XXX:SD The spec is inconsistent in applying padding and
* alignment rules. Here we make changes to the spec. In our
* case object descriptor format header (table 79) is 16B
* instead of 12B, and attributes list length field is 4B
* instead of 2B as defined in spec, and starts at byte 12
* in the header (not 10). ODE is object descriptor entry.
*/
oid = sqlite3_column_int64(stmt, 0);
page = sqlite3_column_int64(stmt, 1);
number = sqlite3_column_int64(stmt, 2);
len = sqlite3_column_bytes(stmt, 3);
if (page == USER_TMSTMP_PG && number == 0) {
/* look ahead in the buf to see if there is space */
if (alloc_len >= 16) {
/* start attr list of 'this' ODE */
set_htonll(tail, oid);
memset(tail + 8, 0, 4); /* reserved */
if (head != tail) {
/* fill attr_list_len of prev ODE */
set_htonl(head, attr_list_len);
head = tail;
attr_list_len = 0;
}
alloc_len -= 16;
tail += 16;
head += 12; /* points to attr-list-len */
*used_outlen += 16;
} else {
if (head != tail) {
/* fill attr_list_len of prev ODE */
set_htonl(head, attr_list_len);
head = tail;
attr_list_len = 0;
}
if (*cont_id == 0)
*cont_id = oid;
}
/* handle overflow: osd2r01 Sec 6.14.2 */
if (*add_len + 16 > *add_len) {
*add_len += 16;
} else {
/* terminate since add_len overflew */
*add_len = (uint64_t) -1;
ret = SQLITE_DONE;
break;
}
continue;
}
if (alloc_len >= 16) {
/* osd_debug("%s: oid %llu, page %u, number %u, len %u",
__func__, llu(oid), page, number, len); */
val = sqlite3_column_blob(stmt, 3);
ret = le_pack_attr(tail, alloc_len, page, number, len,
val);
assert (ret != -EOVERFLOW);
if (ret > 0) {
alloc_len -= ret;
tail += ret;
attr_list_len += ret;
*used_outlen += ret;
if (alloc_len < 16){
set_htonl(head, attr_list_len);
head = tail;
attr_list_len = 0;
if (*cont_id == 0)
*cont_id = oid;
}
} else {
goto out_finalize;
}
} else {
if (head != tail) {
/* fill attr_list_len of this ODE */
set_htonl(head, attr_list_len);
head = tail;
attr_list_len = 0;
if (*cont_id == 0)
*cont_id = oid;
}
}
/* handle overflow: osd2r01 Sec 6.14.2 */
if ((*add_len + roundup8(4+4+2+len)) > *add_len) {
*add_len += roundup8(4+4+2+len);
} else {
/* terminate since add_len overflew */
*add_len = (uint64_t) -1;
ret = SQLITE_DONE;
break;
}
}
if (ret != SQLITE_DONE) {
error_sql(dbc->db, "%s: query execution failed. SQL %s, "
" add_len %llu attr_list_len %u", __func__, SQL,
llu(*add_len), attr_list_len);
goto out_finalize;
}
if (head != tail) {
set_htonl(head, attr_list_len);
head += (4 + attr_list_len);
assert(head == tail);
}
ret = OSD_OK; /* success */
out_finalize:
if (sqlite3_finalize(stmt) != SQLITE_OK) {
ret = -EIO;
error_sql(dbc->db, "%s: finalize", __func__);
}
out:
free(SQL);
return ret;
}
/* only to be used by test_set_one_attr */
static void set_one_attr_int(struct osd_device *osd, uint64_t pid, uint64_t oid,
uint32_t page, uint32_t number, uint64_t val)
{
struct osd_command cmd;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
uint8_t *data_out = NULL;
uint64_t data_out_len;
uint64_t attrval;
int ret;
struct attribute_list attr = {
.type = ATTR_SET,
.page = page,
.number = number,
.len = 8,
.val = &attrval,
};
set_htonll(&attrval, val);
ret = osd_command_set_set_attributes(&cmd, pid, oid);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, &attr, 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
}
static void set_one_attr_val(struct osd_device *osd, uint64_t pid, uint64_t oid,
uint32_t page, uint32_t number, const void *val,
uint16_t len)
{
struct osd_command cmd;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
uint8_t *data_out = NULL;
uint64_t data_out_len;
int ret,i;
struct attribute_list attr = {
.type = ATTR_SET,
.page = page,
.number = number,
.len = len,
.val = (void *)(uintptr_t) val,
};
ret = osd_command_set_set_attributes(&cmd, pid, oid);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, &attr, 1);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
/* output cdbfmt , length , value */
printf("cdbfmt is: %x \n", cmd.cdb[11]);
printf("length is: %x \n", cmd.cdb[61]);
printf("value is: ");
for(i=0; i<=len; i++){
printf("%c", cmd.cdb[62+i]);
}
printf("\n");
}
static void test_set_one_attr (struct osd_device *osd)
{
struct osd_command cmd;
uint64_t pid = USEROBJECT_PID_LB;
uint64_t oid = USEROBJECT_OID_LB;
uint8_t *data_out = NULL;
uint64_t data_out_len;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
int ret;
uint32_t page = USEROBJECT_PG + LUN_PG_LB;
/* create a partition*/
ret = osd_command_set_create_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* creat one object */
ret = osd_command_set_create(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB, 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
/* these cases should not generate error */
set_one_attr_val(osd, pid, oid, page, 1, "test", 5);
set_one_attr_val(osd, pid, oid, page, 1, "test_set_one_attr", 18);
set_one_attr_int(osd, pid, oid, page, 1, 10);
set_one_attr_int(osd, pid, oid, page, 1, 20);
/* these cases must generate error */
/* set_one_attr_val(osd, pid, oid, page, 1, "ttest_set_one_attr", 19); */
/* set_one_attr_val(osd, pid, oid, page, 1, "", 0); */
}
/* only to be used by test_osd_query */
static void set_attr_int(struct osd_device *osd, uint64_t pid, uint64_t oid,
uint32_t page, uint32_t number, uint64_t val)
{
struct osd_command cmd;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
uint8_t *data_out = NULL;
uint64_t data_out_len;
uint64_t attrval;
int ret;
struct attribute_list attr = {
.type = ATTR_SET,
.page = page,
.number = number,
.len = 8,
.val = &attrval,
};
set_htonll(&attrval, val);
ret = osd_command_set_set_attributes(&cmd, pid, oid);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, &attr, 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
}
static void set_attr_val(struct osd_device *osd, uint64_t pid, uint64_t oid,
uint32_t page, uint32_t number, const void *val,
uint16_t len)
{
struct osd_command cmd;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
uint8_t *data_out = NULL;
uint64_t data_out_len;
int ret;
struct attribute_list attr = {
.type = ATTR_SET,
.page = page,
.number = number,
.len = len,
.val = (void *)(uintptr_t) val,
};
ret = osd_command_set_set_attributes(&cmd, pid, oid);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, &attr, 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
}
static void set_qce(uint8_t *cp, uint32_t page, uint32_t number,
uint16_t min_len, const void *min_val,
uint16_t max_len, const void *max_val)
{
uint16_t len = 4 + 4 + 2 + min_len + 2 + max_len;
set_htons(&cp[2], len);
set_htonl(&cp[4], page);
set_htonl(&cp[8], number);
set_htons(&cp[12], min_len);
memcpy(&cp[14], min_val, min_len);
set_htons(&cp[14+min_len], max_len);
memcpy(&cp[16+min_len], max_val, max_len);
}
static int ismember(uint64_t needle, uint64_t *hay, uint64_t haysz)
{
while (haysz--)
if (needle == hay[haysz])
return 1;
return 0;
}
static void check_results(uint8_t *matches, uint64_t matchlen,
uint64_t *idlist, uint64_t idlistlen)
{
uint32_t add_len = get_ntohll(&matches[0]);
assert(add_len == (5+8*idlistlen));
assert(matches[12] == (0x21 << 2));
assert(matchlen == add_len+8);
add_len -= 5;
matches += MIN_ML_LEN;
while (add_len) {
assert(ismember(get_ntohll(matches), idlist, 8));
matches += 8;
add_len -= 8;
}
}
void test_query(struct osd_device *osd)
{
struct osd_command cmd;
uint64_t pid = PARTITION_PID_LB;
uint64_t cid = COLLECTION_OID_LB;
uint64_t oid = USEROBJECT_OID_LB + 1; /* leave room for cid */
uint8_t *data_out = NULL;
uint64_t data_out_len;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
int i, ret;
/* create a collection and stick some objects in it */
ret = osd_command_set_create_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
ret = osd_command_set_create_collection(&cmd, pid, cid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* but don't put all of the objects in the collection */
for (i=0; i<10; i++) {
uint64_t attrval;
struct attribute_list attr = {
.type = ATTR_SET,
.page = USER_COLL_PG,
.number = 1,
.len = 8,
.val = &attrval,
};
set_htonll(&attrval, cid);
ret = osd_command_set_create(&cmd, pid, oid + i, 1);
assert(ret == 0);
if (!(i == 2 || i == 8)) {
ret = osd_command_attr_build(&cmd, &attr, 1);
assert(ret == 0);
}
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
}
/*
* Set some random attributes for querying.
*/
uint32_t page = USEROBJECT_PG + LUN_PG_LB;
set_attr_int(osd, pid, oid, page, 1, 4);
set_attr_int(osd, pid, oid+1, page, 1, 49);
set_attr_int(osd, pid, oid+1, page, 2, 130);
set_attr_int(osd, pid, oid+2, page, 1, 20);
set_attr_int(osd, pid, oid+3, page, 1, 101);
set_attr_int(osd, pid, oid+4, page, 1, 59);
set_attr_int(osd, pid, oid+4, page, 2, 37);
set_attr_int(osd, pid, oid+5, page, 1, 75);
set_attr_int(osd, pid, oid+6, page, 1, 200);
set_attr_int(osd, pid, oid+7, page, 1, 67);
set_attr_int(osd, pid, oid+8, page, 1, 323);
set_attr_int(osd, pid, oid+8, page, 2, 44);
set_attr_int(osd, pid, oid+9, page, 1, 1);
set_attr_int(osd, pid, oid+9, page, 2, 19);
/*
* Various queries.
*/
/* run without query criteria */
uint8_t buf[1024], *cp, *matches;
uint32_t qll;
uint64_t matchlen;
uint64_t idlist[8];
qll = MINQLISTLEN;
memset(buf, 0, 1024);
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[0] = oid;
idlist[1] = oid+1;
idlist[2] = oid+3;
idlist[3] = oid+4;
idlist[4] = oid+5;
idlist[5] = oid+6;
idlist[6] = oid+7;
idlist[7] = oid+9;
check_results(matches, matchlen, idlist, 8);
free(matches);
matches = NULL;
matchlen = 0;
/* run one query without min/max constraints */
qll = 0;
memset(buf, 0, 1024);
cp = buf;
set_qce(&cp[4], page, 2, 0, NULL, 0, NULL);
qll += 4 + (4+4+4+2+2);
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[0] = oid+1;
idlist[1] = oid+4;
idlist[2] = oid+9;
check_results(matches, matchlen, idlist, 3);
free(matches);
matches = NULL;
matchlen = 0;
/* run one query with criteria */
uint64_t min, max;
qll = 0;
min = 40, max= 80;
set_htonll(&min, min);
set_htonll(&max, max);
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x0;
set_qce(&cp[4], page, 1, sizeof(min), &min, sizeof(max), &max);
qll += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[0] = oid+1;
idlist[1] = oid+4;
idlist[2] = oid+5;
idlist[3] = oid+7;
check_results(matches, matchlen, idlist, 4);
free(matches);
matches = NULL;
matchlen = 0;
/* run union of two query criteria */
qll = 0;
/* first query */
min = 100, max = 180;
set_htonll(&min, min);
set_htonll(&max, max);
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x0; /* UNION */
set_qce(&cp[4], page, 1, sizeof(min), &min, sizeof(max), &max);
qll += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
/* second query */
min = 200, max = 323;
set_htonll(&min, min);
set_htonll(&max, max);
set_qce(cp, page, 1, sizeof(min), &min, sizeof(max), &max);
qll += (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += (4+4+4+2+sizeof(min)+2+sizeof(max));
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[0] = oid+3;
idlist[1] = oid+6;
check_results(matches, matchlen, idlist, 2);
free(matches);
matches = NULL;
matchlen = 0;
/* run intersection of 2 query criteria */
qll = 0;
/* first query */
min = 4, max = 100;
set_htonll(&min, min);
set_htonll(&max, max);
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x1; /* INTERSECTION */
set_qce(&cp[4], page, 1, sizeof(min), &min, sizeof(max), &max);
qll += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
/* second query */
min = 10, max = 400;
set_htonll(&min, min);
set_htonll(&max, max);
set_qce(cp, page, 2, sizeof(min), &min, sizeof(max), &max);
qll += (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += (4+4+4+2+sizeof(min)+2+sizeof(max));
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[0] = oid+1;
idlist[1] = oid+4;
check_results(matches, matchlen, idlist, 2);
free(matches);
matches = NULL;
matchlen = 0;
/* run union of 3 query criteria, with missing min/max */
qll = 0;
/* first query */
min = 130, max = 130;
set_htonll(&min, min);
set_htonll(&max, max);
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x0; /* UNION */
set_qce(&cp[4], page, 2, sizeof(min), &min, sizeof(max), &max);
qll += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
/* second query */
min = 150;
set_htonll(&min, min);
set_qce(cp, page, 1, sizeof(min), &min, 0, NULL);
qll += (4+4+4+2+sizeof(min)+2+0);
cp += (4+4+4+2+sizeof(min)+2+0);
/* third query */
max = 10;
set_htonll(&max, max);
set_qce(cp, page, 1, 0, NULL, sizeof(max), &max);
qll += (4+4+4+2+0+2+sizeof(max));
cp += (4+4+4+2+0+2+sizeof(max));
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[3] = oid;
idlist[4] = oid+1;
idlist[5] = oid+6;
idlist[2] = oid+9;
check_results(matches, matchlen, idlist, 4);
free(matches);
matches = NULL;
matchlen = 0;
/* set some attributes with text values */
set_attr_val(osd, pid, oid, page, 1, "hello", 6);
set_attr_val(osd, pid, oid+1, page, 1, "cat", 4);
set_attr_int(osd, pid, oid+1, page, 2, 130);
set_attr_int(osd, pid, oid+2, page, 1, 20);
set_attr_val(osd, pid, oid+3, page, 1, "zebra", 6);
set_attr_int(osd, pid, oid+4, page, 1, 59);
set_attr_int(osd, pid, oid+4, page, 2, 37);
set_attr_int(osd, pid, oid+5, page, 1, 75);
set_attr_val(osd, pid, oid+6, page, 1, "keema", 6);
set_attr_int(osd, pid, oid+7, page, 1, 67);
set_attr_int(osd, pid, oid+8, page, 1, 323);
set_attr_int(osd, pid, oid+8, page, 2, 44);
set_attr_int(osd, pid, oid+9, page, 1, 1);
set_attr_val(osd, pid, oid+9, page, 2, "hotelling", 10);
/* run queries on different datatypes, with diff min max lengths */
qll = 0;
/* first query */
min = 41, max = 169;
set_htonll(&min, min);
set_htonll(&max, max);
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x0; /* UNION */
set_qce(&cp[4], page, 1, sizeof(min), &min, sizeof(max), &max);
qll += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += 4 + (4+4+4+2+sizeof(min)+2+sizeof(max));
/* second query */
set_qce(cp, page, 1, 3, "ab", 5, "keta");
qll += (4+4+4+2+2+2+5);
cp += (4+4+4+2+2+2+5);
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[3] = oid;
idlist[4] = oid+1;
idlist[0] = oid+4;
idlist[1] = oid+5;
idlist[5] = oid+6;
idlist[2] = oid+7;
check_results(matches, matchlen, idlist, 6);
free(matches);
matches = NULL;
matchlen = 0;
/* run intersection of 3 query criteria, with missing min/max */
qll = 0;
/* first query */
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x1; /* INTERSECTION */
set_qce(&cp[4], page, 1, 2, "a", 3, "zz");
qll += 4 + (4+4+4+2+2+2+3);
cp += 4 + (4+4+4+2+2+2+3);
/* second query */
min = 140;
set_htonll(&min, min);
set_qce(cp, page, 1, sizeof(min), &min, 0, NULL);
qll += (4+4+4+2+sizeof(min)+2+0);
cp += (4+4+4+2+sizeof(min)+2+0);
/* third query */
set_qce(cp, page, 2, 0, NULL, 6, "alpha");
qll += (4+4+4+2+0+2+6);
cp += (4+4+4+2+0+2+6);
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
idlist[0] = oid+1;
check_results(matches, matchlen, idlist, 1);
free(matches);
matches = NULL;
matchlen = 0;
/* run intersection of 2 query criteria with empty result */
qll = 0;
/* first query */
memset(buf, 0, 1024);
cp = buf;
cp[0] = 0x1; /* INTERSECTION */
set_qce(&cp[4], page, 1, 3, "aa", 4, "zzz");
qll += 4 + (4+4+4+2+3+2+4);
cp += 4 + (4+4+4+2+3+2+4);
/* second query */
min = 50;
max = 80;
set_htonll(&min, min);
set_htonll(&max, max);
set_qce(cp, page, 1, sizeof(min), &min, sizeof(max), &max);
qll += (4+4+4+2+sizeof(min)+2+sizeof(max));
cp += (4+4+4+2+sizeof(min)+2+sizeof(max));
ret = osd_command_set_query(&cmd, pid, cid, qll, 4096);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, buf, qll, &matches,
&matchlen, sense_out, &senselen_out);
assert(ret == 0);
check_results(matches, matchlen, idlist, 0);
free(matches);
matches = NULL;
matchlen = 0;
/*
* Cleanup.
*/
for (i=0; i<10; i++) {
ret = osd_command_set_remove(&cmd, pid, oid + i);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
}
ret = osd_command_set_remove_collection(&cmd, pid, cid, 0);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
ret = osd_command_set_remove_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
}
struct test_attr {
uint8_t type;
uint64_t oid;
uint32_t page;
uint32_t number;
uint64_t intval;
uint16_t valen;
const void *val;
};
static void ismember_attr(struct test_attr *attr, size_t sz, uint64_t oid,
uint32_t page, uint32_t number, uint64_t valen,
const void *val)
{
size_t i = 0;
for (i = 0; i < sz; i++) {
if (attr[i].oid == oid && attr[i].page == page &&
attr[i].number == number) {
assert(valen <= attr[i].valen);
if (attr[i].type == 1) {
if (valen == attr[i].valen)
assert(attr[i].intval == get_ntohll(val));
} else {
assert(memcmp(attr[i].val, val, valen) == 0);
}
return;
}
}
fprintf(stderr, "unknown attr: oid: %llu, page %u, number %u\n",
llu(oid), page, number);
assert(0); /* unknown attr */
}
static void test_oids_with_attr(struct osd_device *osd, uint64_t pid,
struct attribute_list *getattr, int numattr,
uint64_t alloc_len, uint64_t exp_data_out_len,
uint64_t exp_add_len, uint64_t exp_cont_id,
uint8_t exp_odf, struct test_attr *attrs,
size_t attrs_sz)
{
int ret = 0;
struct osd_command cmd;
uint8_t *cp = NULL;
uint32_t page = 0, number = 0;
uint64_t data_in_len, data_out_len;
const void *data_in;
uint8_t *data_out = NULL;
uint64_t oid = 0;
uint8_t sense_out[OSD_MAX_SENSE];
uint16_t len = 0;
int senselen_out;
uint32_t attr_list_len = 0;
/* execute list with attr, alloc length less than required */
ret = osd_command_set_list(&cmd, pid, 0, alloc_len, 0, 1);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, getattr, numattr);
assert(ret == 0);
data_in = cmd.outdata;
data_in_len = cmd.outlen;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
cp = data_out;
assert(data_out_len == exp_data_out_len);
assert(get_ntohll(cp) == exp_add_len);
cp += 8;
assert(get_ntohll(cp) == exp_cont_id);
cp += 8;
assert(get_ntohl(cp) == 0);
cp += 7;
assert(cp[0] == exp_odf);
cp += 1;
oid = 0;
attr_list_len = 0;
len = 0;
data_out_len -= 24;
while (data_out_len > 0) {
oid = get_ntohll(cp);
cp += 12;
attr_list_len = get_ntohl(cp);
cp += 4;
data_out_len -= 16;
while (attr_list_len > 0) {
page = get_ntohl(cp);
cp += 4;
number = get_ntohl(cp);
cp += 4;
len = get_ntohs(cp);
cp += 2;
attr_list_len -= (4+4+2);
data_out_len -= (4+4+2);
if (len > attr_list_len) {
len = attr_list_len;
}
ismember_attr(attrs, attrs_sz, oid, page, number,
len, cp);
cp += len;
cp += (roundup8(2+len) - (2+len));
data_out_len -= len;
data_out_len -= (roundup8(2+len) - (2+len));
attr_list_len -= len;
attr_list_len -= (roundup8(2+len) - (2+len));
}
}
free(data_out);
data_out = NULL;
osd_command_attr_free(&cmd);
}
void test_list(struct osd_device *osd)
{
struct osd_command cmd;
uint64_t pid = PARTITION_PID_LB;
uint64_t cid = 0;
uint64_t oid = 0;
uint8_t *data_out = NULL;
uint8_t *cp;
uint32_t page = 0, number = 0;
uint64_t data_out_len;
uint64_t idlist[64];
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
int i, ret;
/* create partition */
ret = osd_command_set_create_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create collection */
ret = osd_command_set_create_collection(&cmd, pid, cid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create 6 objects */
ret = osd_command_set_create(&cmd, pid, 0, 6);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create another collection */
ret = osd_command_set_create_collection(&cmd, pid, cid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create 4 objects */
ret = osd_command_set_create(&cmd, pid, 0, 4);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* set attributes on userobjects */
page = USEROBJECT_PG + LUN_PG_LB;
number = 1;
oid = COLLECTION_OID_LB + 1;
struct test_attr attrs[] = {
{1, oid, page, number, 1, 8, NULL},
{1, oid, page+1, number+1, 768, 8, NULL},
{2, oid, page+2, number+2, 0, 5, "sudo"},
{1, oid+1, page+1, number+1, 56, 8, NULL},
{1, oid+1, page+2, number+2, 68, 8, NULL},
{2, oid+2, page+2, number+2, 0, 9, "deadbeef"},
{1, oid+3, page+3, number+3, 1, 8, NULL},
{1, oid+3, page+1, number+1, 111, 8, NULL},
{2, oid+3, page+4, number+4, 0, 5, "sudo"},
{1, oid+3, page+2, number+2, 11, 8, NULL},
{1, oid+3, page+5, number+5, 111111, 8, NULL},
{2, oid+4, page+4, number+4, 0, 6, "milli"},
{2, oid+4, page+5, number+5, 0, 10, "kilometer"},
{2, oid+4, page+3, number+3, 0, 11, "hectameter"},
{2, oid+5, page+1, number+1, 0, 12, "zzzzzzhhhhh"},
{2, oid+5, page+2, number+2, 0, 2, "b"},
{1, oid+5, page+3, number+3, 6, 8, NULL},
{1, oid+7, page+1, number+1, 486, 8, NULL},
{1, oid+7, page+4, number+4, 586, 8, NULL},
{1, oid+7, page+2, number+2, 686, 8, NULL},
{1, oid+8, page, number, 4, 8, NULL},
{2, oid+9, page+1, number+1, 0, 14, "setting these"},
{2, oid+9, page+2, number+2, 0, 11, "attributes"},
{2, oid+9, page+3, number+3, 0, 8, "made me"},
{2, oid+9, page+4, number+4, 0, 12, "mad! really"},
{1, oid+10, page+1, number+1, 1234567890, 8, NULL},
{2, oid+10, page, number, 0, 6, "DelTa"},
};
for (i = 0; i < ARRAY_SIZE(attrs); i++) {
if (attrs[i].type == 1) {
set_attr_int(osd, pid, attrs[i].oid, attrs[i].page,
attrs[i].number, attrs[i].intval);
} else {
set_attr_val(osd, pid, attrs[i].oid, attrs[i].page,
attrs[i].number, attrs[i].val,
attrs[i].valen);
}
}
/* set some attributes on collections */
page = COLLECTION_PG + LUN_PG_LB;
cid = COLLECTION_OID_LB;
set_attr_int(osd, pid, cid, page, 1, 1);
set_attr_int(osd, pid, cid+7, page, 2, 2);
/* execute list command. get only oids first */
ret = osd_command_set_list(&cmd, pid, 0, 4096, 0, 0);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
cp = data_out;
assert(get_ntohll(cp) == 10*8+16);
assert(data_out_len == 10*8+24);
cp += 8;
assert(get_ntohll(cp) == 0);
cp += 8;
assert(get_ntohl(cp) == 0);
cp += 7;
assert(cp[0] == (0x21 << 2));
cp += 1;
data_out_len -= 24;
oid = COLLECTION_OID_LB + 1;
for (i = 0; i < 6; i++)
idlist[i] = oid + i;
oid = COLLECTION_OID_LB + 1 + i + 1;
for (i = 0; i < 4; i++)
idlist[6+i] = oid + i;
while (data_out_len > 0) {
assert(ismember(get_ntohll(cp), idlist, 10));
cp += 8;
data_out_len -= 8;
}
free(data_out);
data_out = NULL;
osd_command_attr_free(&cmd);
/* execute list command with less space */
ret = osd_command_set_list(&cmd, pid, 0, 72, 0, 0);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
cp = data_out;
assert(get_ntohll(cp) == 10*8+16);
assert(data_out_len == 72);
cp += 8;
assert(get_ntohll(cp) == COLLECTION_OID_LB + 8);
cp += 8;
assert(get_ntohl(cp) == 0);
cp += 7;
assert(cp[0] == (0x21 << 2));
cp += 1;
data_out_len -= 24;
oid = COLLECTION_OID_LB + 1;
for (i = 0; i < 6; i++)
idlist[i] = oid + i;
for (i = 0; i < 4; i++)
idlist[6+i] = 0;
while (data_out_len > 0) {
assert(ismember(get_ntohll(cp), idlist, 10));
cp += 8;
data_out_len -= 8;
}
free(data_out);
data_out = NULL;
osd_command_attr_free(&cmd);
page = USEROBJECT_PG + LUN_PG_LB;
number = 1;
struct attribute_list getattr[] = {
{ATTR_GET, page, number, NULL, 0, 0},
{ATTR_GET, page+1, number+1, NULL, 0, 0},
{ATTR_GET, page+2, number+2, NULL, 0, 0},
{ATTR_GET, page+3, number+3, NULL, 0, 0},
{ATTR_GET, page+4, number+4, NULL, 0, 0},
{ATTR_GET, page+5, number+5, NULL, 0, 0},
};
/* execute list with attr */
test_oids_with_attr(osd, pid, getattr, 6, 4096, 792, 784, 0,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* execute list with attr, alloc length less than required */
test_oids_with_attr(osd, pid, getattr, 6, 200, 200, 784, 65539,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* execute list with attr, alloc length less than required */
test_oids_with_attr(osd, pid, getattr, 6, 208, 208, 784, 65539,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* execute list with attr, alloc length less than required */
test_oids_with_attr(osd, pid, getattr, 6, 216, 208, 784, 65539,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* execute list with attr, alloc length less than required */
test_oids_with_attr(osd, pid, getattr, 6, 544, 536, 784, 65544,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* execute list with attr, alloc length less than required */
test_oids_with_attr(osd, pid, getattr, 6, 688, 688, 784, 65546,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* execute list with attr, alloc length less than required */
test_oids_with_attr(osd, pid, getattr, 6, 680, 680, 784, 65546,
(0x22 << 2), attrs, ARRAY_SIZE(attrs));
/* clean up */
oid = USEROBJECT_OID_LB;
for (i=0; i<12; i++) {
if (i == 0 || i == 7)
continue;
ret = osd_command_set_remove(&cmd, pid, oid + i);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
}
cid = COLLECTION_OID_LB;
ret = osd_command_set_remove_collection(&cmd, pid, cid, 0);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
cid = COLLECTION_OID_LB + 7;
ret = osd_command_set_remove_collection(&cmd, pid, cid, 0);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
ret = osd_command_set_remove_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
}
static void test_attr_vals(uint8_t *cp, struct attribute_list *attrs,
size_t sz)
{
size_t i = 0;
uint32_t page = 0;
uint32_t num = 0;
uint16_t len = 0;
uint32_t list_len = 0;
assert((cp[0] & 0x0F) == 0x9);
cp += 4;
list_len = get_ntohl(cp);
cp += 4;
while (list_len > 0) {
page = get_ntohl(cp);
cp += 4;
num = get_ntohl(cp);
cp += 4;
len = get_ntohs(cp);
cp += 2;
for (i = 0; i < sz; i++) {
if (!(attrs[i].page==page && attrs[i].number==num))
continue;
assert(len == attrs[i].len);
if (len == 8) {
assert(get_ntohll(attrs[i].val) ==
get_ntohll(cp));
} else if (len != 0) {
assert(memcmp(attrs[i].val, cp, len) == 0);
}
break;
}
assert(i < sz);
if (len == 0) {
cp += (roundup8(10) - 10);
list_len -= roundup8(4+4+2);
} else {
cp += len;
cp += (roundup8(2+len) - (2+len));
list_len -= roundup8(4+4+2+len);
}
}
assert(list_len == 0);
}
void test_set_member_attributes(struct osd_device *osd)
{
struct osd_command cmd;
uint64_t pid = PARTITION_PID_LB;
uint64_t cid = COLLECTION_OID_LB;
uint64_t oid = 0;
uint8_t *data_out = NULL;
uint32_t page = 0;
const void *data_in;
uint64_t data_out_len, data_in_len;
uint8_t sense_out[OSD_MAX_SENSE];
int senselen_out;
int i, ret;
data_out = NULL;
data_out_len = 0;
/* create partition */
ret = osd_command_set_create_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create collection */
ret = osd_command_set_create_collection(&cmd, pid, cid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create 100 objects */
ret = osd_command_set_create(&cmd, pid, 0, 100);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(osd->ccap.oid == (USEROBJECT_OID_LB + 100));
/* put odd objects into the collection */
oid = USEROBJECT_OID_LB + 1;
for (i = 0; i < 100; i += 2) {
uint64_t attrval;
struct attribute_list attr = {
.type = ATTR_SET,
.page = USER_COLL_PG,
.number = 1,
.len = sizeof(attrval),
.val = &attrval,
};
set_htonll(&attrval, cid);
ret = osd_command_set_set_attributes(&cmd, pid, oid + i);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, &attr, 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
}
/* set attr on collection members */
uint64_t val1 = 123454321, val2 = 987654, val3 = 59999999;
char str1[MAXNAMELEN], str2[MAXNAMELEN], str3[MAXNAMELEN];
set_htonll(&val1, val1);
set_htonll(&val2, val2);
set_htonll(&val3, val3);
sprintf(str1, "GoMtI");
sprintf(str2, "DeViL");
sprintf(str3, "homeopath");
page = USEROBJECT_PG + LUN_PG_LB;
struct attribute_list attrs[] = {
{ATTR_SET, page+1, 2, str1, strlen(str1)+1, 0},
{ATTR_SET, page+21, 4, &val1, sizeof(val1), 0},
{ATTR_SET, page+5, 55, &val2, sizeof(val2), 0},
{ATTR_SET, page+666, 66, str2, strlen(str2)+1, 0},
{ATTR_SET, page+10, 10, str3, strlen(str3)+1, 0},
{ATTR_SET, page+2, 3, &val3, sizeof(val3), 0},
};
ret = osd_command_set_set_member_attributes(&cmd, pid, cid);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, attrs, 6);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
osd_command_attr_free(&cmd);
/* randomly select 5 objects, get their attrs and test */
for (i = 0; i < 6; i++) {
attrs[i].type = ATTR_GET;
}
srand(time(0));
for (i = 0; i < 5; i++) {
int r = (int)(50.0 * (rand()/(RAND_MAX+1.0)));
oid = USEROBJECT_OID_LB + 1 + 2*r;
ret = osd_command_set_get_attributes(&cmd, pid, oid);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, attrs, 6);
assert(ret == 0);
data_in = cmd.outdata;
data_in_len = cmd.outlen;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len,
&data_out, &data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
test_attr_vals(data_out, attrs, 6);
osd_command_attr_free(&cmd);
}
free(data_out);
data_out = NULL;
data_out_len = 0;
/* clean up */
oid = USEROBJECT_OID_LB + 1;
for (i=0; i<100; i++) {
ret = osd_command_set_remove(&cmd, pid, oid + i);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
}
cid = COLLECTION_OID_LB;
ret = osd_command_set_remove_collection(&cmd, pid, cid, 0);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
ret = osd_command_set_remove_partition(&cmd, pid);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
}
void test_atomics(struct osd_device *osd)
{
int ret = 0;
struct osd_command cmd;
int senselen_out;
uint8_t sense_out[OSD_MAX_SENSE];
uint8_t *cp = NULL;
uint8_t *data_out = NULL;
void *data_in = NULL;
uint64_t data_out_len, data_in_len;
/* create partition + empty getpage_setlist */
ret = osd_command_set_create_partition(&cmd, PARTITION_PID_LB);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
/* create 1 object */
ret = osd_command_set_create(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB, 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out, &data_out_len,
sense_out, &senselen_out);
assert(ret == 0);
/* cas */
ret = osd_command_set_cas(&cmd, USEROBJECT_PID_LB, USEROBJECT_OID_LB,
8UL, 0);
assert(ret == 0);
data_in = Malloc(1024);
assert(data_in != NULL);
cp = data_in;
set_htonll(&cp[0], 0UL);
set_htonll(&cp[8], 5UL);
data_in_len = 16;
data_out = NULL;
data_out_len = 0;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(get_ntohll(&data_out[0]) == 0UL);
free(data_out);
data_out = NULL;
data_out_len = 0;
ret = osd_command_set_cas(&cmd, USEROBJECT_PID_LB, USEROBJECT_OID_LB,
8UL, 0);
assert(ret == 0);
cp = data_in;
set_htonll(&cp[0], 5UL);
set_htonll(&cp[8], 0UL);
data_in_len = 16;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(get_ntohll(&data_out[0]) == 5UL);
free(data_out);
data_out = NULL;
data_out_len = 0;
/* fa */
ret = osd_command_set_fa(&cmd, USEROBJECT_PID_LB, USEROBJECT_OID_LB,
8UL, 0);
assert(ret == 0);
cp = data_in;
set_htonll(&cp[0], 4UL);
data_in_len = 8;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(get_ntohll(&data_out[0]) == 0UL);
free(data_out);
data_out = NULL;
data_out_len = 0;
ret = osd_command_set_fa(&cmd, USEROBJECT_PID_LB, USEROBJECT_OID_LB,
8UL, 0);
assert(ret == 0);
cp = data_in;
set_htonll(&cp[0], 16UL);
data_in_len = 8;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(get_ntohll(&data_out[0]) == 4UL);
free(data_out);
data_out = NULL;
data_out_len = 0;
ret = osd_command_set_fa(&cmd, USEROBJECT_PID_LB, USEROBJECT_OID_LB,
8UL, 0);
assert(ret == 0);
cp = data_in;
set_htonll(&cp[0], -20L);
data_in_len = 8;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(get_ntohll(&data_out[0]) == 20UL);
free(data_out);
data_out = NULL;
data_out_len = 0;
ret = osd_command_set_fa(&cmd, USEROBJECT_PID_LB, USEROBJECT_OID_LB,
8UL, 0);
assert(ret == 0);
cp = data_in;
set_htonll(&cp[0], 1L);
data_in_len = 8;
ret = osdemu_cmd_submit(osd, cmd.cdb, data_in, data_in_len, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(get_ntohll(&data_out[0]) == 0UL);
free(data_out);
data_out = NULL;
data_out_len = 0;
free(data_in);
/* gen_cas */
ret = osd_command_set_gen_cas(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB);
assert(ret == 0);
char str1[MAXNAMELEN];
sprintf(str1, "some arbit string");
struct attribute_list attr[] = {
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB, 1, NULL, 0, 0},
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB, 1, str1, strlen(str1)+1,
0
},
{ATTR_RESULT, USEROBJECT_PG+LUN_PG_LB, 1, NULL, 0, 0}
};
ret = osd_command_attr_build(&cmd, attr, 3);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
assert(data_out != NULL);
assert(data_out_len == 24);
attr[2].len = 0;
test_attr_vals(data_out, &attr[2], 1);
osd_command_attr_free(&cmd);
free(data_out);
data_out = NULL;
data_out_len = 0;
ret = osd_command_set_gen_cas(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB);
assert(ret == 0);
char str2[MAXNAMELEN];
sprintf(str2, "a diff str");
struct attribute_list attr1[] = {
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB, 1, str1, strlen(str1)+1,
0
},
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB, 1, str2, strlen(str2)+1,
0
},
{ATTR_RESULT, USEROBJECT_PG+LUN_PG_LB, 1, str1, strlen(str1)+1,
0
}
};
ret = osd_command_attr_build(&cmd, attr1, 3);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
assert(data_out != NULL);
test_attr_vals(data_out, &attr1[2], 1);
osd_command_attr_free(&cmd);
free(data_out);
data_out = NULL;
data_out_len = 0;
/* cond setattr */
ret = osd_command_set_cond_setattr(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB);
assert(ret == 0);
char str3[MAXNAMELEN];
sprintf(str3, "setattr str");
struct attribute_list attr2[] = {
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB, 1, str2, strlen(str2)+1,
0
}, /* cmp */
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB, 1, str1, strlen(str1)+1,
0
}, /* swap */
{ATTR_SET, USEROBJECT_PG+LUN_PG_LB+100, 23, str3,
strlen(str3)+1, 0
},
{ATTR_RESULT, USEROBJECT_PG+LUN_PG_LB, 1, str2, strlen(str2)+1,
0
}
};
ret = osd_command_attr_build(&cmd, attr2, 4);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
assert(data_out != NULL);
test_attr_vals(data_out, &attr2[3], 1);
osd_command_attr_free(&cmd);
free(data_out);
data_out = NULL;
data_out_len = 0;
attr2[2].type = ATTR_GET;
ret = osd_command_set_get_attributes(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB);
assert(ret == 0);
ret = osd_command_attr_build(&cmd, &attr2[2], 1);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, cmd.outdata, cmd.outlen,
&data_out, &data_out_len, sense_out,
&senselen_out);
assert(ret == 0);
assert(data_out != NULL);
test_attr_vals(data_out, &attr2[2], 1);
osd_command_attr_free(&cmd);
free(data_out);
data_out = NULL;
data_out_len = 0;
/* clean up */
ret = osd_command_set_remove(&cmd, USEROBJECT_PID_LB,
USEROBJECT_OID_LB);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
ret = osd_command_set_remove_partition(&cmd, PARTITION_PID_LB);
assert(ret == 0);
ret = osdemu_cmd_submit(osd, cmd.cdb, NULL, 0, &data_out,
&data_out_len, sense_out, &senselen_out);
assert(ret == 0);
}
static void set_member_attributes_speed(struct osd_device *osd, int numiter,
int numobj, int numobj_in_coll,
int numattr, int numattr_set)
{
struct osd_command c;
int i;
uint64_t start, end;
double *v;
uint64_t pid = PARTITION_PID_LB;
uint64_t cid = COLLECTION_OID_LB;
double mu, sd;
struct attribute_list *attr;
uint8_t *attr_val;
uint8_t cidn[8];
if (numobj_in_coll > numobj)
return;
if (numattr_set > numattr)
return;
v = Malloc(numiter * sizeof(*v));
if (!v)
return;
attr = Malloc((numattr + 1) * sizeof(*attr));
if (!attr)
return;
attr_val = Malloc(ATTR_LEN);
if (!attr_val)
return;
memset(attr_val, 0x5a, ATTR_LEN);
osd_command_set_create_partition(&c, pid);
run(osd, &c);
osd_command_set_create_collection(&c, pid, cid);
run(osd, &c);
for (i=0; i<numattr; i++) {
attr[i].type = ATTR_SET;
attr[i].page = LUN_PG_LB;
attr[i].number = 1 + i; /* skip directory */
attr[i].val = attr_val;
attr[i].len = ATTR_LEN;
}
/* one extra attr to stick it in the collection, for some of them */
set_htonll(cidn, cid);
attr[numattr].type = ATTR_SET;
attr[numattr].page = USER_COLL_PG;
attr[numattr].number = 1;
attr[numattr].len = 8;
attr[numattr].val = cidn;
for (i=0; i<numobj; i++) {
int this_numattr;
osd_command_set_create(&c, pid, 0, 1);
this_numattr = numattr;
if (i < numobj_in_coll)
++this_numattr; /* stick in coll */
osd_command_attr_build(&c, attr, this_numattr);
run(osd, &c);
osd_command_attr_free(&c);
}
osd_command_set_set_member_attributes(&c, pid, cid);
osd_command_attr_build(&c, attr, numattr_set);
for (i=0; i<5; i++)
run(osd, &c);
for (i=0; i<numiter; i++) {
rdtsc(start);
run(osd, &c);
rdtsc(end);
v[i] = (double)(end - start) / mhz;
}
osd_command_attr_free(&c);
mu = mean(v, numiter);
sd = stddev(v, mu, numiter);
printf("set_member_attributes numiter %d numobj %d numobj_in_coll %d"
" numattr %d numattr_set %d"
" avg %lf +/- %lf us\n", numiter, numobj, numobj_in_coll,
numattr, numattr_set, mu, sd);
free(attr_val);
free(attr);
free(v);
}