#define NO_SYSLOG
#include "includes.h"
extern BOOL in_client;
extern struct in_addr ipzero;
static void list_devices(void);
static struct cli_state *smb_connect(char *, char *, char *, char *, char *);
static int smb_print(struct cli_state *, char *, FILE *);
int
main(int argc,
char *argv[])
{
int i;
int copies;
char uri[1024],
*sep,
*username,
*password,
*workgroup,
*server,
*printer;
FILE *fp;
int status=0;
struct cli_state *cli;
if (argc > 2 && strncmp(argv[0],"smb://", 6) && !strncmp(argv[1],"smb://", 6)) {
argv++;
argc--;
}
if (argc == 1)
{
list_devices();
return (0);
}
if (argc < 6 || argc > 7)
{
fprintf(stderr, "Usage: %s [DEVICE_URI] job-id user title copies options [file]\n",
argv[0]);
fputs(" The DEVICE_URI environment variable can also contain the\n", stderr);
fputs(" destination printer:\n", stderr);
fputs("\n", stderr);
fputs(" smb://[username:password@][workgroup/]server/printer\n", stderr);
return (1);
}
if (argc == 6)
{
fp = stdin;
copies = 1;
}
else if ((fp = fopen(argv[6], "rb")) == NULL)
{
perror("ERROR: Unable to open print file");
return (1);
}
else
copies = atoi(argv[4]);
if (strncmp(argv[0], "smb://", 6) == 0)
strncpy(uri, argv[0], sizeof(uri) - 1);
else if (getenv("DEVICE_URI") != NULL)
strncpy(uri, getenv("DEVICE_URI"), sizeof(uri) - 1);
else
{
fputs("ERROR: No device URI found in argv[0] or DEVICE_URI environment variable!\n", stderr);
return (1);
}
uri[sizeof(uri) - 1] = '\0';
if ((sep = strrchr(uri, '@')) != NULL)
{
username = uri + 6;
*sep++ = '\0';
server = sep;
if ((password = strchr(username, ':')) != NULL)
*password++ = '\0';
else
password = "";
}
else
{
username = "";
password = "";
server = uri + 6;
}
if ((sep = strchr(server, '/')) == NULL)
{
fputs("ERROR: Bad URI - need printer name!\n", stderr);
return (1);
}
*sep++ = '\0';
printer = sep;
if ((sep = strchr(printer, '/')) != NULL)
{
*sep++ = '\0';
workgroup = server;
server = printer;
printer = sep;
}
else
workgroup = NULL;
setup_logging("smbspool", True);
TimeInit();
charset_initialise();
in_client = True;
if (!lp_load(CONFIGFILE, True, False, False))
{
fprintf(stderr, "ERROR: Can't load %s - run testparm to debug it\n", CONFIGFILE);
return (1);
}
if (workgroup == NULL)
workgroup = lp_workgroup();
codepage_initialise(lp_client_code_page());
load_interfaces();
do
{
if ((cli = smb_connect(workgroup, server, printer, username, password)) == NULL)
{
if (getenv("CLASS") == NULL)
{
perror("ERROR: Unable to connect to SAMBA host, will retry in 60 seconds...");
sleep (60);
}
else
{
perror("ERROR: Unable to connect to SAMBA host, trying next printer...");
return (1);
}
}
}
while (cli == NULL);
if (argc < 7)
CatchSignal(SIGTERM, SIG_IGN);
for (i = 0; i < copies; i ++)
if ((status = smb_print(cli, argv[3] , fp)) != 0)
break;
cli_shutdown(cli);
return (status);
}
static void
list_devices(void)
{
puts("network smb \"Unknown\" \"Windows Printer via SAMBA\"");
}
static struct cli_state *
smb_connect(char *workgroup,
char *server,
char *share,
char *username,
char *password)
{
struct cli_state *c;
struct nmb_name called,
calling;
struct in_addr ip;
pstring myname;
get_myname(myname);
ip = ipzero;
make_nmb_name(&calling, myname, 0x0);
make_nmb_name(&called, server, 0x20);
if ((c = cli_initialise(NULL)) == NULL)
{
fputs("ERROR: cli_initialise() failed...\n", stderr);
return (NULL);
}
if (!cli_set_port(c, SMB_PORT))
{
fputs("ERROR: cli_set_port() failed...\n", stderr);
return (NULL);
}
if (!cli_connect(c, server, &ip))
{
fputs("ERROR: cli_connect() failed...\n", stderr);
return (NULL);
}
if (!cli_session_request(c, &calling, &called))
{
fputs("ERROR: cli_session_request() failed...\n", stderr);
return (NULL);
}
if (!cli_negprot(c))
{
fputs("ERROR: SMB protocol negotiation failed\n", stderr);
cli_shutdown(c);
return (NULL);
}
if (!cli_session_setup(c, username,
password, strlen(password),
password, strlen(password),
workgroup))
{
fprintf(stderr, "ERROR: SMB session setup failed: %s\n", cli_errstr(c));
return (NULL);
}
if (!cli_send_tconX(c, share, "?????",
password, strlen(password)+1))
{
fprintf(stderr, "ERROR: SMB tree connect failed: %s\n", cli_errstr(c));
cli_shutdown(c);
return (NULL);
}
return (c);
}
static int
smb_print(struct cli_state *cli,
char *title,
FILE *fp)
{
int fnum;
int nbytes,
tbytes;
char buffer[8192],
*ptr;
for (ptr = title; *ptr; ptr ++)
if (!isalnum((int)*ptr) && !isspace((int)*ptr))
*ptr = '_';
if ((fnum = cli_open(cli, title, O_RDWR | O_CREAT | O_TRUNC, DENY_NONE)) == -1)
{
fprintf(stderr, "ERROR: %s opening remote file %s\n",
cli_errstr(cli), title);
return (1);
}
if (fp != stdin)
rewind(fp);
tbytes = 0;
while ((nbytes = fread(buffer, 1, sizeof(buffer), fp)) > 0)
{
if (cli_write(cli, fnum, 0, buffer, tbytes, nbytes) != nbytes)
{
fprintf(stderr, "ERROR: Error writing file: %s\n", cli_errstr(cli));
break;
}
tbytes += nbytes;
}
if (!cli_close(cli, fnum))
{
fprintf(stderr, "ERROR: %s closing remote file %s\n",
cli_errstr(cli), title);
return (1);
}
else
return (0);
}