freebsd-nq/contrib/libf2c/libI77/endfile.c
2003-07-11 03:42:19 +00:00

131 lines
2.4 KiB
C

#include "config.h"
#include "f2c.h"
#include "fio.h"
#include <sys/types.h>
#include <unistd.h>
#undef abs
#undef min
#undef max
#include <stdlib.h>
#include <string.h>
extern char *f__r_mode[], *f__w_mode[];
integer
f_end (alist * a)
{
unit *b;
FILE *tf;
if (f__init & 2)
f__fatal (131, "I/O recursion");
if (a->aunit >= MXUNIT || a->aunit < 0)
err (a->aerr, 101, "endfile");
b = &f__units[a->aunit];
if (b->ufd == NULL)
{
char nbuf[10];
sprintf (nbuf, "fort.%ld", (long) a->aunit);
if ((tf = fopen (nbuf, f__w_mode[0])))
fclose (tf);
return (0);
}
b->uend = 1;
return (b->useek ? t_runc (a) : 0);
}
#ifndef HAVE_FTRUNCATE
static int
copy (FILE * from, register long len, FILE * to)
{
int len1;
char buf[BUFSIZ];
while (fread (buf, len1 = len > BUFSIZ ? BUFSIZ : (int) len, 1, from))
{
if (!fwrite (buf, len1, 1, to))
return 1;
if ((len -= len1) <= 0)
break;
}
return 0;
}
#endif /* !defined(HAVE_FTRUNCATE) */
int
t_runc (alist * a)
{
off_t loc, len;
unit *b;
int rc;
FILE *bf;
#ifndef HAVE_FTRUNCATE
FILE *tf;
#endif /* !defined(HAVE_FTRUNCATE) */
b = &f__units[a->aunit];
if (b->url)
return (0); /*don't truncate direct files */
loc = FTELL (bf = b->ufd);
FSEEK (bf, 0, SEEK_END);
len = FTELL (bf);
if (loc >= len || b->useek == 0 || b->ufnm == NULL)
return (0);
#ifndef HAVE_FTRUNCATE
rc = 0;
fclose (b->ufd);
if (!loc)
{
if (!(bf = fopen (b->ufnm, f__w_mode[b->ufmt])))
rc = 1;
if (b->uwrt)
b->uwrt = 1;
goto done;
}
if (!(bf = fopen (b->ufnm, f__r_mode[0])) || !(tf = tmpfile ()))
{
#ifdef NON_UNIX_STDIO
bad:
#endif
rc = 1;
goto done;
}
if (copy (bf, loc, tf))
{
bad1:
rc = 1;
goto done1;
}
if (!(bf = freopen (b->ufnm, f__w_mode[0], bf)))
goto bad1;
FSEEK (tf, 0, SEEK_SET);
if (copy (tf, loc, bf))
goto bad1;
b->uwrt = 1;
b->urw = 2;
#ifdef NON_UNIX_STDIO
if (b->ufmt)
{
fclose (bf);
if (!(bf = fopen (b->ufnm, f__w_mode[3])))
goto bad;
FSEEK (bf, 0, SEEK_END);
b->urw = 3;
}
#endif
done1:
fclose (tf);
done:
f__cf = b->ufd = bf;
#else /* !defined(HAVE_FTRUNCATE) */
fflush (b->ufd);
rc = ftruncate (fileno (b->ufd), loc);
FSEEK (bf, loc, SEEK_SET);
#endif /* !defined(HAVE_FTRUNCATE) */
if (rc)
err (a->aerr, 111, "endfile");
return 0;
}