libatalk/dsi/dsi_read.c

   1 /*
   2  * $Id: dsi_read.c,v 1.5 2009-10-20 04:53:19 didg Exp $
   3  *
   4  * Copyright (c) 1997 Adrian Sun (asun@zoology.washington.edu)
   5  * All rights reserved. See COPYRIGHT.
   6  */
   7
   8 #ifdef HAVE_CONFIG_H
   9 #include "config.h"
  10 #endif /* HAVE_CONFIG_H */
  11
  12 #include <stdio.h>
  13 #include <string.h>
  14 #ifdef HAVE_UNISTD_H
  15 #include <unistd.h>
  16 #endif /* HAVE_UNISTD_H */
  17 #include <signal.h>
  18 #include <sys/types.h>
  19 #include <sys/time.h>
  20 #ifdef HAVE_SYS_FILIO_H
  21 #include <sys/filio.h>
  22 #endif
  23
  24 #include <atalk/dsi.h>
  25 #include <sys/ioctl.h>
  26
  27 #ifndef min
  28 #define min(a,b)   ((a) < (b) ? (a) : (b))
  29 #endif /* ! min */
  30
  31 /* streaming i/o for afp_read. this is all from the perspective of the
  32  * client. it basically does the reverse of dsi_write. on first entry,
  33  * it will send off the header plus whatever is in its command
  34  * buffer. it returns the amount of stuff still to be read
  35  * (constrained by the buffer size). */
  36 ssize_t dsi_readinit(DSI *dsi, void *buf, const size_t buflen,
  37                     const size_t size, const int err)
  38 {
  39
  40   dsi->noreply = 1; /* we will handle our own replies */
  41   dsi->header.dsi_flags = DSIFL_REPLY;
  42   /*dsi->header.dsi_command = DSIFUNC_CMD;*/
  43   dsi->header.dsi_len = htonl(size);
  44   dsi->header.dsi_code = htonl(err);
  45
  46   sigprocmask(SIG_BLOCK, &dsi->sigblockset, &dsi->oldset);
  47   dsi->sigblocked = 1;
  48   if (dsi_stream_send(dsi, buf, buflen)) {
  49     dsi->datasize = size - buflen;
  50     return min(dsi->datasize, buflen);
  51   }
  52
  53   return -1; /* error */
  54 }
  55
  56 void dsi_readdone(DSI *dsi)
  57 {
  58   sigprocmask(SIG_SETMASK, &dsi->oldset, NULL);
  59   dsi->sigblocked = 0;
  60 }
  61
  62 /* */
  63 int dsi_block(DSI *dsi, const int mode)
  64 {
  65 #if 0
  66     dsi->noblocking = mode;
  67     return 0;
  68 #else
  69     int adr = mode;
  70     int ret;
  71
  72     ret = ioctl(dsi->socket, FIONBIO, &adr);
  73     if (!ret) {
  74         dsi->noblocking = mode;
  75     }
  76     return ret;
  77 #endif
  78 }
  79
  80 /* send off the data */
  81 ssize_t dsi_read(DSI *dsi, void *buf, const size_t buflen)
  82 {
  83   size_t len;
  84   int delay = (dsi->datasize != buflen)?1:0;
  85
  86   len  = dsi_stream_write(dsi, buf, buflen, delay);
  87
  88   if (len == buflen) {
  89     dsi->datasize -= len;
  90     return min(dsi->datasize, buflen);
  91   }
  92
  93   return -1;
  94 }