usr/src/cmd/man/src/util/nsgmls.src/lib/OutputByteStream.cxx

   1 // Copyright (c) 1997 James Clark
   2 // See the file COPYING for copying permission.
   3 #pragma ident   "%Z%%M% %I%     %E% SMI"
   4
   5 #include "splib.h"
   6 #include "OutputByteStream.h"
   7
   8 #include <sys/types.h>
   9
  10 #ifdef SP_INCLUDE_IO_H
  11 #include <io.h>         // for open, fstat, lseek, read prototypes
  12 #endif
  13
  14 #ifdef SP_INCLUDE_UNISTD_H
  15 #include <unistd.h>
  16 #endif
  17
  18 #ifdef SP_INCLUDE_OSFCN_H
  19 #include <osfcn.h>
  20 #endif
  21
  22 #include <fcntl.h>
  23 #include <sys/stat.h>
  24 #include <errno.h>
  25 #include <stdio.h>
  26
  27 #ifdef SP_WIDE_SYSTEM
  28 #define STRICT
  29 #include <windows.h>
  30 #endif
  31
  32 #ifndef O_CREAT
  33 #ifdef _O_CREAT
  34 #define O_CREAT _O_CREAT
  35 #endif
  36 #endif
  37
  38 #ifndef O_WRONLY
  39 #ifdef _O_WRONLY
  40 #define O_WRONLY _O_WRONLY
  41 #endif
  42 #endif
  43
  44 #ifndef O_TRUNC
  45 #ifdef _O_TRUNC
  46 #define O_TRUNC _O_TRUNC
  47 #endif
  48 #endif
  49
  50 #ifndef O_BINARY
  51 #ifdef _O_BINARY
  52 #define O_BINARY _O_BINARY
  53 #else
  54 #define O_BINARY 0
  55 #endif
  56 #endif
  57
  58 #ifndef S_IRUSR
  59 #if defined(S_IREAD)
  60 #define S_IRUSR S_IREAD
  61 #elif defined(_S_IREAD)
  62 #define S_IRUSR _S_IREAD
  63 #else
  64 #define S_IRUSR 0400
  65 #endif
  66 #endif
  67
  68 #ifndef S_IWUSR
  69 #if defined(S_IWRITE)
  70 #define S_IWUSR S_IWRITE
  71 #elif defined(_S_IWRITE)
  72 #define S_IWUSR _S_IWRITE
  73 #else
  74 #define S_IWUSR 0200
  75 #endif
  76 #endif
  77
  78 #ifndef S_IRGRP
  79 #if defined(S_IREAD)
  80 #define S_IRGRP S_IREAD
  81 #elif defined(_S_IREAD)
  82 #define S_IRGRP _S_IREAD
  83 #else
  84 #define S_IRGRP 0040
  85 #endif
  86 #endif
  87
  88 #ifndef S_IWGRP
  89 #if defined(S_IWRITE)
  90 #define S_IWGRP S_IWRITE
  91 #elif defined(_S_IWRITE)
  92 #define S_IWGRP _S_IWRITE
  93 #else
  94 #define S_IWGRP 0020
  95 #endif
  96 #endif
  97
  98 #ifndef S_IROTH
  99 #if defined(S_IREAD)
 100 #define S_IROTH S_IREAD
 101 #elif defined(_S_IREAD)
 102 #define S_IROTH _S_IREAD
 103 #else
 104 #define S_IROTH 0004
 105 #endif
 106 #endif
 107
 108 #ifndef S_IWOTH
 109 #if defined(S_IWRITE)
 110 #define S_IWOTH S_IWRITE
 111 #elif defined(_S_IWRITE)
 112 #define S_IWOTH _S_IWRITE
 113 #else
 114 #define S_IWOTH 0002
 115 #endif
 116 #endif
 117
 118 #ifdef SP_NAMESPACE
 119 namespace SP_NAMESPACE {
 120 #endif
 121
 122 const int openFlags = O_CREAT|O_WRONLY|O_TRUNC|O_BINARY;
 123 const int protMode = S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH;
 124 const int bufSize = 8192;
 125
 126 OutputByteStream::OutputByteStream()
 127 : ptr_(0), end_(0)
 128 {
 129 }
 130
 131 OutputByteStream::~OutputByteStream()
 132 {
 133 }
 134
 135 void OutputByteStream::sputn(const char *s, size_t n)
 136 {
 137   for (; n > 0; n--, s++)
 138     sputc(*s);
 139 }
 140
 141 OutputByteStream &OutputByteStream::operator<<(long n)
 142 {
 143   char buf[32];
 144   sprintf(buf, "%ld", n);
 145   return *this << buf;
 146 }
 147
 148 OutputByteStream &OutputByteStream::operator<<(unsigned long n)
 149 {
 150   char buf[32];
 151   sprintf(buf, "%lu", n);
 152   return *this << buf;
 153 }
 154
 155 OutputByteStream &OutputByteStream::operator<<(const char *s)
 156 {
 157   while (*s)
 158     sputc(*s++);
 159   return *this;
 160 }
 161
 162 StrOutputByteStream::StrOutputByteStream()
 163 {
 164 }
 165
 166 void StrOutputByteStream::extractString(String<char> &str)
 167 {
 168   if (ptr_)
 169     buf_.resize(ptr_ - &buf_[0]);
 170   str.resize(0);
 171   buf_.swap(str);
 172   ptr_ = end_ = 0;
 173 }
 174
 175 void StrOutputByteStream::flush()
 176 {
 177 }
 178
 179 void StrOutputByteStream::flushBuf(char c)
 180 {
 181   if (!ptr_) {
 182     buf_.resize(16);
 183     ptr_ = &buf_[0];
 184   }
 185   else {
 186     size_t i = ptr_ - &buf_[0];
 187     buf_.resize(buf_.size()*2);
 188     ptr_ = &buf_[0] + i;
 189   }
 190   end_ = &buf_[0] + buf_.size();
 191   *ptr_++ = c;
 192 }
 193
 194 FileOutputByteStream::FileOutputByteStream()
 195 : fd_(-1)
 196 {
 197 }
 198
 199 FileOutputByteStream::FileOutputByteStream(int fd, Boolean closeFd)
 200 : fd_(-1)
 201 {
 202   attach(fd, closeFd);
 203 }
 204
 205 FileOutputByteStream::~FileOutputByteStream()
 206 {
 207   close();
 208 }
 209
 210 #ifdef SP_WIDE_SYSTEM
 211
 212 Boolean FileOutputByteStream::open(const wchar_t *filename)
 213 {
 214   int fd = _wopen(filename, openFlags, protMode);
 215   if (fd >= 0)
 216     return attach(fd);
 217   // _wopen will always fail on Windows 95
 218   String<char> buf;
 219   int len = WideCharToMultiByte(CP_ACP, 0, filename, -1, 0, 0, 0, 0);
 220   buf.resize(len + 1);
 221   WideCharToMultiByte(CP_ACP, 0, filename, -1, buf.begin(), len, 0, 0);
 222   buf[len] = '\0';
 223   return attach(::open(buf.data(), openFlags, protMode));
 224 }
 225
 226 #else /* not SP_WIDE_SYSTEM */
 227
 228 Boolean FileOutputByteStream::open(const char *filename)
 229 {
 230   return attach(::open(filename, openFlags, protMode));
 231 }
 232
 233 #endif /* not SP_WIDE_SYSTEM */
 234
 235 Boolean FileOutputByteStream::attach(int fd, Boolean closeFd)
 236 {
 237   close();
 238   fd_ = fd;
 239   closeFd_ = closeFd;
 240   return fd_ >= 0;
 241 }
 242
 243 Boolean FileOutputByteStream::close()
 244 {
 245   if (fd_ < 0)
 246     return 0;
 247   flush();
 248   int fd = fd_;
 249   fd_ = -1;
 250   if (!closeFd_)
 251     return 1;
 252   return ::close(fd) == 0;
 253 }
 254
 255 void FileOutputByteStream::flush()
 256 {
 257   if (!buf_.size()) {
 258     if (fd_ < 0)
 259       return;
 260     buf_.resize(bufSize);
 261     ptr_ = &buf_[0];
 262     end_ = ptr_ + buf_.size();
 263   }
 264   size_t n = ptr_ - &buf_[0];
 265   const char *s = buf_.data();
 266   while (n > 0) {
 267     int nw = ::write(fd_, s, n);
 268     if (nw < 0)
 269       break;
 270     n -= nw;
 271     s += nw;
 272   }
 273   ptr_ = &buf_[0];
 274 }
 275
 276 void FileOutputByteStream::flushBuf(char c)
 277 {
 278   flush();
 279   *ptr_++ = c;
 280 }
 281
 282 #ifdef SP_NAMESPACE
 283 }
 284 #endif