/*
 * Copyright (c) 1991-1994  Sony Corporation
 * 
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 * 
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL SONY CORPORATION BE LIABLE FOR ANY CLAIM,
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR
 * THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 * 
 * Except as contained in this notice, the name of Sony Corporation
 * shall not be used in advertising or otherwise to promote the sale, use
 * or other dealings in this Software without prior written authorization
 * from Sony Corporation.
 *
 */

/*
 * $SonyRCSfile: depend.c,v $  
 * $SonyRevision: 1.4 $ 
 * $SonyDate: 1997/10/07 07:39:45 $
 */

#include <config.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <unistd.h>
#include "sj_kcnv.h"
#include <errno.h>
#include <sys/file.h>
#include <sys/stat.h>
#include "Dict.h"
#include "sj3err.h"

#include "sj_kanakan.h"

#if defined(__FD_SET) && !defined(FD_SET)
# define FD_SET __FD_SET
# define FD_CLR __FD_CLR
#endif

DictFile *dictlink = NULL;
StdyFile *stdylink = NULL;

static long
get4byte(u_char *p)
{
	long	i;

	i = *p;
	i = (i << 8) + *++p;
	i = (i << 8) + *++p;
	return ((i << 8) + *++p);
}
static void
Put4byte(u_char *p, long n)
{
	p += 3;
	*p-- = n; n >>= 8;
	*p-- = n; n >>= 8;
	*p-- = n; n >>= 8;
	*p = n;
}
#define put4byte(p, n)	Put4byte((p), (long)(n))



static int
fgetfile(FILE *fp, long pos, long len, u_char *p)
{
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; return ERROR;
	}
	if (fread(p, len, 1, fp) != 1) {
		serv_errno = SJ3_FileReadError; return ERROR;
	}
	return SJ3_NormalEnd;
}
static int
fputfile(FILE *fp, long pos, int len, u_char *p)
{
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; return ERROR;
	}
	if (fwrite(p, len, 1, fp) != 1) {
		serv_errno = SJ3_FileWriteError; return ERROR;
	}
	return SJ3_NormalEnd;
}
static int
getfile(int fd, int pos, int len, u_char *p)
{
	if (lseek(fd, pos, L_SET) == ERROR) {
		serv_errno = SJ3_FileSeekError; return ERROR;
	}
	if (read(fd, p, len) != len) {
		serv_errno = SJ3_FileReadError; return ERROR;
	}
	return SJ3_NormalEnd;
}
static int
putfile(int fd, int pos, int len, u_char *p)
{
	if (lseek(fd, pos, L_SET) == ERROR) {
		serv_errno = SJ3_FileSeekError; return ERROR;
	}
	if (write(fd, p, len) != len) {
		serv_errno = SJ3_FileWriteError; return ERROR;
	}
	return SJ3_NormalEnd;
}



static int
check_passwd(u_char *buf, char *passwd)
{
	buf += PASSWDPOS;
	return (*buf && strncmp(passwd, buf, PASSWDLEN)) ? FALSE : TRUE;
}
static void
set_passwd(u_char *buf, char *passwd)
{
	strncpy(buf + PASSWDPOS, passwd, PASSWDLEN);
}



static int
check_dictfile(u_char *buf)
{
	return (DICTVERSION != get4byte(buf + VERSIONPOS)) ? FALSE : TRUE;
}



static DictFile *
search_same_dict(ino_t ino)
{
	DictFile *p;

	for (p = dictlink ; p != NULL ; p = p -> link)
		if (p->dict.dicid == ino) return p;

	return NULL;
}



