print.c revision 87866 
1/*
2 * Copyright (c) 2000, Boris Popov
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *    This product includes software developed by Boris Popov.
16 * 4. Neither the name of the author nor the names of any co-contributors
17 *    may be used to endorse or promote products derived from this software
18 *    without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 *
32 * $Id: print.c,v 1.4 2001/01/28 07:35:01 bp Exp $
33 */
34#include <sys/param.h>
35#include <sys/errno.h>
36#include <sys/stat.h>
37#include <err.h>
38#include <fcntl.h>
39#include <stdio.h>
40#include <unistd.h>
41#include <strings.h>
42#include <stdlib.h>
43#include <sysexits.h>
44
45#include <cflib.h>
46
47#include <netsmb/smb_lib.h>
48#include <netsmb/smb_conn.h>
49
50#include "common.h"
51
52int
53cmd_print(int argc, char *argv[])
54{
55	struct smb_ctx sctx, *ctx = &sctx;
56	smbfh fh;
57	off_t offset;
58	char buf[8192];
59	char *filename;
60	char fnamebuf[256];
61	int error, opt, i, file, count;
62
63	if (argc < 2)
64		view_usage();
65	if (smb_ctx_init(ctx, argc, argv, SMBL_SHARE, SMBL_SHARE, SMB_ST_PRINTER) != 0)
66		exit(1);
67	if (smb_ctx_readrc(ctx) != 0)
68		exit(1);
69	if (smb_rc)
70		rc_close(smb_rc);
71	while ((opt = getopt(argc, argv, STDPARAM_OPT)) != EOF) {
72		switch(opt){
73		    case STDPARAM_ARGS:
74			error = smb_ctx_opt(ctx, opt, optarg);
75			if (error)
76				exit(1);
77			break;
78		    default:
79			view_usage();
80			/*NOTREACHED*/
81		}
82	}
83	if (optind + 1 >= argc)
84		print_usage();
85	filename = argv[optind + 1];
86
87	if (strcmp(filename, "-") == 0) {
88		file = 0;	/* stdin */
89		filename = "stdin";
90	} else {
91		file = open(filename, O_RDONLY, 0);
92		if (file < 0) {
93			smb_error("could not open file %s\n", errno, filename);
94			exit(1);
95		}
96	}
97
98	if (smb_ctx_resolve(ctx) != 0)
99		exit(1);
100	error = smb_ctx_lookup(ctx, SMBL_SHARE, SMBLK_CREATE);
101	if (error) {
102		smb_error("could not login to server %s", error, ctx->ct_ssn.ioc_srvname);
103		exit(1);
104	}
105	snprintf(fnamebuf, sizeof(fnamebuf), "%s_%s_%s", ctx->ct_ssn.ioc_user,
106	    ctx->ct_ssn.ioc_srvname, filename);
107	error = smb_smb_open_print_file(ctx, 0, 1, fnamebuf, &fh);
108	if (error) {
109		smb_error("could not open print job", error);
110		exit(1);
111	}
112	offset = 0;
113	error = 0;
114	for(;;) {
115		count = read(file, buf, sizeof(buf));
116		if (count == 0)
117			break;
118		if (count < 0) {
119			error = errno;
120			smb_error("error reading input file\n", error);
121			break;
122		}
123		i = smb_write(ctx, fh, offset, count, buf);
124		if (i < 0) {
125			error = errno;
126			smb_error("error writing spool file\n", error);
127			break;
128		}
129		if (i != count) {
130			smb_error("incomplete write to spool file\n", 0);
131			error = EIO;
132			break;
133		}
134		offset += count;
135	}
136	close(file);
137	error = smb_smb_close_print_file(ctx, fh);
138	if (error)
139		smb_error("an error while closing spool file\n", error);
140	return error ? 1 : 0;
141}
142
143
144void
145print_usage(void)
146{
147	printf(
148	"usage: smbutil print [connection optinons] //user@server/share\n"
149	);
150	exit(1);
151}
152
153