sys/netsmb/smb_rq.c

53 static int  smb_rq_reply(struct smb_rq *rqp);
54 static int  smb_rq_enqueue(struct smb_rq *rqp);
57 static int  smb_rq_new(struct smb_rq *rqp, u_char cmd);
64 	struct smb_rq *rqp;
67 	rqp = malloc(sizeof(*rqp), M_SMBRQ, M_WAITOK);
68 	if (rqp == NULL)
70 	error = smb_rq_init(rqp, layer, cmd, scred);
71 	rqp->sr_flags |= SMBR_ALLOCED;
73 		smb_rq_done(rqp);
76 	*rqpp = rqp;
83 smb_rq_init(struct smb_rq *rqp, struct smb_connobj *layer, u_char cmd,
88 	bzero(rqp, sizeof(*rqp));
89 	smb_sl_init(&rqp->sr_slock, "srslock");
90 	error = smb_rq_getenv(layer, &rqp->sr_vc, &rqp->sr_share);
93 	error = smb_vc_access(rqp->sr_vc, scred, SMBM_EXEC);
96 	if (rqp->sr_share) {
97 		error = smb_share_access(rqp->sr_share, scred, SMBM_EXEC);
101 	rqp->sr_cred = scred;
102 	rqp->sr_mid = smb_vc_nextmid(rqp->sr_vc);
103 	return smb_rq_new(rqp, cmd);
107 smb_rq_new(struct smb_rq *rqp, u_char cmd)
109 	struct smb_vc *vcp = rqp->sr_vc;
110 	struct mbchain *mbp = &rqp->sr_rq;
114 	rqp->sr_sendcnt = 0;
116 	md_done(&rqp->sr_rp);
132 		rqp->sr_rqsig = NULL;
135 		rqp->sr_rqsig = (u_int8_t *)mb_reserve(mbp, 8);
138 	rqp->sr_rqtid = mb_reserve(mbp, sizeof(u_int16_t));
140 	rqp->sr_rquid = mb_reserve(mbp, sizeof(u_int16_t));
141 	mb_put_uint16le(mbp, rqp->sr_mid);
146 smb_rq_done(struct smb_rq *rqp)
148 	mb_done(&rqp->sr_rq);
149 	md_done(&rqp->sr_rp);
150 	smb_sl_destroy(&rqp->sr_slock);
151 	if (rqp->sr_flags & SMBR_ALLOCED)
152 		free(rqp, M_SMBRQ);
159 smb_rq_simple(struct smb_rq *rqp)
161 	struct smb_vc *vcp = rqp->sr_vc;
165 		rqp->sr_flags &= ~SMBR_RESTART;
166 		rqp->sr_timo = vcp->vc_timo;
167 		rqp->sr_state = SMBRQ_NOTSENT;
168 		error = smb_rq_enqueue(rqp);
171 		error = smb_rq_reply(rqp);
174 		if ((rqp->sr_flags & (SMBR_RESTART | SMBR_NORESTART)) != SMBR_RESTART)
181 smb_rq_enqueue(struct smb_rq *rqp)
183 	struct smb_share *ssp = rqp->sr_share;
186 	if (ssp == NULL || rqp->sr_cred == &rqp->sr_vc->vc_iod->iod_scred) {
187 		return smb_iod_addrq(rqp);
194 			if (smb_td_intr(rqp->sr_cred->scr_td))
202 			error = smb_iod_request(rqp->sr_vc->vc_iod,
207 		error = smb_iod_addrq(rqp);
215 smb_rq_wstart(struct smb_rq *rqp)
217 	rqp->sr_wcount = mb_reserve(&rqp->sr_rq, sizeof(u_int8_t));
218 	rqp->sr_rq.mb_count = 0;
222 smb_rq_wend(struct smb_rq *rqp)
224 	if (rqp->sr_wcount == NULL) {
228 	if (rqp->sr_rq.mb_count & 1)
230 	*rqp->sr_wcount = rqp->sr_rq.mb_count / 2;
234 smb_rq_bstart(struct smb_rq *rqp)
236 	rqp->sr_bcount = mb_reserve(&rqp->sr_rq, sizeof(u_short));
237 	rqp->sr_rq.mb_count = 0;
241 smb_rq_bend(struct smb_rq *rqp)
245 	if (rqp->sr_bcount == NULL) {
249 	bcnt = rqp->sr_rq.mb_count;
252 	le16enc(rqp->sr_bcount, bcnt);
256 smb_rq_intr(struct smb_rq *rqp)
258 	if (rqp->sr_flags & SMBR_INTR)
260 	return smb_td_intr(rqp->sr_cred->scr_td);
264 smb_rq_getrequest(struct smb_rq *rqp, struct mbchain **mbpp)
266 	*mbpp = &rqp->sr_rq;
271 smb_rq_getreply(struct smb_rq *rqp, struct mdchain **mbpp)
273 	*mbpp = &rqp->sr_rp;
322 smb_rq_reply(struct smb_rq *rqp)
324 	struct mdchain *mdp = &rqp->sr_rp;
329 	error = smb_iod_waitrq(rqp);
336 	if (rqp->sr_vc->vc_hflags2 & SMB_FLAGS2_ERR_STATUS) {
337 		error = md_get_uint32le(mdp, &rqp->sr_error);
339 		error = md_get_uint8(mdp, &rqp->sr_errclass);
341 		error = md_get_uint16le(mdp, &rqp->sr_serror);
343 			rperror = smb_maperror(rqp->sr_errclass, rqp->sr_serror);
345 	error = md_get_uint8(mdp, &rqp->sr_rpflags);
346 	error = md_get_uint16le(mdp, &rqp->sr_rpflags2);
352 	error = md_get_uint16le(mdp, &rqp->sr_rptid);
353 	error = md_get_uint16le(mdp, &rqp->sr_rppid);
354 	error = md_get_uint16le(mdp, &rqp->sr_rpuid);
355 	error = md_get_uint16le(mdp, &rqp->sr_rpmid);
358 	    (rqp->sr_vc->vc_hflags2 & SMB_FLAGS2_SECURITY_SIGNATURE))
359 		error = smb_rq_verify(rqp);
362 	    rqp->sr_rpmid, rqp->sr_rppid, rqp->sr_rpuid, rqp->sr_rptid,
363 	    rqp->sr_errclass, rqp->sr_serror);
444 	struct smb_rq *rqp = t2p->t2_rq;
450 	error = smb_rq_reply(rqp);
457 		SMBRQ_SLOCK(rqp);
458 		md_next_record(&rqp->sr_rp);
459 		SMBRQ_SUNLOCK(rqp);
469 	mdp = &rqp->sr_rp;
535 		SMBRQ_SLOCK(rqp);
536 		md_next_record(&rqp->sr_rp);
537 		SMBRQ_SUNLOCK(rqp);
538 		error = smb_rq_reply(rqp);
556 	struct smb_rq *rqp;
580 	    SMB_COM_TRANSACTION : SMB_COM_TRANSACTION2, scred, &rqp);
583 	rqp->sr_flags |= SMBR_MULTIPACKET;
584 	t2p->t2_rq = rqp;
585 	rqp->sr_t2 = t2p;
586 	mbp = &rqp->sr_rq;
587 	smb_rq_wstart(rqp);
627 	smb_rq_wend(rqp);
628 	smb_rq_bstart(rqp);
650 	smb_rq_bend(rqp);	/* incredible, but thats it... */
651 	error = smb_rq_enqueue(rqp);
661 		error = smb_rq_new(rqp, t2p->t_name ?
665 		mbp = &rqp->sr_rq;
666 		smb_rq_wstart(rqp);
700 		smb_rq_wend(rqp);
701 		smb_rq_bstart(rqp);
719 		smb_rq_bend(rqp);
720 		rqp->sr_state = SMBRQ_NOTSENT;
737 	smb_iod_removerq(rqp);
739 	smb_rq_done(rqp);
741 		if (rqp->sr_flags & SMBR_RESTART)