static int
getofs(DictFile *dp)
{
	idxofs = dp -> ofsptr;
	return 0;
}
static int
getidx(DictFile *dp)
{
	idxbuf = dp -> buffer + dp -> idxstrt;
	return 0;
}
/* XXX: not IFunc */
static int
getdic(DictFile *dp, TypeDicSeg seg)
{
	if (seg >= dp->dict.segunit) return -1;
	dicbuf = dp->buffer + dp->segstrt + dp->dict.seglen * seg;
	return 0;
}
static int
putidx(DictFile *dp)
{
	return putfile(dp->fd, dp->idxstrt, dp->dict.idxlen, idxbuf);
}
/* XXX: not IFunc */
static int
putdic(DictFile *dp, TypeDicSeg seg)
{
	u_char	*p;
	long	i, j;

	if (seg >= dp->dict.maxunit) return -1;
	i = dp->segstrt + dp->dict.seglen * (seg + 1);
	if (i > dp->bufsiz) {
		if (!(p = (u_char *)malloc(i))) return -1;
		memcpy(p, dp->buffer, dp->bufsiz);

		if (idxbuf - dp->buffer == dp->idxstrt)
			idxbuf = p + dp->idxstrt;
		if ((dicbuf >= dp->buffer) &&
		    ((j = dicbuf - dp->buffer) < dp->bufsiz))
			dicbuf = p + j;

		free((char *)(dp -> buffer));
		dp -> buffer = p;
		dp -> bufsiz = i;
	}

	i = dp->segstrt + dp->dict.seglen * seg;
	if ((p = dp->buffer + i) != dicbuf) memcpy(p, dicbuf, dp->dict.seglen);

	return putfile(dp->fd, i, dp->dict.seglen, p);
}
/* XXX: not IFunc */
static int
rszdic(DictFile *dp, TypeDicSeg seg)
{
	long	i;
	u_char	*p;

	i = dp->segstrt + dp->dict.seglen * seg;
	if (dp->bufsiz != i) {
		if (!(p = (u_char *)malloc(i))) return -1;
		memcpy(p, dp->buffer, ((i > dp->bufsiz) ? dp->bufsiz : i));

		if (idxbuf - dp->buffer == dp->idxstrt)
			idxbuf = p + dp->idxstrt;
		if ((dicbuf >= dp->buffer) &&
		    ((i = dicbuf - dp->buffer) < dp->bufsiz))
			dicbuf = p + i;

		free((char *)(dp -> buffer));
		dp -> buffer = p;
		dp -> bufsiz = i;
		if (ftruncate(dp->fd, i) == -1) return -1;
	}
	put4byte(dp->buffer + DICTSEGNUM, seg);
	return putfile(dp->fd, 0, HEADERLENGTH, dp->buffer);
}



DictFile *
opendict(char *name, char *passwd)
{
	FILE		*fp;
	struct stat	sbuf;
	DictFile	*dfp;
	u_char		tmp[HEADERLENGTH];
	int		i;
	u_char		*dp;

	if (stat(name, &sbuf) == ERROR) {
		if (errno == ENOENT)
			serv_errno = SJ3_FileNotExist;
		else
			serv_errno = SJ3_CannotAccessFile;
		return NULL;
	}

	if ((dfp = search_same_dict(sbuf.st_ino)) != NULL) {
		dfp -> refcnt++; return dfp;
	}

	if ((fp = fopen(name, "r+")) == NULL) {
		if ((fp = fopen(name, "r")) == NULL) {
			serv_errno = SJ3_CannotOpenFile; return NULL;
		}
		i = FALSE;
	}
	else
		i = TRUE;

	if (fgetfile(fp, 0L, sizeof(tmp), tmp) == ERROR) goto error1;

	if (!check_dictfile(tmp)) {
		serv_errno = SJ3_IllegalDictFile; goto error1;
	}

	if (!check_passwd(tmp, passwd)) {
		serv_errno = SJ3_IncorrectPasswd; goto error1;
	}

	if ((dp = (u_char *)malloc((long)sbuf.st_size)) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error1;
	}
	if ((dfp = (DictFile *)malloc(sizeof(*dfp))) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error2;
	}
	memset(dfp, '\0', sizeof(*dfp));

	if (fgetfile(fp, 0L, (long)sbuf.st_size, dp) == ERROR) goto error3;

	dfp -> dict.dicid   = sbuf.st_ino;
	dfp -> dict.idxlen  = get4byte(dp + DICTIDXLEN);
	dfp -> dict.seglen  = get4byte(dp + DICTSEGLEN);
	dfp -> dict.segunit = get4byte(dp + DICTSEGNUM);
	dfp -> dict.maxunit = i ? get4byte(dp + DICTSEGMAX) : 0 ;
	dfp -> dict.getofs  = getofs;
	dfp -> dict.getidx  = getidx;
	dfp -> dict.getdic  = getdic;
	dfp -> dict.putidx  = putidx;
	dfp -> dict.putdic  = putdic;
	dfp -> dict.rszdic  = rszdic;
	dfp -> refcnt       = 1;
	dfp -> fp           = fp;
	dfp -> fd           = fileno(fp);
	dfp -> buffer       = dp;
	dfp -> bufsiz       = sbuf.st_size;
	dfp -> idxstrt      = get4byte(dp + DICTIDXPOS);
	dfp -> segstrt      = get4byte(dp + DICTSEGPOS);

	i = dfp -> dict.segunit;
	if (i < dfp -> dict.maxunit) i = dfp -> dict.maxunit;
	i *= sizeof(TypeIdxOfs);
	if ((dfp -> ofsptr = (TypeIdxOfs *)malloc(i)) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error3;
	}
	mkidxtbl(&(dfp -> dict));

	dfp -> link         = dictlink;
	dictlink = dfp;

	return dfp;


error3:	free((char *)dfp);
error2:	free((char *)dp);
error1:	fclose(fp);

	return NULL;
}



