/* gensio.hpp -- generic char buffer and I/O facilities
* by
[email protected] at Tue Feb 26 13:30:02 CET 2002
*/
#ifdef __GNUC__
#ifndef __clang__
#pragma interface
#endif
#endif
#ifndef GENSIO_HPP
#define GENSIO_HPP 1
#include "config2.h"
#include <stdarg.h> /* va_list */
#include "gensi.hpp"
#include <stdio.h>
/* Naming conventions: *Encode is a specific, well-known PostScript or PDF
* encoding filter documented by Adobe, *Encoder is something more general.
*/
/** Writing 0 bytes for vi_write is interpreted as a close() operation.
* Constructor must not write anything to the underlying (lower-level)
* Encoder. Implementors must redefine vi_write(). Encoders may be stacked
* atop of each other.
*/
class Encoder: public GenBuffer::Writable { public:
/** Default: calls vi_write. */
virtual void vi_putcc(char c);
/** vi_write() must be called with positive slen_t for normal writing, and
* vi_write(?,0); must be called to signal EOF. After that, it is prohibited
* to call vi_write() either way.
*/
virtual void vi_write(char const*buf, slen_t len) =0;
/** Copies all data (till EOF) from stream `f' to (this). */
static void writeFrom(GenBuffer::Writable& out, FILE *f);
/** Copies all data (till EOF) from stream `f' to (this). */
static void writeFrom(GenBuffer::Writable& out, GenBuffer::Readable &in);
}; /* class Encoder */
/** Implementors must redefine vi_read(), and they may redefine vi_getcc().
* A .vi_read(?,0) has special meaning in a Decoder (but not in a normal
* Readable).
*/
class Decoder: public GenBuffer::Readable { public:
/** Calls vi_read(&ret,1). Note that this is the inverse of
* GenBuffer::Readable, because there vi_read() calls vi_getcc(). Decoders
* may be stacked atop of each other.
*/
virtual int vi_getcc();
/** vi_read() must be called with positive slen_t for normal reading, and
* vi_read(?,0); must be called to signal that the caller would not read from
* the Decoder again, so the decoder is allowed to release the resources
* used. After that, it is prohibited to call vi_read() either way.
*/
virtual slen_t vi_read(char *to_buf, slen_t max) =0;
}; /* class Decoder */
class DecoderTeller: public Decoder { public:
/** Equivalent of ftell() */
virtual long vi_tell() const=0;
};
class Filter { public:
/** Starts a child process (with popen(pipe_tmpl,"wb")), pipes data to it,
* writes output to a temporary file. After _everything_ is written, the
* temporary file is read and fed to out_.
*/
class PipeE: public Encoder {
public:
PipeE(GenBuffer::Writable &out_, char const*pipe_tmpl, slendiff_t i=0);
virtual ~PipeE();
virtual void vi_write(char const*buf, slen_t len);
protected:
FILE *p;
SimBuffer::B tmpname;
GenBuffer::Writable &out;
SimBuffer::B tmpename, redir_cmd;
/** Temporary source file name `%S', forces system() instead of popen() */
SimBuffer::B tmpsname;
/** vi_check() is called by this->vi_write(...) (and also when Broken Pipe)
* to check whether the encoding succeeded. vi_check() should raise an Error
* if it detects failure or simply return if it detects success. Default
* implementation: empty body which simly returns.
*/
virtual void vi_check();
protected:
/** Copies the rest of (seekable) file `f' to `out' (subsequent filters).
* `f' is initially positioned at the beginning. Must call fclose(f).
* vi_copy() should raise Error::...s. The default implementation
* just copies the data bytes verbatim. `out.vi_write(0,0);' will be called
* by vi_write().
*/
virtual void vi_copy(FILE *f);
};
/** Gobbles all data written to it, just like /dev/null */
class NullE: public Encoder {
public:
inline NullE() {}
inline virtual ~NullE() {}
inline virtual void vi_putcc(char) {}
inline virtual void vi_write(char const*, slen_t) {}
};
class VerbatimE: public Encoder {
public:
inline VerbatimE(GenBuffer::Writable& out_): out(&out_) {}
inline virtual void vi_write(char const*buf, slen_t len) { out->vi_write(buf,len); }
inline void setOut(GenBuffer::Writable& out_) { out=&out_; }
inline GenBuffer::Writable& getOut() const { return *out; }
protected:
GenBuffer::Writable* out;
};
class VerbatimCountE: public Encoder {
public:
inline VerbatimCountE(GenBuffer::Writable& out_): out(&out_), count(0) {}
inline virtual void vi_write(char const*buf, slen_t len) { out->vi_write(buf,len); count+=len; }
inline void setOut(GenBuffer::Writable& out_) { out=&out_; }
inline GenBuffer::Writable& getOut() const { return *out; }
inline slen_t getCount() const { return count; }
protected:
GenBuffer::Writable* out;
/** Number of bytes already written */
slen_t count;
};
class FILEE: public Encoder {
public:
inline FILEE(FILE *f_,bool closep_): f(f_), closep(closep_) {}
FILEE(char const* filename);
inline virtual void vi_putcc(char c) { MACRO_PUTC(c,f); }
virtual void vi_write(char const*buf, slen_t len);
void close();
protected:
FILE *f;
bool closep;
};
class PipeD: public Decoder { public:
PipeD(GenBuffer::Readable &in_, char const*pipe_tmpl, slendiff_t i=0);
virtual ~PipeD();
/** Equivalent to vi_read(&ret,1). */
virtual int vi_getcc();
/** Upon the first non-zero call, opens `p', calls vi_precopy() and
* closes `in'. Upon all non-zero calls,
* reads the temporary file with fread(). Upon vi_read(?,0), removes the
* the temporary files.
*/
virtual slen_t vi_read(char *to_buf, slen_t max);
protected:
/** 0: never-read (initial state), 1: vi_precopy() already called, 2: EOF reached, `in' closed */
int state;
/** opened with popen(?,"w") in state:0->1, then with fopen(?,"rb") in state==1 */
FILE *p;
/** Temporary destination file name `%D' */
SimBuffer::B tmpname;
GenBuffer::Readable ∈
SimBuffer::B tmpename, redir_cmd;
/** Temporary source file name `%S', forces system() instead of popen() */
SimBuffer::B tmpsname;
/** Copies the whole `in' to writable pipe `p'. `p' will be closed by the
* caller; in.read(0,0) will be called by the caller.
* vi_precopy() should raise Error::...s. The default implementation
* just copies the data bytes verbatim.
*/
virtual void vi_precopy();
/** vi_check() is called by this->vi_write(...) (and also when Broken Pipe)
* to check whether the encoding succeeded. vi_check() should raise an Error
* if it detects failure or simply return if it detects success. Default
* implementation: empty body which simly returns.
*/
virtual void vi_check();
private:
void do_close();
};
class VerbatimD: public Decoder {
public:
inline VerbatimD(GenBuffer::Readable& in_): in(in_) {}
/** Works fine even if len==0. */
inline virtual slen_t vi_read(char *buf, slen_t len) { return in.vi_read(buf,len); }
protected:
GenBuffer::Readable& in;
};
class FILED: public DecoderTeller {
public:
inline FILED(FILE *f_,bool closep_): f(f_), closep(closep_) {}
FILED(char const* filename);
inline virtual ~FILED() { close(); }
inline virtual int vi_getcc() { return MACRO_GETC(f); }
virtual slen_t vi_read(char *buf, slen_t len);
void close();
inline virtual long vi_tell() const { return ftell(f); }
protected:
FILE *f;
bool closep;
};
/**
* Sat Apr 19 12:05:49 CEST 2003
* Always opens the file in binary mode.
* First reads from the unget buffer, then reads from a FILE*. A typical
* usage is:
* { Filter::UngetFILED f("in.txt"); // fopen("in.txt","rb"), true);
* f.getUnget() << "Prepend this in front of first line.\n";
* int c;
* while ((c=f.vi_getcc()!=-1)) putchar(c);
* }
*/
class UngetFILED: public DecoderTeller {
public:
GenBuffer::Writable& getUnget() { return unget; }
BEGIN_STATIC_ENUM(unsigned char, closeMode_t)
CM_closep=1,
CM_unlinkp=2,
CM_keep_stdinp=4, /* CM_unlinkp, but keep STDIN open */
CM_seekablep=8, /* it is sure that this->f is seekable */
CM_MAX=4
END_STATIC_ENUM()
inline UngetFILED(FILE *f_, closeMode_t closep_): f(f_), closeMode(closep_), ftell_at(0), ofs(0) {}
UngetFILED(char const* filename_, FILE* stdin_f=(FILE*)NULLP, closeMode_t closeMode_=CM_closep);
inline virtual ~UngetFILED() { close(); }
virtual int vi_getcc();
virtual slen_t vi_read(char *buf, slen_t len);
/** Appends a line to buf, including delimiter unless EOF.
* @param delimiter: a char or negative to read to EOF
*/
void appendLine(GenBuffer::Writable &buf, int delimiter='\n');
void close();
// void checkFILE();
inline virtual long vi_tell() const { return ftell_at; } // return ftell_add+ofs+(f!=NULLP ? ftell(f) : 0); }
/**
* @return equivalent to getc(f) if this->f is not seekable, but returns
* -2 if this->f is seekable.
*/
int getc_seekable();
/**
* Actively tests whether FILE* is seekable. Doesn't work for regular files
* of 0 bytes.
*/
bool isSeekable();
/** Reading (this) and reading the returned FILE* will be equivalent
* (unless this->getUnget() is used later).
* Creates a temporary file if this->unget is not empty.
* @param seekable_p if true, the returned FILE* must be seekable. Possibly
* creates a temporary file.
*/
FILE* getFILE(bool seekable_p);
/** Implies a getFILE() */
void seek(long abs_ofs);
/** Tries to do an fseek(f, -slen, SEEK_CUR). On failure, appends to unget.
* The user should only unread() data obtained from vi_read().
*/
void unread(char const *s, slen_t slen);
inline char const* getFilename() const { return filename; }
inline char const* getFilenameDefault(char const *def) const { return filename==NULLP ? def : filename; }
inline bool hadError() const { return f!=NULLP && ferror(f); }
protected:
FILE *f;
unsigned char closeMode;
slen_t ftell_at;
slen_t ofs;
SimBuffer::B unget;
char const* filename;
};
/** Reads from a memory of a GenBuffer via first_sub and next_sub. The
* GenBuffer should not be changed during the read. The GenBuffer isn't
* delete()d by (this)
*/
class BufR: public GenBuffer::Readable {
public:
BufR(GenBuffer const& buf_);
virtual int vi_getcc();
virtual slen_t vi_read(char *to_buf, slen_t max);
virtual void vi_rewind();
protected:
GenBuffer const* bufp;
GenBuffer::Sub sub;
};
/** Reads from a consecutive memory area, which won't be
* delete()d by (this)
*/
class FlatD: public DecoderTeller /*GenBuffer::Readable*/ {
public:
FlatD(char const* s_, slen_t slen_);
FlatD(char const* s_);
virtual int vi_getcc();
virtual slen_t vi_read(char *to_buf, slen_t max);
virtual void vi_rewind();
inline int getcc() { return slen!=0 ? (slen--, *(unsigned char const*)s++) : -1; }
inline virtual long vi_tell() const { return s-sbeg; }
inline long tell() const { return s-sbeg; }
protected:
char const *s, *sbeg;
slen_t slen;
};
}; /* class Filter */
class Files {
public:
/** Formerly `class WritableFILE' */
class FILEW: public GenBuffer::Writable {
public:
inline FILEW(FILE *f_): f(f_) {}
inline virtual void vi_putcc(char c) { MACRO_PUTC(c,f); }
inline virtual void vi_write(char const*buf, slen_t len) { fwrite(buf, 1, len, f); }
virtual GenBuffer::Writable& vformat(slen_t n, char const *fmt, va_list ap);
/** appends; uses SimBuffer::B as temp */
virtual GenBuffer::Writable& vformat(char const *fmt, va_list ap);
inline void close() { fclose(f); }
inline void setF(FILE *f_) { f=f_; }
private:
FILE *f;
};
/** Formerly `class ReadableFILE'. Doesn't close the its FILE* automatically. */
class FILER: public GenBuffer::Readable {
public:
inline FILER(FILE *f_): f(f_) {}
inline virtual int vi_getcc() { return MACRO_GETC(f); }
inline virtual slen_t vi_read(char *to_buf, slen_t max) { return fread(to_buf, 1, max, f); }
inline void close() { fclose(f); }
inline virtual void vi_rewind() { rewind(f); }
private:
FILE *f;
};
/** @return the beginning of the last substring in param filename that does
* not contain '/' (dir separator)
*/
static char const* only_fext(char const*filename);
/** true iff temporary files should be removed at program finish end */
static bool tmpRemove;
/** @param fname must start with '/' (dir separator)
* @return true if file successfully created
*/
static FILE *try_dir(SimBuffer::B &dir, SimBuffer::B const&fname, char const*s1, char const*s2, char const*open_mode="wb");
/* The file will be opened for writing only. It won't be registered for
* for automatic removal.
* @param dir `dir' is empty: appends a unique filename for a temporary
* file. Otherwise: returns a unique filename in the specified directory.
* Creates the new file with 0 size.
* @param extension NULLP or a string specifying the extension of the file
* to create (should beginn with ".")
* @param open_mode "w", "wb" or "wb+"
* @return FILE* opened for writing for success, NULLP on failure
* --return true on success, false on failure
*/
static FILE *open_tmpnam(SimBuffer::B &dir, char const*open_mode="wb", char const*extension=(char const*)NULLP);
// static FILE *open_tmpnam(SimBuffer::B &dir, bool binary_p=true, char const*extension=(char const*)NULLP);
static bool find_tmpnam(SimBuffer::B &dir);
/** Calls lstat().
* @return (slen_t)-1 on error, the size otherwise
*/
static slen_t statSize(char const* filename);
/** Ensures that file will be removed (if possible...) when the process
* terminates. Makes a copy of the filename array, it won't use filename
* after returning.
*/
static void tmpRemoveCleanup(char const* filename);
/** Ensures that file will be removed (if possible...) when the process
* terminates. Copies the string filename. The file will be removed iff
* (*p!=NULLP) when the cleanup handler runs. The cleanup handler fclose()s
* the file before removing it.
*/
static void tmpRemoveCleanup(char const* filename, FILE**p);
/** Removes the file/entity if exists.
* @return 0 on success (file hadn't existed, a directory component
* hadn't existed, or the file has been successfully removed),
* 1 otherwise
*/
static int removeIf(char const *filename);
/** Sat Sep 7 20:58:54 CEST 2002 */
static void doSignalCleanup();
/** Set the specified file descriptor to binary mode. Useful for stdin (0),
* stdout (1), stderr (2) on Win32 and DOS systems. On these systems, this
* call has the effect of `fopen(..., "rb");' On UNIX, file
* descriptors are always binary.
*/
#if HAVE_DOS_BINARY
static void set_binary_mode(int fd, bool binary);
#else
static inline void set_binary_mode(int,bool) {}
#endif
/** Like the system(3) call, but it is able to run a string containing
* multiple lines of commands. On Win32, it creates a batch file if
* necessary.
*/
static int system3(char const *commands);
};
#endif