mirror of
https://git.FreeBSD.org/src.git
synced 2024-12-16 10:20:30 +00:00
131 lines
2.4 KiB
C
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;
|
|
}
|