size_t create_rpc_blob(RPC_DATA_BLOB *str, size_t len)
{
str->buffer = TALLOC_ZERO(get_talloc_ctx(), len);
if (str->buffer == NULL)
smb_panic("create_rpc_blob: talloc fail\n");
return len;
}
static void do_smb_resolve(struct mdns_smbsrv_result *browsesrv)
{
DNSServiceRef mdns_conn_sdref = NULL;
int mdnsfd;
int fdsetsz;
int ret;
fd_set *fdset = NULL;
struct timeval tv;
DNSServiceErrorType err;
TALLOC_CTX * ctx = talloc_tos();
err = DNSServiceResolve(&mdns_conn_sdref, 0 /* flags */,
browsesrv->ifIndex,
browsesrv->serviceName, browsesrv->regType, browsesrv->domain,
do_smb_resolve_reply, NULL);
if (err != kDNSServiceErr_NoError) {
return;
}
mdnsfd = DNSServiceRefSockFD(mdns_conn_sdref);
for (;;) {
if (fdset != NULL) {
TALLOC_FREE(fdset);
}
if (mdnsfd < 0 || mdnsfd >= FD_SETSIZE) {
errno = EBADF;
break;
}
fdsetsz = howmany(mdnsfd + 1, NFDBITS) * sizeof(fd_mask);
fdset = TALLOC_ZERO(ctx, fdsetsz);
FD_SET(mdnsfd, fdset);
tv.tv_sec = 1;
tv.tv_usec = 0;
/* Wait until response received from mDNS daemon */
ret = sys_select(mdnsfd + 1, fdset, NULL, NULL, &tv);
if (ret <= 0 && errno != EINTR) {
break;
}
if (FD_ISSET(mdnsfd, fdset)) {
/* Invoke callback function */
DNSServiceProcessResult(mdns_conn_sdref);
break;
}
}
TALLOC_FREE(fdset);
DNSServiceRefDeallocate(mdns_conn_sdref);
}
void init_regval_buffer(REGVAL_BUFFER *str, const uint8 *buf, size_t len)
{
ZERO_STRUCTP(str);
/* max buffer size (allocated size) */
str->buf_max_len = len;
str->offset = 0;
str->buf_len = buf != NULL ? len : 0;
if (buf != NULL) {
SMB_ASSERT(str->buf_max_len >= str->buf_len);
str->buffer = TALLOC_ZERO(get_talloc_ctx(), str->buf_max_len);
if (str->buffer == NULL)
smb_panic("init_regval_buffer: talloc fail\n");
memcpy(str->buffer, buf, str->buf_len);
}
}
void *vfs_add_fsp_extension_notype(vfs_handle_struct *handle, files_struct *fsp, size_t ext_size)
{
struct vfs_fsp_data *ext;
void * ext_data;
/* Prevent VFS modules adding multiple extensions. */
if ((ext_data = vfs_fetch_fsp_extension(handle, fsp))) {
return ext_data;
}
ext = (struct vfs_fsp_data *)TALLOC_ZERO(
handle->conn->mem_ctx, sizeof(struct vfs_fsp_data) + ext_size);
if (ext == NULL) {
return NULL;
}
ext->owner = handle;
ext->next = fsp->vfs_extension;
fsp->vfs_extension = ext;
return EXT_DATA_AREA(ext);
}
void init_string2(STRING2 *str, const char *buf, size_t max_len, size_t str_len)
{
/* set up string lengths. */
SMB_ASSERT(max_len >= str_len);
/* Ensure buf is valid if str_len was set. Coverity check. */
if (str_len && !buf) {
return;
}
str->str_max_len = max_len;
str->offset = 0;
str->str_str_len = str_len;
/* store the string */
if(str_len != 0) {
str->buffer = TALLOC_ZERO(get_talloc_ctx(), str->str_max_len);
if (str->buffer == NULL)
smb_panic("init_string2: malloc fail\n");
memcpy(str->buffer, buf, str_len);
}
}
int do_smb_browse(void)
{
int mdnsfd;
int fdsetsz;
int ret;
fd_set *fdset = NULL;
struct mdns_browse_state bstate;
struct mdns_smbsrv_result *resptr;
struct timeval tv;
DNSServiceRef mdns_conn_sdref = NULL;
DNSServiceErrorType err;
TALLOC_CTX * ctx = talloc_stackframe();
ZERO_STRUCT(bstate);
err = DNSServiceBrowse(&mdns_conn_sdref, 0, 0, "_smb._tcp", "",
do_smb_browse_reply, &bstate);
if (err != kDNSServiceErr_NoError) {
d_printf("Error connecting to the Multicast DNS daemon\n");
TALLOC_FREE(ctx);
return 1;
}
mdnsfd = DNSServiceRefSockFD(mdns_conn_sdref);
for (;;) {
if (fdset != NULL) {
TALLOC_FREE(fdset);
}
fdsetsz = howmany(mdnsfd + 1, NFDBITS) * sizeof(fd_mask);
fdset = TALLOC_ZERO(ctx, fdsetsz);
FD_SET(mdnsfd, fdset);
tv.tv_sec = 1;
tv.tv_usec = 0;
/* Wait until response received from mDNS daemon */
ret = sys_select(mdnsfd + 1, fdset, NULL, NULL, &tv);
if (ret <= 0 && errno != EINTR) {
break;
}
if (FD_ISSET(mdnsfd, fdset)) {
/* Invoke callback function */
if (DNSServiceProcessResult(mdns_conn_sdref)) {
break;
}
if (bstate.browseDone) {
break;
}
}
}
DNSServiceRefDeallocate(mdns_conn_sdref);
if (bstate.listhead != NULL) {
resptr = bstate.listhead;
while (resptr != NULL) {
struct mdns_smbsrv_result *oldresptr;
oldresptr = resptr;
/* Resolve smb service instance */
do_smb_resolve(resptr);
resptr = resptr->nextResult;
}
}
TALLOC_FREE(ctx);
return 0;
}
