--- np2/common/textfile.c 2003/10/16 17:57:10 1.1.1.1 +++ np2/common/textfile.c 2005/05/15 20:18:31 1.9 @@ -1,119 +1,577 @@ #include "compiler.h" -#include "dosio.h" +#include "strres.h" #include "textfile.h" +#if defined(SUPPORT_TEXTCNV) +#include "textcnv.h" +#endif +#include "dosio.h" -TEXTFILEH textfile_open(const char *filename, UINT buffersize) { +enum { + TFMODE_READ = 0x01, + TFMODE_WRITE = 0x02 +}; + +struct _textfile; +typedef struct _textfile _TEXTFILE; +typedef struct _textfile *TEXTFILE; + +typedef BRESULT (*READFN)(TEXTFILE tf, void *buffer, UINT size); + +struct _textfile { + UINT8 mode; + UINT8 width; +#if defined(SUPPORT_TEXTCNV) + UINT8 access; + UINT8 xendian; +#endif + FILEH fh; + long fpos; + UINT8 *buf; + UINT bufsize; + UINT bufpos; + UINT bufrem; + READFN readfn; +#if defined(SUPPORT_TEXTCNV) + UINT8 *cnvbuf; + UINT cnvbufsize; + TCTOOEM tooem; + TCFROMOEM fromoem; +#endif +}; - FILEH fh; - TEXTFILEH ret; +// ---- ReadA + +static UINT readbufferA(TEXTFILE tf) { + + UINT rsize; + + if (tf->bufrem == 0) { + rsize = file_read(tf->fh, tf->buf, tf->bufsize); + rsize = rsize / sizeof(char); + tf->fpos += rsize * sizeof(char); + tf->bufpos = 0; + tf->bufrem = rsize; + } + return(tf->bufrem); +} + +static BRESULT readlineA(TEXTFILE tf, void *buffer, UINT size) { + + char *dst; + BOOL crlf; + BRESULT ret; + char c; +const char *src; + UINT pos; + + if (size == 0) { + dst = NULL; + size = 0; + } + else { + dst = (char *)buffer; + size--; + } + + crlf = FALSE; + ret = FAILURE; + c = 0; + do { + if (readbufferA(tf) == 0) { + break; + } + ret = SUCCESS; + src = (char *)tf->buf; + src += tf->bufpos; + pos = 0; + while(posbufrem) { + c = src[pos]; + pos++; + if ((c == 0x0d) || (c == 0x0a)) { + crlf = TRUE; + break; + } + if (size) { + size--; + *dst++ = c; + } + } + tf->bufpos += pos; + tf->bufrem -= pos; + } while(!crlf); + if ((crlf) && (c == 0x0d)) { + if (readbufferA(tf) != 0) { + src = (char *)tf->buf; + src += tf->bufpos; + if (*src == 0x0a) { + tf->bufpos++; + tf->bufrem--; + } + } + } + if (dst) { + *dst = '\0'; + } + return(ret); +} + + +// ---- ReadW + +static UINT readbufferW(TEXTFILE tf) { + + UINT rsize; + UINT8 *buf; + + if (tf->bufrem == 0) { + buf = tf->buf; + rsize = file_read(tf->fh, buf, tf->bufsize); + rsize = rsize / sizeof(UINT16); + tf->fpos += rsize * sizeof(UINT16); + tf->bufpos = 0; + tf->bufrem = rsize; +#if defined(SUPPORT_TEXTCNV) + if (tf->xendian) { + textcnv_swapendian16(buf, rsize); + } +#endif // defined(SUPPORT_TEXTCNV) + } + return(tf->bufrem); +} + +static BRESULT readlineW(TEXTFILE tf, void *buffer, UINT size) { + + UINT16 *dst; + BOOL crlf; + BRESULT ret; + UINT16 c; +const UINT16 *src; + UINT pos; + + if (size == 0) { + dst = NULL; + size = 0; + } + else { + dst = (UINT16 *)buffer; + size--; + } + crlf = FALSE; + ret = FAILURE; + c = 0; + do { + if (readbufferW(tf) == 0) { + break; + } + ret = SUCCESS; + src = (UINT16 *)tf->buf; + src += tf->bufpos; + pos = 0; + while(posbufrem) { + c = src[pos]; + pos++; + if ((c == 0x0d) || (c == 0x0a)) { + crlf = TRUE; + break; + } + if (size) { + size--; + *dst++ = c; + } + } + tf->bufpos += pos; + tf->bufrem -= pos; + } while(!crlf); + if ((crlf) && (c == 0x0d)) { + if (readbufferW(tf) != 0) { + src = (UINT16 *)tf->buf; + src += tf->bufpos; + if (*src == 0x0a) { + tf->bufpos++; + tf->bufrem--; + } + } + } + if (dst) { + *dst = '\0'; + } + return(ret); +} + + +// ---- read with convert + +#if defined(SUPPORT_TEXTCNV) +static BRESULT readlineAtoOEM(TEXTFILE tf, void *buffer, UINT size) { + + BRESULT ret; + + ret = readlineA(tf, tf->cnvbuf, tf->cnvbufsize); + if (ret == SUCCESS) { + (tf->tooem)(buffer, size, tf->cnvbuf, (UINT)-1); + } + return(ret); +} + +static BRESULT readlineWtoOEM(TEXTFILE tf, void *buffer, UINT size) { + + BRESULT ret; + + ret = readlineW(tf, tf->cnvbuf, tf->cnvbufsize / 2); + if (ret == SUCCESS) { + (tf->tooem)(buffer, size, tf->cnvbuf, (UINT)-1); + } + return(ret); +} +#endif // defined(SUPPORT_TEXTCNV) + + +// ---- write + +static BRESULT flushwritebuffer(TEXTFILE tf) { + + UINT size; + UINT wsize; + + if (tf->bufpos) { + size = tf->bufpos * tf->width; + wsize = file_write(tf->fh, tf->buf, size); + tf->fpos += wsize; + if (wsize != size) { + return(FAILURE); + } + } + return(SUCCESS); +} + +static BRESULT writebufferA(TEXTFILE tf, const void *buffer, UINT size) { + + BRESULT ret; +const UINT8 *p; + UINT wsize; + + ret = SUCCESS; + p = (UINT8 *)buffer; + while(size) { + wsize = min(size, tf->bufrem); + if (wsize) { + CopyMemory(tf->buf + tf->bufpos, p, wsize); + p += wsize; + size -= wsize; + tf->bufpos += wsize; + tf->bufrem -= wsize; + } + if (tf->bufrem == 0) { + ret = flushwritebuffer(tf); + tf->bufpos = 0; + tf->bufrem = tf->bufsize / sizeof(char); + } + } + return(ret); +} + +static BRESULT writebufferW(TEXTFILE tf, const void *buffer, UINT size) { + + BRESULT ret; +const UINT8 *p; + UINT8 *q; + UINT wsize; + + ret = SUCCESS; + p = (UINT8 *)buffer; + while(size) { + wsize = min(size, tf->bufrem); + if (wsize) { + q = tf->buf + (tf->bufpos * sizeof(UINT16)); + CopyMemory(q, p, wsize * sizeof(UINT16)); +#if defined(SUPPORT_TEXTCNV) + if (tf->xendian) { + textcnv_swapendian16(q, wsize); + } +#endif // defined(SUPPORT_TEXTCNV) + p += wsize * sizeof(UINT16); + size -= wsize; + tf->bufpos += wsize; + tf->bufrem -= wsize; + } + if (tf->bufrem == 0) { + ret = flushwritebuffer(tf); + tf->bufpos = 0; + tf->bufrem = tf->bufsize / sizeof(UINT16); + } + } + return(ret); +} + + +// ---- + +static TEXTFILEH registfile(FILEH fh, UINT buffersize, + const UINT8 *hdr, UINT hdrsize) { + + UINT cnvbufsize; +#if defined(SUPPORT_TEXTCNV) + TCINF inf; +#endif + long fpos; + UINT8 width; + READFN readfn; + TEXTFILE ret; + + buffersize = buffersize & (~3); if (buffersize < 256) { buffersize = 256; } - fh = file_open(filename); - if (fh != FILEH_INVALID) { - ret = (TEXTFILEH)_MALLOC(sizeof(_TEXTFILE) + buffersize, filename); - if (ret) { - ZeroMemory(ret, sizeof(_TEXTFILE) + buffersize); - ret->fh = (void *)fh; - ret->buffersize = buffersize; - return(ret); + cnvbufsize = 0; +#if defined(SUPPORT_TEXTCNV) + if (textcnv_getinfo(&inf, hdr, hdrsize) == 0) { + return(NULL); + } + fpos = inf.hdrsize; + width = inf.width; + if (inf.width == 1) { + readfn = (inf.tooem != NULL)?readlineAtoOEM:readlineA; + } + else if (inf.width == 2) { + buffersize = buffersize * 2; + readfn = (inf.tooem != NULL)?readlineWtoOEM:readlineW; + } + else { + return(NULL); + } + if ((inf.tooem != NULL) || (inf.fromoem != NULL)) { + cnvbufsize = buffersize; + } +#else // defined(SUPPORT_TEXTCNV) + fpos = 0; + width = 1; + if ((hdrsize >= 3) && + (hdr[0] == 0xef) && (hdr[1] == 0xbb) && (hdr[2] == 0xbf)) { + // UTF-8 + fpos = 3; + } + else if ((hdrsize >= 2) && (hdr[0] == 0xff) && (hdr[1] == 0xfe)) { + // UCSLE + fpos = 2; + width = 2; +#if defined(BYTESEX_BIG) + return(NULL); +#endif // defined(BYTESEX_BIG) + } + else if ((hdrsize >= 2) && (hdr[0] == 0xfe) && (hdr[1] == 0xff)) { + // UCS2BE + fpos = 2; + width = 2; +#if defined(BYTESEX_LITTLE) + return(NULL); +#endif // defined(BYTESEX_LITTLE) + } + if (width != sizeof(OEMCHAR)) { + return(NULL); + } + buffersize = buffersize * sizeof(OEMCHAR); + readfn = (width == 2)?readlineW:readlineA; +#endif // defined(SUPPORT_TEXTCNV) + + ret = (TEXTFILE)_MALLOC(sizeof(_TEXTFILE) + buffersize + cnvbufsize, + "TEXTFILE"); + if (ret == NULL) { + return(NULL); + } + ZeroMemory(ret, sizeof(_TEXTFILE)); +// ret->mode = 0; + ret->width = width; + ret->fh = fh; + ret->fpos = fpos; + ret->buf = (UINT8 *)(ret + 1); + ret->bufsize = buffersize; + ret->readfn = readfn; +#if defined(SUPPORT_TEXTCNV) + ret->access = inf.caps; + ret->xendian = inf.xendian; + ret->cnvbuf = ret->buf + buffersize; + ret->cnvbufsize = cnvbufsize; + ret->tooem = inf.tooem; + ret->fromoem = inf.fromoem; +#endif // defined(SUPPORT_TEXTCNV) + return((TEXTFILEH)ret); +} + +static BRESULT flushfile(TEXTFILE tf) { + + BRESULT ret; + long fpos; + UINT size; + UINT wsize; + + ret = SUCCESS; + if (tf->mode & TFMODE_READ) { + fpos = tf->fpos - (tf->bufrem * tf->width); + tf->fpos = file_seek(tf->fh, fpos, FSEEK_SET); + if (tf->fpos != fpos) { + ret = FAILURE; + } + } + else if (tf->mode & TFMODE_WRITE) { + if (tf->bufpos) { + size = tf->bufpos * tf->width; + wsize = file_write(tf->fh, tf->buf, size); + if (wsize != size) { + ret = FAILURE; + } + tf->fpos += wsize; + } + } + else { + fpos = file_seek(tf->fh, tf->fpos, FSEEK_SET); + if (tf->fpos != fpos) { + ret = FAILURE; } - file_close(fh); + tf->fpos = fpos; + } + tf->mode = 0; + tf->bufpos = 0; + tf->bufrem = 0; + return(ret); +} + + +// ---- + +TEXTFILEH textfile_open(const OEMCHAR *filename, UINT buffersize) { + + FILEH fh; + UINT8 hdr[4]; + UINT hdrsize; + TEXTFILEH ret; + + fh = file_open_rb(filename); + if (fh == FILEH_INVALID) { + goto tfo_err; + } + hdrsize = file_read(fh, hdr, sizeof(hdr)); + ret = registfile(fh, buffersize, hdr, hdrsize); + if (ret) { + return(ret); } + file_close(fh); + +tfo_err: return(NULL); } +TEXTFILEH textfile_create(const OEMCHAR *filename, UINT buffersize) { -void textfile_close(TEXTFILEH fh) { + FILEH fh; +const UINT8 *hdr; + UINT hdrsize; + TEXTFILEH ret; - if (fh) { - file_close((FILEH)fh->fh); - _MFREE(fh); + fh = file_create(filename); + if (fh == FILEH_INVALID) { + goto tfc_err1; } +#if defined(OSLANG_UTF8) + hdr = str_utf8; + hdrsize = sizeof(str_utf8); +#elif defined(OSLANG_UCS2) + hdr = (UINT8 *)str_ucs2; + hdrsize = sizeof(str_ucs2); +#else + hdr = NULL; + hdrsize = 0; +#endif + if ((hdrsize) && (file_write(fh, hdr, hdrsize) != hdrsize)) { + goto tfc_err2; + } + ret = registfile(fh, buffersize, hdr, hdrsize); + if (ret) { + return(ret); + } + +tfc_err2: + file_close(fh); + +tfc_err1: + return(NULL); } +BRESULT textfile_read(TEXTFILEH tfh, OEMCHAR *buffer, UINT size) { -static BOOL getnextstring(TEXTFILEH fh) { + TEXTFILE tf; - UINT rsize; + tf = (TEXTFILE)tfh; + if (tf == NULL) { + return(FAILURE); + } +#if defined(SUPPORT_TEXTCNV) + if (!(tf->access & TEXTCNV_READ)) { + return(FAILURE); + } +#endif // defined(SUPPORT_TEXTCNV) + if (!(tf->mode & TFMODE_READ)) { + flushfile(tf); + tf->mode = TFMODE_READ; + } + return((tf->readfn)(tf, buffer, size)); +} + +BRESULT textfile_write(TEXTFILEH tfh, const OEMCHAR *buffer) { - if (!fh) { + TEXTFILE tf; +const void *buf; + UINT leng; + + tf = (TEXTFILE)tfh; + if (tf == NULL) { return(FAILURE); } - if (file_seek((FILEH)fh->fh, fh->fhpos, 0) != fh->fhpos) { +#if defined(SUPPORT_TEXTCNV) + if (!(tf->access & TEXTCNV_WRITE)) { return(FAILURE); } - rsize = file_read((FILEH)fh->fh, fh + 1, fh->buffersize); - if (rsize == (UINT)-1) { +#endif // defined(SUPPORT_TEXTCNV) + if (!(tf->mode & TFMODE_WRITE)) { + flushfile(tf); + tf->mode = TFMODE_WRITE; + } + leng = OEMSTRLEN(buffer); +#if defined(SUPPORT_TEXTCNV) + if (tf->fromoem != NULL) { + leng = (tf->fromoem)(tf->cnvbuf, tf->cnvbufsize / tf->width, + buffer, leng); + buf = tf->cnvbuf; + } + else { + buf = buffer; + } +#else // defined(SUPPORT_TEXTCNV) + buf = buffer; +#endif // defined(SUPPORT_TEXTCNV) + if (tf->width == 1) { + return(writebufferA(tf, buf, leng)); + } + else if (tf->width == 2) { + return(writebufferW(tf, buf, leng)); + } + else { return(FAILURE); } - fh->fhpos += rsize; - fh->pos = 0; - fh->remain = rsize; - return(SUCCESS); } +void textfile_close(TEXTFILEH tfh) { -BOOL textfile_read(TEXTFILEH fh, char *buffer, UINT size) { + TEXTFILE tf; - char c = '\0'; - char *p; - BOOL crlf; - BOOL ret = FAILURE; - - if ((fh) && (size > 0)) { - size--; - crlf = FALSE; - do { - if ((!fh->remain) && (getnextstring(fh))) { - return(FAILURE); - } - if (!fh->remain) { - break; - } - ret = SUCCESS; - p = (char *)(fh + 1); - p += fh->pos; - while((fh->remain) && (size)) { - c = *p++; - fh->pos++; - fh->remain--; - if ((c == 0x0d) || (c == 0x0a)) { - crlf = TRUE; - break; - } - *buffer++ = c; - size--; - } - if (!crlf) { - while((fh->remain) && (size)) { - c = *p++; - fh->pos++; - fh->remain--; - if ((c == 0x0d) || (c == 0x0a)) { - crlf = TRUE; - break; - } - } - } - } while(!crlf); - if ((crlf) && (c == 0x0d)) { - if ((!fh->remain) && (getnextstring(fh))) { - return(FAILURE); - } - if (fh->remain) { - p = (char *)(fh + 1); - p += fh->pos; - if (*p == 0x0a) { - fh->pos++; - fh->remain--; - } - } + tf = (TEXTFILE)tfh; + if (tf) { + if (tf->mode & TFMODE_WRITE) { + flushfile(tf); } - *buffer = '\0'; + file_close(tf->fh); + _MFREE(tf); } - return(ret); }