static int ipc_mgmt_req(struct tgtadm_req *req, struct concat_buf *b) { int err, fd = 0, done = 0; req->len = sizeof(*req) + b->size; err = ipc_mgmt_connect(&fd); if (err < 0) { eprintf("can't connect to tgt daemon, %m\n"); goto out; } err = write(fd, req, sizeof(*req)); if (err < 0 || err != sizeof(*req)) { eprintf("failed to send request hdr to tgt daemon, %m\n"); err = errno; goto out; } while (done < b->size) { err = concat_write(b, fd, done); if (err > 0) done += err; else if (errno != EAGAIN) { eprintf("failed to send request buf to " "tgt daemon, %m\n"); err = errno; goto out; } } dprintf("sent to tgtd %d\n", req->len); err = ipc_mgmt_rsp(fd, req); out: if (fd > 0) close(fd); concat_buf_release(b); return err; }
static void mtask_recv_send_handler(int fd, int events, void *data) { int err, len; char *p; struct mgmt_task *mtask = data; struct tgtadm_req *req = &mtask->req; struct tgtadm_rsp *rsp = &mtask->rsp; switch (mtask->mtask_state) { case MTASK_STATE_HDR_RECV: len = sizeof(*req) - mtask->done; err = read(fd, (char *)req + mtask->done, len); if (err > 0) { mtask->done += err; if (mtask->done == sizeof(*req)) { mtask->req_bsize = req->len - sizeof(*req); if (!mtask->req_bsize) { err = mtask_received(mtask, fd); if (err) goto out; } else { /* the pdu exists */ if (mtask->req_bsize > MAX_MGT_BUFSIZE) { eprintf("mtask buffer len: %d too large\n", mtask->req_bsize); mtask->req_bsize = 0; goto out; } mtask->req_buf = zalloc(mtask->req_bsize); if (!mtask->req_buf) { eprintf("can't allocate mtask buffer len: %d\n", mtask->req_bsize); mtask->req_bsize = 0; goto out; } mtask->mtask_state = MTASK_STATE_PDU_RECV; mtask->done = 0; } } } else if (errno != EAGAIN) goto out; break; case MTASK_STATE_PDU_RECV: len = mtask->req_bsize - mtask->done; err = read(fd, mtask->req_buf + mtask->done, len); if (err > 0) { mtask->done += err; if (mtask->done == mtask->req_bsize) { err = mtask_received(mtask, fd); if (err) goto out; } } else if (errno != EAGAIN) goto out; break; case MTASK_STATE_HDR_SEND: p = (char *)rsp + mtask->done; len = sizeof(*rsp) - mtask->done; err = write(fd, p, len); if (err > 0) { mtask->done += err; if (mtask->done == sizeof(*rsp)) { if (rsp->len == sizeof(*rsp)) goto out; mtask->done = 0; mtask->mtask_state = MTASK_STATE_PDU_SEND; } } else if (errno != EAGAIN) goto out; break; case MTASK_STATE_PDU_SEND: err = concat_write(&mtask->rsp_concat, fd, mtask->done); if (err >= 0) { mtask->done += err; if (mtask->done == (rsp->len - sizeof(*rsp))) goto out; } else if (errno != EAGAIN) goto out; break; default: eprintf("unknown state %d\n", mtask->mtask_state); } return; out: if (req->mode == MODE_SYSTEM && req->op == OP_DELETE && !rsp->err) system_active = 0; tgt_event_del(fd); close(fd); mtask_free(mtask); }