/*
* (c) The GRASP/AQUA Project, Glasgow University, 1994-1998
*
- * $Id: openFile.c,v 1.1 1998/04/10 10:54:42 simonm Exp $
+ * $Id: openFile.c,v 1.18 2000/05/15 09:20:11 simonmar Exp $
*
* openFile Runtime Support
*/
+/* We use lstat, which is sadly not POSIX */
+#define NON_POSIX_SOURCE
+
#include "Rts.h"
#include "stgio.h"
#include <fcntl.h>
#endif
-StgAddr
-openFile(StgByteArray file, StgByteArray how)
+#if defined(mingw32_TARGET_OS) && !defined(O_NOCTTY)
+#define O_NOCTTY 0
+#endif
+
+IOFileObject*
+openStdFile(StgInt fd, StgInt rd)
+{
+ IOFileObject* fo;
+ long fd_flags;
+
+ if ((fo = malloc(sizeof(IOFileObject))) == NULL)
+ return NULL;
+ fo->fd = fd;
+ fo->buf = NULL;
+ fo->bufWPtr = 0;
+ fo->bufRPtr = 0;
+ fo->flags = FILEOBJ_STD | ( rd ? FILEOBJ_READ : FILEOBJ_WRITE);
+ fo->connectedTo = NULL;
+
+#if !defined(_WIN32) || defined(__CYGWIN__) || defined(__CYGWIN32__)
+ /* Set the non-blocking flag on this file descriptor.
+ *
+ * Don't do it for stdout and stderr: some shells (actually most)
+ * don't reset the nonblocking flag after running a program, and
+ * this causes all sorts of problems. --SDM (12/99)
+ *
+ * MS Win32 CRT doesn't support fcntl() -- the workaround is to
+ * start using 'completion ports', but I'm punting on implementing
+ * support for using those.
+ */
+ if (fd != 1 && fd != 2) {
+ fd_flags = fcntl(fd, F_GETFL);
+ fcntl(fd, F_SETFL, fd_flags | O_NONBLOCK);
+ }
+#endif
+
+ return fo;
+}
+
+#define OPENFILE_APPEND 0
+#define OPENFILE_WRITE 1
+#define OPENFILE_READ_ONLY 2
+#define OPENFILE_READ_WRITE 3
+
+IOFileObject*
+openFile(StgByteArray file, StgInt how, StgInt binary)
{
- FILE *fp;
int fd;
int oflags;
- int exclusive;
+ int for_writing;
int created = 0;
struct stat sb;
+ IOFileObject* fo;
+ int flags = 0;
+
+#if defined(_WIN32) && !(defined(__CYGWIN__) || defined(__CYGWIN32__))
+#define O_NONBLOCK 0
+#endif
/*
* Since we aren't supposed to succeed when we're opening for writing and
- * there's another writer, we can't just do an fopen() for "w" mode.
+ * there's another writer, we can't just do an open() with O_WRONLY.
*/
- switch (how[0]) {
- case 'a':
- oflags = O_WRONLY | O_NOCTTY | O_APPEND;
- exclusive = 1;
+ switch (how) {
+ case OPENFILE_APPEND:
+ oflags = O_NONBLOCK | O_WRONLY | O_NOCTTY | O_APPEND;
+ for_writing = 1;
+ flags |= FILEOBJ_WRITE;
break;
- case 'w':
- oflags = O_WRONLY | O_NOCTTY;
- exclusive = 1;
+ case OPENFILE_WRITE:
+ oflags = O_NONBLOCK | O_WRONLY | O_NOCTTY;
+ flags |= FILEOBJ_WRITE;
+ for_writing = 1;
break;
- case 'r':
- oflags = how[1] == '+' ? O_RDWR | O_NOCTTY : O_RDONLY | O_NOCTTY;
- exclusive = 0;
+ case OPENFILE_READ_ONLY:
+ oflags = O_NONBLOCK | O_RDONLY | O_NOCTTY;
+ flags |= FILEOBJ_READ;
+ for_writing = 0;
+ break;
+ case OPENFILE_READ_WRITE:
+ oflags = O_NONBLOCK | O_RDWR | O_NOCTTY;
+ flags |= FILEOBJ_READ | FILEOBJ_WRITE;
+ for_writing = 1;
break;
default:
- fprintf(stderr, "openFile: unknown mode `%s'\n", how);
+ fprintf(stderr, "openFile: unknown mode `%d'\n", how);
exit(EXIT_FAILURE);
}
+#if HAVE_O_BINARY
+ if (binary) {
+ oflags |= O_BINARY;
+ flags |= FILEOBJ_BINARY;
+ }
+#endif
+
/* First try to open without creating */
while ((fd = open(file, oflags, 0666)) < 0) {
if (errno == ENOENT) {
- if (how[0] == 'r' && how[1] == '\0') {
+ if ( how == OPENFILE_READ_ONLY ) {
/* For ReadMode, just bail out now */
ghc_errtype = ERR_NOSUCHTHING;
ghc_errstr = "file does not exist";
return NULL;
} else {
/* If it is a dangling symlink, break off now, too. */
+#ifndef mingw32_TARGET_OS
struct stat st;
if ( lstat(file,&st) == 0) {
ghc_errtype = ERR_NOSUCHTHING;
ghc_errstr = "dangling symlink";
return NULL;
}
+#endif
}
/* Now try to create it */
while ((fd = open(file, oflags | O_CREAT | O_EXCL, 0666)) < 0) {
}
/* Use our own personal locking */
- if (lockFile(fd, exclusive) < 0) {
+ if (lockFile(fd, for_writing, 1/*enforce single-writer, if needs be.*/) < 0) {
cvtErrno();
switch (ghc_errno) {
default:
* ftruncate() is non-POSIX, so we truncate with a second open, which may fail.
*/
- if (how[0] == 'w') {
- int fd2;
+ if ( how == OPENFILE_WRITE ) {
+ int fd2, oflags2;
- oflags |= O_TRUNC;
- while ((fd2 = open(file, oflags, 0666)) < 0) {
+ oflags2 = oflags | O_TRUNC;
+ while ((fd2 = open(file, oflags2, 0666)) < 0) {
if (errno != EINTR) {
cvtErrno();
if (created)
}
close(fd2);
}
- errno = 0; /* Just in case fdopen() is lame */
- while ((fp = fdopen(fd, how)) == NULL) {
- if (errno != EINTR) {
- cvtErrno();
- if (created)
- (void) unlink(file);
- (void) close(fd);
- return NULL;
+
+ /* Allocate a IOFileObject to hold the information
+ we need to record per-handle for the various C stubs.
+ This chunk of memory is wrapped up inside a foreign object,
+ so it will be finalised and freed properly when we're
+ through with the handle.
+ */
+ if ((fo = malloc(sizeof(IOFileObject))) == NULL)
+ return NULL;
+
+ fo->fd = fd;
+ fo->buf = NULL;
+ fo->bufWPtr = 0;
+ fo->bufRPtr = 0;
+ fo->flags = flags;
+ fo->connectedTo = NULL;
+ return fo;
+}
+
+/* `Lock' file descriptor and return file object. */
+IOFileObject*
+openFd(StgInt fd, StgInt oflags, StgInt flags)
+{
+ int for_writing;
+ IOFileObject* fo;
+
+ for_writing = ( ((oflags & O_WRONLY) || (oflags & O_RDWR)) ? 1 : 0);
+
+ if (lockFile(fd, for_writing, 1/* enforce single-writer */ ) < 0) {
+ cvtErrno();
+ switch (ghc_errno) {
+ default:
+ stdErrno();
+ break;
+ case GHC_EACCES:
+ case GHC_EAGAIN:
+ ghc_errtype = ERR_RESOURCEBUSY;
+ ghc_errstr = "file is locked";
+ break;
}
+ return NULL;
}
- return (StgAddr) fp;
+ /* See openFileObject() comment */
+ if ((fo = malloc(sizeof(IOFileObject))) == NULL)
+ return NULL;
+ fo->fd = fd;
+ fo->buf = NULL;
+ fo->bufWPtr = 0;
+ fo->bufRPtr = 0;
+ fo->flags = flags | ( oflags & O_RDONLY ? FILEOBJ_READ
+ : oflags & O_RDWR ? FILEOBJ_READ
+ : 0)
+ | ( oflags & O_WRONLY ? FILEOBJ_WRITE
+ : oflags & O_RDWR ? FILEOBJ_WRITE
+ : 0);
+ fo->connectedTo = NULL;
+ return fo;
}