/*
* smb_encode_stream_info
*
* This function encodes the streams information.
* The following rules about how have been derived from observed NT
* behaviour.
*
* If the target is a file:
* 1. If there are no named streams, the response should still contain
* an entry for the unnamed stream.
* 2. If there are named streams, the response should contain an entry
* for the unnamed stream followed by the entries for the named
* streams.
*
* If the target is a directory:
* 1. If there are no streams, the response is complete. Directories
* do not report the unnamed stream.
* 2. If there are streams, the response should contain entries for
* those streams but there should not be an entry for the unnamed
* stream.
*
* Note that the stream name lengths exclude the null terminator but
* the field lengths (i.e. next offset calculations) need to include
* the null terminator and be padded to a multiple of 8 bytes. The
* last entry does not seem to need any padding.
*
* If an error is encountered when trying to read the stream entries
* (smb_odir_read_streaminfo) it is treated as if there are no [more]
* entries. The entries that have been read so far are returned and
* no error is reported.
*
* If the response buffer is not large enough to return all of the
* named stream entries, the entries that do fit are returned and
* a warning code is set (NT_STATUS_BUFFER_OVERFLOW). The next_offset
* value in the last returned entry must be 0.
*/
static void
smb_encode_stream_info(smb_request_t *sr, smb_xa_t *xa, smb_queryinfo_t *qinfo)
{
char *stream_name;
uint32_t next_offset;
uint32_t stream_nlen;
uint32_t pad;
u_offset_t datasz, allocsz;
boolean_t is_dir;
smb_streaminfo_t *sinfo, *sinfo_next;
int rc = 0;
boolean_t done = B_FALSE;
boolean_t eos = B_FALSE;
uint16_t odid;
smb_odir_t *od = NULL;
smb_node_t *fnode = qinfo->qi_node;
smb_attr_t *attr = &qinfo->qi_attr;
ASSERT(fnode);
if (SMB_IS_STREAM(fnode)) {
fnode = fnode->n_unode;
ASSERT(fnode);
}
ASSERT(fnode->n_magic == SMB_NODE_MAGIC);
ASSERT(fnode->n_state != SMB_NODE_STATE_DESTROYING);
sinfo = kmem_alloc(sizeof (smb_streaminfo_t), KM_SLEEP);
sinfo_next = kmem_alloc(sizeof (smb_streaminfo_t), KM_SLEEP);
is_dir = ((attr->sa_dosattr & FILE_ATTRIBUTE_DIRECTORY) != 0);
datasz = attr->sa_vattr.va_size;
allocsz = attr->sa_allocsz;
odid = smb_odir_openat(sr, fnode);
if (odid != 0)
od = smb_tree_lookup_odir(sr->tid_tree, odid);
if (od != NULL)
rc = smb_odir_read_streaminfo(sr, od, sinfo, &eos);
if ((od == NULL) || (rc != 0) || (eos))
done = B_TRUE;
/* If not a directory, encode an entry for the unnamed stream. */
if (!is_dir) {
stream_name = "::$DATA";
stream_nlen = smb_ascii_or_unicode_strlen(sr, stream_name);
next_offset = SMB_STREAM_ENCODE_FIXED_SZ + stream_nlen +
smb_ascii_or_unicode_null_len(sr);
/* Can unnamed stream fit in response buffer? */
if (MBC_ROOM_FOR(&xa->rep_data_mb, next_offset) == 0) {
done = B_TRUE;
smbsr_warn(sr, NT_STATUS_BUFFER_OVERFLOW,
ERRDOS, ERROR_MORE_DATA);
} else {
/* Can first named stream fit in rsp buffer? */
if (!done && !smb_stream_fits(sr, xa, sinfo->si_name,
next_offset)) {
done = B_TRUE;
smbsr_warn(sr, NT_STATUS_BUFFER_OVERFLOW,
ERRDOS, ERROR_MORE_DATA);
}
if (done)
next_offset = 0;
(void) smb_mbc_encodef(&xa->rep_data_mb, "%llqqu", sr,
next_offset, stream_nlen, datasz, allocsz,
stream_name);
}
}
/*
* If there is no next entry, or there is not enough space in
* the response buffer for the next entry, the next_offset and
* padding are 0.
*/
while (!done) {
stream_nlen = smb_ascii_or_unicode_strlen(sr, sinfo->si_name);
sinfo_next->si_name[0] = 0;
rc = smb_odir_read_streaminfo(sr, od, sinfo_next, &eos);
if ((rc != 0) || (eos)) {
done = B_TRUE;
} else {
next_offset = SMB_STREAM_ENCODE_FIXED_SZ +
stream_nlen +
smb_ascii_or_unicode_null_len(sr);
pad = smb_pad_align(next_offset, 8);
next_offset += pad;
/* Can next named stream fit in response buffer? */
if (!smb_stream_fits(sr, xa, sinfo_next->si_name,
next_offset)) {
done = B_TRUE;
smbsr_warn(sr, NT_STATUS_BUFFER_OVERFLOW,
ERRDOS, ERROR_MORE_DATA);
}
}
if (done) {
next_offset = 0;
pad = 0;
}
rc = smb_mbc_encodef(&xa->rep_data_mb, "%llqqu#.",
sr, next_offset, stream_nlen,
sinfo->si_size, sinfo->si_alloc_size,
sinfo->si_name, pad);
(void) memcpy(sinfo, sinfo_next, sizeof (smb_streaminfo_t));
}
kmem_free(sinfo, sizeof (smb_streaminfo_t));
kmem_free(sinfo_next, sizeof (smb_streaminfo_t));
if (od) {
smb_odir_close(od);
smb_odir_release(od);
}
}
/*
* smb_encode_stream_info
*
* This function encodes the streams information.
* The following rules about how have been derived from observed NT
* behaviour.
*
* If the target is a file:
* 1. If there are no named streams, the response should still contain
* an entry for the unnamed stream.
* 2. If there are named streams, the response should contain an entry
* for the unnamed stream followed by the entries for the named
* streams.
*
* If the target is a directory:
* 1. If there are no streams, the response is complete. Directories
* do not report the unnamed stream.
* 2. If there are streams, the response should contain entries for
* those streams but there should not be an entry for the unnamed
* stream.
*
* Note that the stream name lengths exclude the null terminator but
* the field lengths (i.e. next offset calculations) need to include
* the null terminator and be padded to a multiple of 8 bytes. The
* last entry does not seem to need any padding.
*
* If an error is encountered when trying to read the stream entries
* (smb_odir_read_streaminfo) it is treated as if there are no [more]
* entries. The entries that have been read so far are returned and
* no error is reported.
*
* Offset calculation:
* 2 dwords + 2 quadwords => 4 + 4 + 8 + 8 => 24
*/
static void
smb_encode_stream_info(smb_request_t *sr, smb_xa_t *xa, smb_queryinfo_t *qinfo)
{
char *stream_name;
uint32_t next_offset;
uint32_t stream_nlen;
uint32_t pad;
u_offset_t datasz, allocsz;
boolean_t is_dir;
smb_streaminfo_t *sinfo, *sinfo_next;
int rc = 0;
boolean_t done = B_FALSE;
boolean_t eos = B_FALSE;
uint16_t odid;
smb_odir_t *od = NULL;
smb_node_t *fnode = qinfo->qi_node;
smb_attr_t *attr = &qinfo->qi_attr;
ASSERT(fnode);
if (SMB_IS_STREAM(fnode)) {
fnode = fnode->n_unode;
ASSERT(fnode);
}
ASSERT(fnode->n_magic == SMB_NODE_MAGIC);
ASSERT(fnode->n_state != SMB_NODE_STATE_DESTROYING);
sinfo = kmem_alloc(sizeof (smb_streaminfo_t), KM_SLEEP);
sinfo_next = kmem_alloc(sizeof (smb_streaminfo_t), KM_SLEEP);
is_dir = (attr->sa_vattr.va_type == VDIR);
datasz = attr->sa_vattr.va_size;
allocsz = attr->sa_allocsz;
odid = smb_odir_openat(sr, fnode);
if (odid != 0)
od = smb_tree_lookup_odir(sr->tid_tree, odid);
if (od != NULL)
rc = smb_odir_read_streaminfo(sr, od, sinfo, &eos);
if ((od == NULL) || (rc != 0) || (eos))
done = B_TRUE;
/* If not a directory, encode an entry for the unnamed stream. */
if (!is_dir) {
stream_name = "::$DATA";
stream_nlen = smb_ascii_or_unicode_strlen(sr, stream_name);
if (done)
next_offset = 0;
else
next_offset = 24 + stream_nlen +
smb_ascii_or_unicode_null_len(sr);
(void) smb_mbc_encodef(&xa->rep_data_mb, "%llqqu", sr,
next_offset, stream_nlen, datasz, allocsz, stream_name);
}
/*
* Since last packet does not have a pad we need to check
* for the next stream before we encode the current one
*/
while (!done) {
stream_nlen = smb_ascii_or_unicode_strlen(sr, sinfo->si_name);
sinfo_next->si_name[0] = 0;
rc = smb_odir_read_streaminfo(sr, od, sinfo_next, &eos);
if ((rc != 0) || (eos)) {
done = B_TRUE;
next_offset = 0;
pad = 0;
} else {
next_offset = 24 + stream_nlen +
smb_ascii_or_unicode_null_len(sr);
pad = smb_pad_align(next_offset, 8);
next_offset += pad;
}
(void) smb_mbc_encodef(&xa->rep_data_mb, "%llqqu#.",
sr, next_offset, stream_nlen,
sinfo->si_size, sinfo->si_alloc_size,
sinfo->si_name, pad);
(void) memcpy(sinfo, sinfo_next, sizeof (smb_streaminfo_t));
}
kmem_free(sinfo, sizeof (smb_streaminfo_t));
kmem_free(sinfo_next, sizeof (smb_streaminfo_t));
if (od) {
smb_odir_close(od);
smb_odir_release(od);
}
}