void cli_shutdown(struct cli_state *cli)
{
BOOL allocated = cli->allocated;
cli_close_connection(cli);
ZERO_STRUCTP(cli);
if (allocated)
SAFE_FREE(cli);
}
BOOL attempt_netbios_session_request(struct cli_state *cli, const char *srchost, const char *desthost,
struct in_addr *pdest_ip)
{
struct nmb_name calling, called;
make_nmb_name(&calling, srchost, 0x0);
/*
* If the called name is an IP address
* then use *SMBSERVER immediately.
*/
if(is_ipaddress(desthost))
make_nmb_name(&called, "*SMBSERVER", 0x20);
else
make_nmb_name(&called, desthost, 0x20);
if (!cli_session_request(cli, &calling, &called)) {
struct nmb_name smbservername;
make_nmb_name(&smbservername , "*SMBSERVER", 0x20);
/*
* If the name wasn't *SMBSERVER then
* try with *SMBSERVER if the first name fails.
*/
if (nmb_name_equal(&called, &smbservername)) {
/*
* The name used was *SMBSERVER, don't bother with another name.
*/
DEBUG(0,("attempt_netbios_session_request: %s rejected the session for name *SMBSERVER with error %s.\n",
desthost, cli_errstr(cli) ));
return False;
}
/*
* We need to close the connection here but can't call cli_shutdown as
* will free an allocated cli struct. cli_close_connection was invented
* for this purpose. JRA. Based on work by "Kim R. Pedersen" <*****@*****.**>.
*/
cli_close_connection(cli);
if (!cli_initialise(cli) || !cli_connect(cli, desthost, pdest_ip) ||
!cli_session_request(cli, &calling, &smbservername)) {
DEBUG(0,("attempt_netbios_session_request: %s rejected the session for name *SMBSERVER with error %s\n",
desthost, cli_errstr(cli) ));
return False;
}
}
return True;
}
static void cli_alarm(UNUSED int sig)
{
if (clif_conn && _clif_command(clif_conn, "P", SHOW_NO_OUTPUT)) {
printf("Connection to lldpad lost - trying to reconnect\n");
cli_close_connection();
}
if (!clif_conn) {
clif_conn = clif_open();
if (clif_conn) {
printf("Connection to lldpad re-established\n");
if (clif_attach(clif_conn, NULL) == 0)
cli_attached = 1;
else
printf(
"Warning: Failed to attach to lldpad.\n");
}
}
if (clif_conn)
cli_recv_pending(clif_conn, 1);
alarm(1);
}
struct cli_state *cli_initialise(struct cli_state *cli)
{
BOOL alloced_cli = False;
/* Check the effective uid - make sure we are not setuid */
if (is_setuid_root()) {
DEBUG(0,("libsmb based programs must *NOT* be setuid root.\n"));
return NULL;
}
if (!cli) {
cli = (struct cli_state *)malloc(sizeof(*cli));
if (!cli)
return NULL;
ZERO_STRUCTP(cli);
alloced_cli = True;
}
if (cli->initialised)
cli_close_connection(cli);
ZERO_STRUCTP(cli);
cli->port = 0;
cli->fd = -1;
cli->cnum = -1;
cli->pid = (uint16)sys_getpid();
cli->mid = 1;
cli->vuid = UID_FIELD_INVALID;
cli->protocol = PROTOCOL_NT1;
cli->timeout = 20000; /* Timeout is in milliseconds. */
cli->bufsize = CLI_BUFFER_SIZE+4;
cli->max_xmit = cli->bufsize;
cli->outbuf = (char *)malloc(cli->bufsize);
cli->inbuf = (char *)malloc(cli->bufsize);
cli->oplock_handler = cli_oplock_ack;
/* Set the CLI_FORCE_DOSERR environment variable to test
client routines using DOS errors instead of STATUS32
ones. This intended only as a temporary hack. */
if (getenv("CLI_FORCE_DOSERR"))
cli->force_dos_errors = True;
if (!cli->outbuf || !cli->inbuf)
goto error;
if ((cli->mem_ctx = talloc_init()) == NULL)
goto error;
memset(cli->outbuf, 0, cli->bufsize);
memset(cli->inbuf, 0, cli->bufsize);
cli->nt_pipe_fnum = 0;
cli->initialised = 1;
cli->allocated = alloced_cli;
return cli;
/* Clean up after malloc() error */
error:
SAFE_FREE(cli->inbuf);
SAFE_FREE(cli->outbuf);
if (alloced_cli)
SAFE_FREE(cli);
return NULL;
}
int main(int argc, char *argv[])
{
int interactive;
int raw = 0;
int warning_displayed = 0;
int c;
int ret = 0;
for (;;) {
c = getopt(argc, argv, "hvrR");
if (c < 0)
break;
switch (c) {
case 'h':
usage();
return 0;
case 'r':
if (raw) {
usage();
return -1;
}
raw = SHOW_RAW;
break;
case 'R':
if (raw) {
usage();
return -1;
}
raw = (SHOW_RAW | SHOW_RAW_ONLY);
break;
case 'v':
printf("%s\n", cli_version);
return 0;
default:
usage();
return -1;
}
}
show_raw = raw;
interactive = argc == optind;
if (interactive) {
printf("%s\n\n%s\n\n", cli_version,
cli_license);
}
for (;;) {
clif_conn = clif_open();
if (clif_conn) {
if (warning_displayed)
printf("Connection established.\n");
break;
}
if (!interactive) {
perror("Failed to connect to lldpad - clif_open");
return -1;
}
if (!warning_displayed) {
printf("Could not connect to lldpad - re-trying\n");
warning_displayed = 1;
}
sleep(1);
continue;
}
signal(SIGINT, cli_terminate);
signal(SIGTERM, cli_terminate);
signal(SIGALRM, cli_alarm);
if (interactive) {
if (clif_attach(clif_conn, NULL) == 0)
cli_attached = 1;
else
printf("Warning: Failed to attach to lldpad.\n");
cli_interactive(raw);
} else
ret = request(clif_conn, argc - optind, &argv[optind], raw);
cli_close_connection();
return ret;
}
static void cli_terminate(UNUSED int sig)
{
cli_close_connection();
exit(0);
}