int
closedict(DictFile *dfp)
{
	DictFile	*df;

	if (--dfp->refcnt > 0) return 0;

	if (dictlink == dfp)
		dictlink = dfp -> link;
	else {
		for (df = dictlink ; df ; df = df -> link)
			if (df -> link == dfp) {
				df -> link = dfp -> link;
				break;
			}
	}

	if (dfp -> fp) fclose(dfp ->fp);

	free((char *)(dfp -> ofsptr));
	free((char *)(dfp -> buffer));
	free((char *)dfp);

	return 0;
}

static	fd_set	zero_fd_set; /* XXX */

void
lock_dict(DictFile *p, int fd)
{
	FD_SET(fd, &(p -> lock));
}

void
unlock_dict(DictFile *p, int fd)
{
	FD_CLR(fd, &(p -> lock));
}

int is_dict_locked(DictFile *p)
{
	return memcmp(&(p -> lock), &zero_fd_set, sizeof(zero_fd_set));
}

static int
check_stdyfile(u_char *buf)
{
	return (STDYVERSION != get4byte(buf + VERSIONPOS)) ? FALSE : TRUE;
}

static StdyFile *
search_same_stdy(ino_t ino)
{
	StdyFile *p;

	for (p = stdylink ; p != NULL ; p = p -> link)
		if (p -> inode == ino) return p;

	return NULL;
}

StdyFile *
openstdy(char *name, char *passwd)
{
	FILE		*fp;
	struct stat	sbuf;
	StdyFile	*sfp;
	u_char		*hd;
	STDYIN		*sp;
	u_short		*cip;
	u_char		*clp;
	long		stdycnt, stdypos, stdylen, stdymax;
	long		clidxpos, clidxlen;
	long		clstdypos, clstdylen, clstdystep;
	long		len;

	if (stat(name, &sbuf) == ERROR) {
		if (errno == ENOENT)
			serv_errno = SJ3_FileNotExist;
		else
			serv_errno = SJ3_CannotAccessFile;
		return NULL;
	}

	if ((sfp = search_same_stdy(sbuf.st_ino)) != NULL) {
		sfp -> refcnt++; return sfp;
	}

	if ((hd = (u_char *)malloc(HEADERLENGTH + COMMENTLENGTH)) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; return NULL;
	}

	if ((fp = fopen(name, "r+")) == NULL) {
		serv_errno = SJ3_CannotOpenFile; goto error0;
	}

	if (fgetfile(fp, 0L, HEADERLENGTH + COMMENTLENGTH, hd) == ERROR)
		goto error1;

	if (!check_stdyfile(hd)) {
		serv_errno = SJ3_IllegalStdyFile; goto error1;
	}

	if (!check_passwd(hd, passwd)) {
		serv_errno = SJ3_IncorrectPasswd; goto error1;
	}

	stdycnt    = get4byte(hd + STDYNORMCNT);
	stdypos    = get4byte(hd + STDYNORMPOS);
	stdylen    = get4byte(hd + STDYNORMLEN);
	stdymax    = get4byte(hd + STDYNORMNUM);
	clidxpos   = get4byte(hd + STDYCLIDXPOS);
	clidxlen   = get4byte(hd + STDYCLIDXLEN);
	clstdypos  = get4byte(hd + STDYCLSEGPOS);
	clstdylen  = get4byte(hd + STDYCLSEGLEN);
	clstdystep = get4byte(hd + STDYCLIDXSTEP);
	len = sizeof(STDYIN) * stdymax;

	if ((sfp = (StdyFile *)malloc(sizeof(*sfp))) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error1;
	}
	memset(sfp, '\0', sizeof(*sfp));
	if ((sp = (STDYIN *)malloc(len)) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error2;
	}
	if ((cip = (u_short *)malloc(clidxlen)) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error3;
	}
	if ((clp = (u_char *)malloc(clstdylen)) == NULL) {
		serv_errno = SJ3_NotEnoughMemory; goto error4;
	}

	if (fgetfile(fp, clidxpos, clidxlen, (u_char *)cip) == ERROR) goto error5; /* XXX */
	if (fgetfile(fp, clstdypos, clstdylen, (u_char *)clp) == ERROR) goto error5; /* XXX */
	if (fgetfile(fp, stdypos, len, (u_char *)sp) == ERROR) stdycnt = 0; /* XXX */

	sfp -> stdy.stdycnt    = stdycnt;
	sfp -> stdy.stdymax    = stdymax;
	sfp -> stdy.stdydic    = sp;
	sfp -> stdy.clstdystep = clstdystep;
	sfp -> stdy.clstdyidx  = cip;
	sfp -> stdy.clstdylen  = clstdylen;
	sfp -> stdy.clstdydic  = clp;
	sfp -> refcnt          = 1;
	sfp -> inode           = sbuf.st_ino;
	sfp -> fp              = fp;
	sfp -> fd              = fileno(fp);
	sfp -> header          = hd;

	sfp -> link            = stdylink;
	stdylink = sfp;

	return sfp;

error5:	free((char *)clp);
error4:	free((char *)cip);
error3:	free((char *)sp);
error2:	free((char *)sfp);
error1:	fclose(fp);
error0:	free((char *)hd);

	return NULL;
}

