// ------- * MAIN * -------
int main(int argc, char *argv[]) {
int status = 0;
struct parsed_url p_url = {{0}};
int sock = 0;
int64_t filesize = 0;
FILE *out_file = NULL;
// parse args
if (argc == 3 && strcmp(argv[2], "--debug") == 0)
DEBUG = 1;
else if (argc != 2) {
status = 1;
fprintf(stderr, "Error: Invalid argument.\n");
printf("%s URL [--debug]\n URL: host:port/filename\n\n", argv[0]);
goto quit;
}
if ((status = parse_url(argv[1], &p_url)) != 0) {
fprintf(stderr, "Error: Failed to parse given URL.\n");
goto quit;
}
// --------
// open socket
if (DEBUG) printf("Opening socket...\n");
if ((sock = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
fprintf(stderr, "Error: Failed to open stream socket.\n");
goto quit;
}
// --------
// connect to server
if (DEBUG) printf("Connecting to server...\n");
status = connect_to_server(p_url.hostname, p_url.port, sock);
if (status != 0) {
fprintf(stderr, "Error: Failed to connect to server.\n");
goto close_socket;
}
// send file request; get status and filesize
if (DEBUG) printf("Exchanging initial information with server...\n");
status = exchange_info(p_url.filename, &filesize, sock);
if (status != 0) {
if (status == -1)
fprintf(stderr, "Error: File '%s' not found on server.\n", p_url.filename);
else if (status == -2)
fprintf(stderr, "Error: Server is busy, try again later.\n");
else if (status == -3)
fprintf(stderr, "Error: Failed to open file '%s' on server.\n", p_url.filename);
else
fprintf(stderr, "Error: Failed during initial data exchange with server.\n");
goto close_socket;
}
// --------
// open file
if (DEBUG) printf("Opening empty file...\n");
out_file = fopen(p_url.filename, "w");
if (out_file == NULL) {
fprintf(stderr, "Error: Failed to create file '%s' in current working directory.\n", p_url.filename);
goto close_socket;
}
// --------
// download data to file
printf("Downloading file...\n");
status = download_file(sock, out_file, filesize);
if (status != 0) {
fprintf(stderr, "Error: Failed to download file.\n");
}
else
printf("Download complete.\n");
// --------
// cleanup
fclose(out_file);
close_socket:
close(sock);
// --------
quit:
if (status != 0)
return EXIT_FAILURE;
return EXIT_SUCCESS;
}
static void verbs_add_device (struct ib_device *dev)
{
int ret;
struct ib_qp_init_attr attrs;
if (ib_dev)
return;
/* durty hack for ib_dma_map_single not to segfault */
dev->dma_ops = NULL;
ib_dev = dev;
printk (KERN_INFO "IB add device called. Name = %s\n", dev->name);
ret = ib_query_device (dev, &dev_attr);
if (ret) {
printk (KERN_INFO "ib_quer_device failed: %d\n", ret);
return;
}
printk (KERN_INFO "IB device caps: max_qp %d, max_mcast_grp: %d, max_pkeys: %d\n",
dev_attr.max_qp, dev_attr.max_mcast_grp, (int)dev_attr.max_pkeys);
/* We'll work with first port. It's a sample module, anyway. Who is that moron which decided
* to count ports from one? */
ret = ib_query_port (dev, 1, &port_attr);
if (ret) {
printk (KERN_INFO "ib_query_port failed: %d\n", ret);
return;
}
printk (KERN_INFO "Port info: lid: %u, sm_lid: %u, max_msg_size: %u\n",
(unsigned)port_attr.lid, (unsigned)port_attr.sm_lid, port_attr.max_msg_sz);
pd = ib_alloc_pd (dev);
if (IS_ERR (pd)) {
ret = PTR_ERR (pd);
printk (KERN_INFO "pd allocation failed: %d\n", ret);
return;
}
printk (KERN_INFO "PD allocated\n");
mr = ib_get_dma_mr (pd, IB_ACCESS_LOCAL_WRITE);
if (IS_ERR (mr)) {
ret = PTR_ERR (mr);
printk (KERN_INFO "get_dma_mr failed: %d\n", ret);
return;
}
send_cq = ib_create_cq (dev, NULL, NULL, NULL, 1, 1);
if (IS_ERR (send_cq)) {
ret = PTR_ERR (send_cq);
printk (KERN_INFO "ib_create_cq failed: %d\n", ret);
return;
}
recv_cq = ib_create_cq (dev, verbs_comp_handler_recv, NULL, NULL, 1, 1);
if (IS_ERR (recv_cq)) {
ret = PTR_ERR (recv_cq);
printk (KERN_INFO "ib_create_cq failed: %d\n", ret);
return;
}
ib_req_notify_cq (recv_cq, IB_CQ_NEXT_COMP);
printk (KERN_INFO "CQs allocated\n");
ib_query_pkey (dev, 1, 0, &pkey);
/* allocate memory */
send_buf = kmalloc (buf_size + 40, GFP_KERNEL);
recv_buf = kmalloc (buf_size + 40, GFP_KERNEL);
if (!send_buf || !recv_buf) {
printk (KERN_INFO "Memory allocation error\n");
return;
}
printk (KERN_INFO "Trying to register regions\n");
if (ib_dev->dma_ops)
printk (KERN_INFO "DMA ops are defined\n");
memset (send_buf, 0, buf_size+40);
memset (send_buf, 0, buf_size+40);
send_key = ib_dma_map_single (ib_dev, send_buf, buf_size, DMA_FROM_DEVICE);
printk (KERN_INFO "send_key obtained %llx\n", send_key);
recv_key = ib_dma_map_single (ib_dev, recv_buf, buf_size, DMA_TO_DEVICE);
printk (KERN_INFO "recv_key obtained %llx\n", recv_key);
if (ib_dma_mapping_error (ib_dev, send_key)) {
printk (KERN_INFO "Error mapping send buffer\n");
return;
}
if (ib_dma_mapping_error (ib_dev, recv_key)) {
printk (KERN_INFO "Error mapping recv buffer\n");
return;
}
memset (&attrs, 0, sizeof (attrs));
attrs.qp_type = IB_QPT_UD;
attrs.sq_sig_type = IB_SIGNAL_ALL_WR;
attrs.event_handler = verbs_qp_event;
attrs.cap.max_send_wr = CQ_SIZE;
attrs.cap.max_recv_wr = CQ_SIZE;
attrs.cap.max_send_sge = 1;
attrs.cap.max_recv_sge = 1;
attrs.send_cq = send_cq;
attrs.recv_cq = recv_cq;
qp = ib_create_qp (pd, &attrs);
if (IS_ERR (qp)) {
ret = PTR_ERR (qp);
printk (KERN_INFO "qp allocation failed: %d\n", ret);
return;
}
printk (KERN_INFO "Create QP with num %x\n", qp->qp_num);
if (init_qp (qp)) {
printk (KERN_INFO "Failed to initialize QP\n");
return;
}
ret = ib_query_gid (ib_dev, 1, 0, &local_info.gid);
if (ret) {
printk (KERN_INFO "query_gid failed %d\n", ret);
return;
}
local_info.qp_num = qp->qp_num;
local_info.lid = port_attr.lid;
/* now we are ready to send our QP number and other stuff to other party */
if (!server_addr) {
schedule_work (&sock_accept);
flush_scheduled_work ();
}
else
exchange_info (server_addr);
if (!have_remote_info) {
printk (KERN_INFO "Have no remote info, give up\n");
return;
}
ret = path_rec_lookup_start ();
if (ret) {
printk (KERN_INFO "path_rec lookup start failed: %d\n", ret);
return;
}
/* post receive request */
verbs_post_recv_req ();
mod_timer (&verbs_timer, NEXTJIFF(1));
}
