* or implied, of GRNET S.A.
*/
+#define _GNU_SOURCE
#include <stdio.h>
+#include <stdlib.h>
#include <unistd.h>
+#include <sys/syscall.h>
#include <sys/types.h>
#include <pthread.h>
#include <xseg/xseg.h>
#include <peer.h>
#include <time.h>
#include <sys/util.h>
+#include <signal.h>
+#include <bench-xseg.h>
struct timespec delay = {0, 4000000};
{
fprintf(stderr, "Custom peer options: \n"
" --------------------------------------------\n"
- " -ts | No | Total I/O size\n"
- " -os | 4MB | Object size\n"
+ " -ts | None | Total I/O size\n"
+ " -os | 4M | Object size\n"
+ " -bs | 4k | Block size\n"
+ " -dp | None | Destination port\n"
" --iodepth | 1 | Number of in-flight I/O requests\n"
"\n");
}
+
+/*
+ * Convert string to size in bytes.
+ * If syntax is invalid, return 0. Values such as zero and non-integer
+ * multiples of segment's page size should not be accepted.
+ */
+static uint64_t str2num(char *str)
+{
+ char *unit;
+ uint64_t num;
+
+ num = strtoll(str, &unit, 10);
+ if (strlen(unit) > 1) //Invalid syntax
+ return 0;
+ else if (strlen(unit) < 1) //Plain number in bytes
+ return num;
+
+ switch (*unit) {
+ case 'g':
+ case 'G':
+ num *= 1024;
+ case 'm':
+ case 'M':
+ num *= 1024;
+ case 'k':
+ case 'K':
+ num *= 1024;
+ break;
+ default:
+ num = 0;
+ }
+ return num;
+}
+
int custom_peer_init(struct peerd *peer, int argc, char *argv[])
{
+ struct bench *prefs;
+ char total_size[MAX_ARG_LEN + 1];
+ char object_size[MAX_ARG_LEN + 1];
+ char block_size[MAX_ARG_LEN + 1];
+ struct xseg *xseg = peer->xseg;
+ unsigned int xseg_page_size = 1 << xseg->config.page_shift;
+ long dst_port = -1;
+
+ total_size[0] = 0;
+ block_size[0] = 0;
+ object_size[0] = 0;
+
+ prefs = malloc(sizeof(struct bench));
+ if (!prefs) {
+ perror("malloc");
+ return -1;
+ }
+
+ //Begin reading the benchmark-specific arguments
+ BEGIN_READ_ARGS(argc, argv);
+ READ_ARG_STRING("-ts", total_size, MAX_ARG_LEN);
+ READ_ARG_STRING("-os", object_size, MAX_ARG_LEN);
+ READ_ARG_STRING("-bs", block_size, MAX_ARG_LEN);
+ READ_ARG_ULONG("--iodepth", prefs->iodepth);
+ READ_ARG_ULONG("-dp", dst_port);
+ END_READ_ARGS();
+
+ /*************************
+ * Check size parameters *
+ *************************/
+
+ //Block size (bs): Defaults to 4K.
+ //It must be a number followed by one of these characters: [k|K|m|M|g|G].
+ //If not, it will be considered as size in bytes.
+ //Must be integer multiple of segment's page size (typically 4k).
+ if (!block_size[0])
+ strcpy(block_size,"4k");
+
+ prefs->bs = str2num(block_size);
+ if (!prefs->bs) {
+ XSEGLOG2(&lc, E, "Invalid syntax: %s\n", block_size);
+ goto arg_fail;
+ } else if (prefs->bs % xseg_page_size) {
+ XSEGLOG2(&lc, E, "Misaligned block size: %s\n", block_size);
+ goto arg_fail;
+ }
+
+ //Total I/O size (ts): Must be supplied by user.
+ //Must have the same format as "total size"
+ //Must be integer multiple of "block size"
+ if (!total_size[0]) {
+ XSEGLOG2(&lc, E, "Total I/O size needs to be supplied\n");
+ goto arg_fail;
+ }
+
+ prefs->ts = str2num(total_size);
+ if (!prefs->ts) {
+ XSEGLOG2(&lc, E, "Invalid syntax: %s\n", total_size);
+ goto arg_fail;
+ } else if (prefs->ts % prefs->bs) {
+ XSEGLOG2(&lc, E, "Misaligned total I/O size: %s\n", total_size);
+ goto arg_fail;
+ } else if (prefs->ts > xseg->segment_size) {
+ XSEGLOG2(&lc, E, "Total I/O size exceeds segment size\n", total_size);
+ goto arg_fail;
+ }
+
+ //Object size (os): Defaults to 4M.
+ //Must have the same format as "total size"
+ //Must be integer multiple of "block size"
+ if (!object_size[0])
+ strcpy(object_size,"4M");
+
+ prefs->os = str2num(object_size);
+ if (!prefs->os) {
+ XSEGLOG2(&lc, E, "Invalid syntax: %s\n", object_size);
+ goto arg_fail;
+ } else if (prefs->os % prefs->bs) {
+ XSEGLOG2(&lc, E, "Misaligned object size: %s\n", object_size);
+ goto arg_fail;
+ }
+
+ /*************************
+ * Check port parameters *
+ *************************/
+
+ if (dst_port < 0){
+ XSEGLOG2(&lc, E, "Destination port needs to be supplied\n");
+ goto arg_fail;
+ }
+
+ prefs->src_port = peer->portno_start; //TODO: allow user to change this
+ prefs->dst_port = (xport) dst_port;
+
+ /**************************
+ * Customize struct peerd *
+ **************************/
+
+ prefs->total_tm = malloc(sizeof(struct timer));
+ prefs->get_tm = malloc(sizeof(struct timer));
+ prefs->sub_tm = malloc(sizeof(struct timer));
+ prefs->rec_tm = malloc(sizeof(struct timer));
+ if (!prefs->total_tm || !prefs->get_tm || !prefs->sub_tm ||
+ !prefs->rec_tm) {
+ perror("malloc");
+ return -1;
+ }
+ peer->custom_peerd_loop = custom_peerd_loop;
+ peer->priv = (void *) prefs;
+ return 0;
+
+arg_fail:
+ free(prefs);
+ custom_peer_usage();
+ return -1;
+}
+
+
+int send_request(struct peerd *peer, struct bench *prefs)
+{
+ struct xseg_request *req;
+ struct xseg *xseg = peer->xseg;
+ xport srcport = prefs->src_port;
+ xport dstport = prefs->dst_port;
+ xport p;
+
+ int r;
+ uint32_t targetlen=10; //FIXME: handle it better
+ uint64_t size = prefs->os;
+
+ //srcport and dstport must already be provided by the user.
+ //returns struct xseg_request with basic initializations
+ req = xseg_get_request(xseg, srcport, dstport, X_ALLOC);
+ if (!req) {
+ fprintf(stderr, "No request\n");
+ return -1;
+ }
+
+ //Allocate enough space for the data and the target's name
+ r = xseg_prep_request(xseg, req, targetlen, size);
+ if (r < 0) {
+ fprintf(stderr, "Cannot prepare request! (%lu, %llu)\n",
+ (unsigned long)targetlen, (unsigned long long)size);
+ xseg_put_request(xseg, req, srcport);
+ return -1;
+ }
+
+#if 0
+ //TODO: allow strcpy, memcpy
+ //Copy target's name to the newly allocated space
+ req_target = xseg_get_target(xseg, req);
+ strncpy(req_target, target, targetlen);
+
+ //Copy data buffer to the newly allocated space
+ req_data = xseg_get_data(xseg, req);
+ memcpy(req_data, buf, size);
+ req->offset = offset;
+ req->size = size;
+ req->op = X_WRITE;
+#endif
+
+ //Submit the request from the source port to the target port
+ timer_start(prefs->sub_tm);
+ p = xseg_submit(xseg, req, srcport, X_ALLOC);
+ if (p == NoPort) {
+ fprintf(stderr, "Cannot submit\n");
+ return -1;
+ }
+ timer_stop(prefs->sub_tm);
+
+ //Send SIGIO to the process that has binded this port to inform that
+ //IO is possible
+ xseg_signal(xseg, p);
+
+ return 0;
+}
+
+/*
+ * This function substitutes the default peerd_loop of peer.c.
+ * It's plugged to struct peerd at custom peer's initialisation
+ */
+int custom_peerd_loop(struct peerd *peer)
+{
+#ifdef MT
+ int i;
+ if (peer->interactive_func)
+ peer->interactive_func();
+ for (i = 0; i < peer->nr_threads; i++) {
+ pthread_join(peer->thread[i].tid, NULL);
+ }
+#else
+ struct xseg *xseg = peer->xseg;
+ struct bench *prefs = peer->priv;
+
+ xport portno_start = peer->portno_start;
+ xport portno_end = peer->portno_end;
+ uint64_t threshold=1000/(1 + portno_end - portno_start);
+ pid_t pid =syscall(SYS_gettid);
+ XSEGLOG2(&lc, I, "Peer has tid %u.\n", pid);
+ xseg_init_local_signal(xseg, peer->portno_start);
+ uint64_t loops;
+
+ uint64_t remaining = prefs->ts;
+
+ while (!isTerminate()
+ && xq_count(&peer->free_reqs) == peer->nr_ops
+ && remaining) {
+
+ while (prefs->sub_tm->completed - prefs->sub_tm->completed <
+ prefs->iodepth){
+ send_request(peer, prefs);
+ }
+
+ for (loops = threshold; loops > 0; loops--) {
+ if (loops == 1)
+ xseg_prepare_wait(xseg, peer->portno_start);
+ if (check_ports(peer))
+ loops = threshold;
+ }
+#ifdef ST_THREADS
+ if (ta){
+ st_sleep(0);
+ continue;
+ }
+#endif
+ XSEGLOG2(&lc, I, "Peer goes to sleep\n");
+ xseg_wait_signal(xseg, 10000000UL);
+ xseg_cancel_wait(xseg, peer->portno_start);
+ XSEGLOG2(&lc, I, "Peer woke up\n");
+ }
+ custom_peer_finalize(peer);
+ xseg_quit_local_signal(xseg, peer->portno_start);
+#endif
return 0;
}
else {
// printf("completing req id: %u (remote %u)\n", (unsigned int) (pr - peer->peer_reqs), (unsigned int) pr->req->priv);
// nanosleep(&delay,NULL);
- print_req(peer->xseg, pr->req);
+// print_req(peer->xseg, pr->req);
complete(peer, pr);
}
return 0;