int
closestdy(StdyFile *sfp)
{
	StdyFile	*sf;

	if (--sfp->refcnt > 0) return 0;

	if (stdylink == sfp)
		stdylink = sfp -> link;
	else {
		for (sf = stdylink ; sf ; sf = sf -> link)
			if (sf -> link == sfp) {
				sf -> link = sfp -> link;
				break;
			}
	}

	if (sfp -> fp) fclose(sfp ->fp);

	free((char *)sfp -> stdy.stdydic);
	free((char *)sfp -> stdy.clstdyidx);
	free((char *)sfp -> stdy.clstdydic);
	free((char *)sfp -> header);
	free((char *)sfp);

	return 0;
}



int
putstydic()
{
	int	fd;
	u_char	*hd;
	StdyFile *sf;
	long	len;

	sf = (StdyFile *)stdy_base;
	fd = sf -> fd;
	hd = sf -> header;

	put4byte(hd + STDYNORMCNT, sf -> stdy.stdycnt);

	len = sizeof(STDYIN) * sf -> stdy.stdymax;
	put4byte(hd + STDYNORMLEN, len);

	if (putfile(fd, 0, HEADERLENGTH + COMMENTLENGTH, hd)) return ERROR;

	return putfile(fd, get4byte(hd + STDYNORMPOS), len, (u_char *)sf -> stdy.stdydic);
}

int
putcldic()
{
	int	fd;
	u_char	*hd;
	StdyFile *sf;

	sf = (StdyFile *)stdy_base;
	fd = sf -> fd;
	hd = sf -> header;

	if (putfile(fd, get4byte(hd + STDYCLIDXPOS),
		get4byte(hd + STDYCLIDXLEN), (u_char *)sf -> stdy.clstdyidx)) /* XXX */
		return ERROR;
	return putfile(fd, get4byte(hd + STDYCLSEGPOS),
			get4byte(hd + STDYCLSEGLEN), sf -> stdy.clstdydic);
}

int
makedict(char *path, int idxlen, int seglen, int segnum)
{
	FILE	*fp;
	u_char	tmp[HEADERLENGTH + COMMENTLENGTH];
	long	pos = HEADERLENGTH + COMMENTLENGTH;
	int	ret = ERROR;
	int	i;

	if (idxlen < MINIDXLEN || seglen < MINSEGLEN || segnum < MINSEGNUM) {
		serv_errno = SJ3_IllegalParameter; return ret;
	}

	if (!(fp = fopen(path, "w"))) {
		serv_errno = SJ3_CannotCreateFile; return ret;
	}
	memset(tmp, '\0', sizeof(tmp));
	put4byte(tmp + VERSIONPOS, DICTVERSION);

	put4byte(tmp + DICTIDXPOS, pos);
	put4byte(tmp + DICTIDXLEN, idxlen);
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; goto error;
	}
	for (i = 0 ; i < idxlen ; i++)
		if (putc(0, fp))  {
			serv_errno = SJ3_FileWriteError; goto error;
		}
	pos += idxlen;

	put4byte(tmp + DICTSEGPOS, pos);
	put4byte(tmp + DICTSEGLEN, seglen);
	put4byte(tmp + DICTSEGNUM, 1);
	put4byte(tmp + DICTSEGMAX, segnum);
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; goto error;
	}
	if (putc(0, fp)) {
			serv_errno = SJ3_FileWriteError; goto error;
	}
	for (i = 1 ; i < seglen ; i++)
		if (putc(0xff, fp) != 0xff) {
			serv_errno = SJ3_FileWriteError; goto error;
		}

	if (fputfile(fp, 0L, sizeof(tmp), tmp) == SJ3_NormalEnd)
		ret = SJ3_NormalEnd;
