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); }