#include "includes.h"
-/*
- destroy a socket
- */
-static int sock_destructor(void *ptr)
-{
- struct smbcli_socket *sock = ptr;
- if (sock->fd != -1) {
- close(sock->fd);
- sock->fd = -1;
- }
- return 0;
-}
-
/*
create a smbcli_socket context
*/
}
ZERO_STRUCTP(sock);
- sock->fd = -1;
+ sock->sock = NULL;
sock->port = 0;
/* 20 second default timeout */
sock->timeout = 20000;
sock->hostname = NULL;
- talloc_set_destructor(sock, sock_destructor);
-
return sock;
}
*/
BOOL smbcli_sock_connect(struct smbcli_socket *sock, struct in_addr *ip, int port)
{
- if (getenv("LIBSMB_PROG")) {
- sock->fd = sock_exec(getenv("LIBSMB_PROG"));
- return sock->fd != -1;
- }
+ NTSTATUS status;
if (port == 0) {
int i;
return False;
}
- sock->dest_ip = *ip;
- sock->port = port;
- sock->fd = open_socket_out(SOCK_STREAM,
- &sock->dest_ip,
- sock->port,
- LONG_CONNECT_TIMEOUT);
- if (sock->fd == -1) {
+ status = socket_create("ip", SOCKET_TYPE_STREAM, &sock->sock, 0);
+ if (!NT_STATUS_IS_OK(status)) {
return False;
}
+ talloc_steal(sock, sock->sock);
- set_blocking(sock->fd, False);
- set_socket_options(sock->fd, lp_socket_options());
+ status = socket_connect(sock->sock, NULL, 0, inet_ntoa(*ip), port, 0);
+ if (!NT_STATUS_IS_OK(status)) {
+ talloc_free(sock->sock);
+ sock->sock = NULL;
+ return False;
+ }
+
+ sock->dest_ip = *ip;
+ sock->port = port;
+
+ socket_set_option(sock->sock, lp_socket_options(), NULL);
return True;
}
****************************************************************************/
void smbcli_sock_dead(struct smbcli_socket *sock)
{
- if (sock->fd != -1) {
- close(sock->fd);
- sock->fd = -1;
+ if (sock->sock != NULL) {
+ talloc_free(sock->sock);
+ sock->sock = NULL;
}
}
****************************************************************************/
void smbcli_sock_set_options(struct smbcli_socket *sock, const char *options)
{
- set_socket_options(sock->fd, options);
+ socket_set_option(sock->sock, options, NULL);
}
/****************************************************************************
****************************************************************************/
ssize_t smbcli_sock_write(struct smbcli_socket *sock, const char *data, size_t len)
{
- if (sock->fd == -1) {
+ NTSTATUS status;
+ DATA_BLOB blob;
+ size_t nsent;
+
+ if (sock->sock == NULL) {
errno = EIO;
return -1;
}
- return write(sock->fd, data, len);
+ blob.data = discard_const(data);
+ blob.length = len;
+
+ status = socket_send(sock->sock, &blob, &nsent, 0);
+ if (NT_STATUS_IS_ERR(status)) {
+ return -1;
+ }
+
+ return nsent;
}
****************************************************************************/
ssize_t smbcli_sock_read(struct smbcli_socket *sock, char *data, size_t len)
{
- if (sock->fd == -1) {
+ NTSTATUS status;
+ size_t nread;
+
+ if (sock->sock == NULL) {
errno = EIO;
return -1;
}
- return read(sock->fd, data, len);
+ status = socket_recv(sock->sock, data, len, &nread, 0);
+ if (NT_STATUS_IS_ERR(status)) {
+ return -1;
+ }
+
+ return nread;
}
/****************************************************************************
char *name, *p;
BOOL ret;
+#if 0
if (getenv("LIBSMB_PROG")) {
sock->fd = sock_exec(getenv("LIBSMB_PROG"));
return sock->fd != -1;
}
+#endif
name = talloc_strdup(sock, host);
ZERO_STRUCT(transport->called);
- fde.fd = sock->fd;
+ fde.fd = socket_get_fd(sock->sock);
fde.flags = EVENT_FD_READ;
fde.handler = smbcli_transport_event_handler;
fde.private = transport;
{
smbcli_transport_process_send(transport);
smbcli_transport_process_recv(transport);
- if (transport->socket->fd == -1) {
+ if (transport->socket->sock == NULL) {
return False;
}
return True;
void smbcli_transport_send(struct smbcli_request *req)
{
/* check if the transport is dead */
- if (req->transport->socket->fd == -1) {
+ if (req->transport->socket->sock == NULL) {
req->state = SMBCLI_REQUEST_ERROR;
req->status = NT_STATUS_NET_WRITE_FAULT;
return;