contrib/smbfs/smbutil/print.c

   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
  35 #include <sys/cdefs.h>
  36 __FBSDID("$FreeBSD$");
  37
  38 #include <sys/param.h>
  39 #include <sys/errno.h>
  40 #include <sys/stat.h>
  41 #include <err.h>
  42 #include <fcntl.h>
  43 #include <stdio.h>
  44 #include <unistd.h>
  45 #include <strings.h>
  46 #include <stdlib.h>
  47 #include <sysexits.h>
  48
  49 #include <cflib.h>
  50
  51 #include <netsmb/smb_lib.h>
  52 #include <netsmb/smb_conn.h>
  53
  54 #include "common.h"
  55
  56 int
  57 cmd_print(int argc, char *argv[])
  58 {
  59         struct smb_ctx sctx, *ctx = &sctx;
  60         smbfh fh;
  61         off_t offset;
  62         char buf[8192];
  63         char *filename;
  64         char fnamebuf[256];
  65         int error, opt, i, file, count;
  66
  67         if (argc < 2)
  68                 view_usage();
  69         if (smb_ctx_init(ctx, argc, argv, SMBL_SHARE, SMBL_SHARE, SMB_ST_PRINTER) != 0)
  70                 exit(1);
  71         if (smb_ctx_readrc(ctx) != 0)
  72                 exit(1);
  73         if (smb_rc)
  74                 rc_close(smb_rc);
  75         while ((opt = getopt(argc, argv, STDPARAM_OPT)) != EOF) {
  76                 switch(opt){
  77                     case STDPARAM_ARGS:
  78                         error = smb_ctx_opt(ctx, opt, optarg);
  79                         if (error)
  80                                 exit(1);
  81                         break;
  82                     default:
  83                         view_usage();
  84                         /*NOTREACHED*/
  85                 }
  86         }
  87         if (optind + 1 >= argc)
  88                 print_usage();
  89         filename = argv[optind + 1];
  90
  91         if (strcmp(filename, "-") == 0) {
  92                 file = 0;       /* stdin */
  93                 filename = "stdin";
  94         } else {
  95                 file = open(filename, O_RDONLY, 0);
  96                 if (file < 0) {
  97                         smb_error("could not open file %s\n", errno, filename);
  98                         exit(1);
  99                 }
 100         }
 101
 102         if (smb_ctx_resolve(ctx) != 0)
 103                 exit(1);
 104         error = smb_ctx_lookup(ctx, SMBL_SHARE, SMBLK_CREATE);
 105         if (error) {
 106                 smb_error("could not login to server %s", error, ctx->ct_ssn.ioc_srvname);
 107                 exit(1);
 108         }
 109         snprintf(fnamebuf, sizeof(fnamebuf), "%s_%s_%s", ctx->ct_ssn.ioc_user,
 110             ctx->ct_ssn.ioc_srvname, filename);
 111         error = smb_smb_open_print_file(ctx, 0, 1, fnamebuf, &fh);
 112         if (error) {
 113                 smb_error("could not open print job", error);
 114                 exit(1);
 115         }
 116         offset = 0;
 117         error = 0;
 118         for(;;) {
 119                 count = read(file, buf, sizeof(buf));
 120                 if (count == 0)
 121                         break;
 122                 if (count < 0) {
 123                         error = errno;
 124                         smb_error("error reading input file\n", error);
 125                         break;
 126                 }
 127                 i = smb_write(ctx, fh, offset, count, buf);
 128                 if (i < 0) {
 129                         error = errno;
 130                         smb_error("error writing spool file\n", error);
 131                         break;
 132                 }
 133                 if (i != count) {
 134                         smb_error("incomplete write to spool file\n", 0);
 135                         error = EIO;
 136                         break;
 137                 }
 138                 offset += count;
 139         }
 140         close(file);
 141         error = smb_smb_close_print_file(ctx, fh);
 142         if (error)
 143                 smb_error("an error while closing spool file\n", error);
 144         return error ? 1 : 0;
 145 }
 146
 147
 148 void
 149 print_usage(void)
 150 {
 151         printf(
 152         "usage: smbutil print [connection options] //user@server/share\n"
 153         );
 154         exit(1);
 155 }
 156