error:
	fclose(fp);
	if (ret != SJ3_NormalEnd) unlink(path);
	return ret;
}

int
makestdy(char *path, int stynum, int clstep, int cllen)
{
	FILE	*fp;
	u_char	tmp[HEADERLENGTH + COMMENTLENGTH];
	long	pos = HEADERLENGTH + COMMENTLENGTH;
	int	ret = ERROR;
	int	i, j;

	if (stynum == 0) stynum = MINSTYNUM;
	if (clstep == 0) clstep = MINCLSTEP;
	if (cllen == 0) cllen = MINCLLEN;

	if (stynum < MINSTYNUM || clstep < MINCLSTEP || cllen < MINCLLEN) {
		serv_errno = SJ3_IllegalParameter; return ret;
	}
	if (stynum > MAXSTYNUM || clstep > MAXCLSTEP || cllen > MAXCLLEN) {
		serv_errno = SJ3_IllegalParameter; return ret;
	}

	if (!(fp = fopen(path, "w"))) {
		serv_errno = SJ3_CannotCreateFile; return ret;
	}
	memset(tmp, '\0', sizeof(tmp));
	put4byte(tmp + VERSIONPOS, STDYVERSION);

	j = 256 * sizeof(u_short);
	put4byte(tmp + STDYCLIDXPOS, pos);
	put4byte(tmp + STDYCLIDXLEN, j);
	put4byte(tmp + STDYCLIDXSTEP, clstep);
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; goto error;
	}
	for (i = 0 ; i < j ; i++)
		if (putc(0, fp)) {
			serv_errno = SJ3_FileWriteError; goto error;
		}
	pos += j;

	put4byte(tmp + STDYCLSEGPOS, pos);
	put4byte(tmp + STDYCLSEGLEN, cllen);
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; goto error;
	}
	for (i = 0 ; i < cllen ; i++)
		if (putc(0, fp)) {
			serv_errno = SJ3_FileWriteError; goto error;
		}
	pos += cllen;

	j = sizeof(STDYIN) * stynum;
	put4byte(tmp + STDYNORMPOS, pos);
	put4byte(tmp + STDYNORMLEN, j);
	put4byte(tmp + STDYNORMNUM, stynum);
	put4byte(tmp + STDYNORMCNT, 0);
	if (fseek(fp, pos, 0) == ERROR) {
		serv_errno = SJ3_FileSeekError; goto error;
	}
	for (i = 0 ; i < j ; i++) {
		if (putc(0, fp)) {
			serv_errno = SJ3_FileWriteError; goto error;
		}
	}
	pos += j;

	if (fputfile(fp, 0L, sizeof(tmp), tmp) == SJ3_NormalEnd)
		ret = SJ3_NormalEnd;
error:
	fclose(fp);
	if (ret != SJ3_NormalEnd) unlink(path);
	return ret;
}

void
sj_closeall()
{
	while (dictlink) closedict(dictlink);
	while (stdylink) closestdy(stdylink);
}

int
set_dictpass(DictFile *dp, char *pass)
{
	set_passwd(dp -> buffer, pass);
	return putfile(dp->fd, 0, HEADERLENGTH + COMMENTLENGTH, dp->buffer);
}

int
set_stdypass(char *pass)
{
	StdyFile	*sp;

	sp = (StdyFile *)stdy_base;
	set_passwd(sp -> header, pass);
	return putfile(sp->fd, 0, HEADERLENGTH + COMMENTLENGTH, sp->header);
}

static
void
set_comment(u_char *buf, char *comment)
{
	strlcpy(buf + HEADERLENGTH, comment, COMMENTLENGTH);
}

int
set_dictcmnt(DictFile *dp, char *cmnt)
{
	set_comment(dp -> buffer, cmnt);
	return putfile(dp->fd, 0, HEADERLENGTH + COMMENTLENGTH, dp->buffer);
}

int
set_stdycmnt(char *cmnt)
{
	StdyFile	*sp;

	sp = (StdyFile *)stdy_base;
	set_comment(sp -> header, cmnt);
	return putfile(sp->fd, 0, HEADERLENGTH + COMMENTLENGTH, sp->header);
}

int
get_stdysize(int *stynum, int *clstep, int *cllen)
{
	*stynum = STUDYMAX;
	*clstep = CLSTUDYSTEP;
	*cllen  = CLSTUDYLEN;
	return 1;
}