commit da132b31db47c1c6002a4e43f9bb874286968366
parent 36dca2a4d1ae37768409ef355b323dafd4386488
Author: Michael Savage <mikejsavage@gmail.com>
Date: Thu, 6 Sep 2018 00:07:58 +0300
Initial Windows support!
Diffstat:
17 files changed, 1714 insertions(+), 18 deletions(-)
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,10 @@
-mudgangster
build
release
gen.mk
+
+mudgangster
+*.exe
+*.exp
+*.ilk
+*.lib
+*.pdb
diff --git a/libs/lfs.lua b/libs/lfs.lua
@@ -0,0 +1,3 @@
+lib( "lfs", { "libs/lfs/lfs" } )
+-- obj_replace_cxxflags( "libs/lpeg/%", "-c -O2 -x c" )
+obj_cxxflags( "libs/lfs/%", "/c /TC /I libs/lua" )
diff --git a/libs/lfs/lfs.cc b/libs/lfs/lfs.cc
@@ -0,0 +1,941 @@
+/*
+** LuaFileSystem
+** Copyright Kepler Project 2003 - 2017 (http://keplerproject.github.io/luafilesystem)
+**
+** File system manipulation library.
+** This library offers these functions:
+** lfs.attributes (filepath [, attributename | attributetable])
+** lfs.chdir (path)
+** lfs.currentdir ()
+** lfs.dir (path)
+** lfs.link (old, new[, symlink])
+** lfs.lock (fh, mode)
+** lfs.lock_dir (path)
+** lfs.mkdir (path)
+** lfs.rmdir (path)
+** lfs.setmode (filepath, mode)
+** lfs.symlinkattributes (filepath [, attributename])
+** lfs.touch (filepath [, atime [, mtime]])
+** lfs.unlock (fh)
+*/
+
+#ifndef LFS_DO_NOT_USE_LARGE_FILE
+#ifndef _WIN32
+#ifndef _AIX
+#define _FILE_OFFSET_BITS 64 /* Linux, Solaris and HP-UX */
+#else
+#define _LARGE_FILES 1 /* AIX */
+#endif
+#endif
+#endif
+
+#ifndef LFS_DO_NOT_USE_LARGE_FILE
+#define _LARGEFILE64_SOURCE
+#endif
+
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <time.h>
+#include <sys/stat.h>
+
+#ifdef _WIN32
+ #include <direct.h>
+ #include <windows.h>
+ #include <io.h>
+ #include <sys/locking.h>
+ #ifdef __BORLANDC__
+ #include <utime.h>
+ #else
+ #include <sys/utime.h>
+ #endif
+ #include <fcntl.h>
+ /* MAX_PATH seems to be 260. Seems kind of small. Is there a better one? */
+ #define LFS_MAXPATHLEN MAX_PATH
+#else
+ #include <unistd.h>
+ #include <dirent.h>
+ #include <fcntl.h>
+ #include <sys/types.h>
+ #include <utime.h>
+ #include <sys/param.h> /* for MAXPATHLEN */
+ #define LFS_MAXPATHLEN MAXPATHLEN
+#endif
+
+#include <lua.h>
+#include <lauxlib.h>
+#include <lualib.h>
+
+#include "lfs.h"
+
+#define LFS_VERSION "1.7.0"
+#define LFS_LIBNAME "lfs"
+
+#if LUA_VERSION_NUM >= 503 /* Lua 5.3 */
+
+#ifndef luaL_optlong
+#define luaL_optlong luaL_optinteger
+#endif
+
+#endif
+
+#if LUA_VERSION_NUM >= 502
+# define new_lib(L, l) (luaL_newlib(L, l))
+#else
+# define new_lib(L, l) (lua_newtable(L), luaL_register(L, NULL, l))
+#endif
+
+/* Define 'strerror' for systems that do not implement it */
+#ifdef NO_STRERROR
+#define strerror(_) "System unable to describe the error"
+#endif
+
+#define DIR_METATABLE "directory metatable"
+typedef struct dir_data {
+ int closed;
+#ifdef _WIN32
+ intptr_t hFile;
+ char pattern[MAX_PATH+1];
+#else
+ DIR *dir;
+#endif
+} dir_data;
+
+#define LOCK_METATABLE "lock metatable"
+
+#ifdef _WIN32
+ #ifdef __BORLANDC__
+ #define lfs_setmode(file, m) (setmode(_fileno(file), m))
+ #define STAT_STRUCT struct stati64
+ #else
+ #define lfs_setmode(file, m) (_setmode(_fileno(file), m))
+ #define STAT_STRUCT struct _stati64
+ #endif
+#define STAT_FUNC _stati64
+#define LSTAT_FUNC STAT_FUNC
+#else
+#define _O_TEXT 0
+#define _O_BINARY 0
+#define lfs_setmode(file, m) ((void)file, (void)m, 0)
+#define STAT_STRUCT struct stat
+#define STAT_FUNC stat
+#define LSTAT_FUNC lstat
+#endif
+
+#ifdef _WIN32
+ #define lfs_mkdir _mkdir
+#else
+ #define lfs_mkdir(path) (mkdir((path), \
+ S_IRUSR | S_IWUSR | S_IXUSR | S_IRGRP | S_IWGRP | S_IXGRP | S_IROTH | S_IXOTH))
+#endif
+
+/*
+** Utility functions
+*/
+static int pusherror(lua_State *L, const char *info)
+{
+ lua_pushnil(L);
+ if (info==NULL)
+ lua_pushstring(L, strerror(errno));
+ else
+ lua_pushfstring(L, "%s: %s", info, strerror(errno));
+ lua_pushinteger(L, errno);
+ return 3;
+}
+
+static int pushresult(lua_State *L, int res, const char *info) {
+ if (res == -1) {
+ return pusherror(L, info);
+ } else {
+ lua_pushboolean(L, 1);
+ return 1;
+ }
+}
+
+
+/*
+** This function changes the working (current) directory
+*/
+static int change_dir (lua_State *L) {
+ const char *path = luaL_checkstring(L, 1);
+ if (chdir(path)) {
+ lua_pushnil (L);
+ lua_pushfstring (L,"Unable to change working directory to '%s'\n%s\n",
+ path, chdir_error);
+ return 2;
+ } else {
+ lua_pushboolean (L, 1);
+ return 1;
+ }
+}
+
+/*
+** This function returns the current directory
+** If unable to get the current directory, it returns nil
+** and a string describing the error
+*/
+static int get_dir (lua_State *L) {
+#ifdef NO_GETCWD
+ lua_pushnil(L);
+ lua_pushstring(L, "Function 'getcwd' not provided by system");
+ return 2;
+#else
+ char *path = NULL;
+ /* Passing (NULL, 0) is not guaranteed to work. Use a temp buffer and size instead. */
+ size_t size = LFS_MAXPATHLEN; /* initial buffer size */
+ int result;
+ while (1) {
+ path = realloc(path, size);
+ if (!path) /* failed to allocate */
+ return pusherror(L, "get_dir realloc() failed");
+ if (getcwd(path, size) != NULL) {
+ /* success, push the path to the Lua stack */
+ lua_pushstring(L, path);
+ result = 1;
+ break;
+ }
+ if (errno != ERANGE) { /* unexpected error */
+ result = pusherror(L, "get_dir getcwd() failed");
+ break;
+ }
+ /* ERANGE = insufficient buffer capacity, double size and retry */
+ size *= 2;
+ }
+ free(path);
+ return result;
+#endif
+}
+
+/*
+** Check if the given element on the stack is a file and returns it.
+*/
+static FILE *check_file (lua_State *L, int idx, const char *funcname) {
+#if LUA_VERSION_NUM == 501
+ FILE **fh = (FILE **)luaL_checkudata (L, idx, "FILE*");
+ if (*fh == NULL) {
+ luaL_error (L, "%s: closed file", funcname);
+ return 0;
+ } else
+ return *fh;
+#elif LUA_VERSION_NUM >= 502 && LUA_VERSION_NUM <= 503
+ luaL_Stream *fh = (luaL_Stream *)luaL_checkudata (L, idx, "FILE*");
+ if (fh->closef == 0 || fh->f == NULL) {
+ luaL_error (L, "%s: closed file", funcname);
+ return 0;
+ } else
+ return fh->f;
+#else
+#error unsupported Lua version
+#endif
+}
+
+
+/*
+**
+*/
+static int _file_lock (lua_State *L, FILE *fh, const char *mode, const long start, long len, const char *funcname) {
+ int code;
+#ifdef _WIN32
+ /* lkmode valid values are:
+ LK_LOCK Locks the specified bytes. If the bytes cannot be locked, the program immediately tries again after 1 second. If, after 10 attempts, the bytes cannot be locked, the constant returns an error.
+ LK_NBLCK Locks the specified bytes. If the bytes cannot be locked, the constant returns an error.
+ LK_NBRLCK Same as _LK_NBLCK.
+ LK_RLCK Same as _LK_LOCK.
+ LK_UNLCK Unlocks the specified bytes, which must have been previously locked.
+
+ Regions should be locked only briefly and should be unlocked before closing a file or exiting the program.
+
+ http://msdn.microsoft.com/library/default.asp?url=/library/en-us/vclib/html/_crt__locking.asp
+ */
+ int lkmode;
+ switch (*mode) {
+ case 'r': lkmode = LK_NBLCK; break;
+ case 'w': lkmode = LK_NBLCK; break;
+ case 'u': lkmode = LK_UNLCK; break;
+ default : return luaL_error (L, "%s: invalid mode", funcname);
+ }
+ if (!len) {
+ fseek (fh, 0L, SEEK_END);
+ len = ftell (fh);
+ }
+ fseek (fh, start, SEEK_SET);
+#ifdef __BORLANDC__
+ code = locking (fileno(fh), lkmode, len);
+#else
+ code = _locking (fileno(fh), lkmode, len);
+#endif
+#else
+ struct flock f;
+ switch (*mode) {
+ case 'w': f.l_type = F_WRLCK; break;
+ case 'r': f.l_type = F_RDLCK; break;
+ case 'u': f.l_type = F_UNLCK; break;
+ default : return luaL_error (L, "%s: invalid mode", funcname);
+ }
+ f.l_whence = SEEK_SET;
+ f.l_start = (off_t)start;
+ f.l_len = (off_t)len;
+ code = fcntl (fileno(fh), F_SETLK, &f);
+#endif
+ return (code != -1);
+}
+
+#ifdef _WIN32
+typedef struct lfs_Lock {
+ HANDLE fd;
+} lfs_Lock;
+static int lfs_lock_dir(lua_State *L) {
+ size_t pathl; HANDLE fd;
+ lfs_Lock *lock;
+ char *ln;
+ const char *lockfile = "/lockfile.lfs";
+ const char *path = luaL_checklstring(L, 1, &pathl);
+ ln = (char*)malloc(pathl + strlen(lockfile) + 1);
+ if(!ln) {
+ lua_pushnil(L); lua_pushstring(L, strerror(errno)); return 2;
+ }
+ strcpy(ln, path); strcat(ln, lockfile);
+ if((fd = CreateFile(ln, GENERIC_WRITE, 0, NULL, CREATE_NEW,
+ FILE_ATTRIBUTE_NORMAL | FILE_FLAG_DELETE_ON_CLOSE, NULL)) == INVALID_HANDLE_VALUE) {
+ int en = GetLastError();
+ free(ln); lua_pushnil(L);
+ if(en == ERROR_FILE_EXISTS || en == ERROR_SHARING_VIOLATION)
+ lua_pushstring(L, "File exists");
+ else
+ lua_pushstring(L, strerror(en));
+ return 2;
+ }
+ free(ln);
+ lock = (lfs_Lock*)lua_newuserdata(L, sizeof(lfs_Lock));
+ lock->fd = fd;
+ luaL_getmetatable (L, LOCK_METATABLE);
+ lua_setmetatable (L, -2);
+ return 1;
+}
+static int lfs_unlock_dir(lua_State *L) {
+ lfs_Lock *lock = (lfs_Lock *)luaL_checkudata(L, 1, LOCK_METATABLE);
+ if(lock->fd != INVALID_HANDLE_VALUE) {
+ CloseHandle(lock->fd);
+ lock->fd=INVALID_HANDLE_VALUE;
+ }
+ return 0;
+}
+#else
+typedef struct lfs_Lock {
+ char *ln;
+} lfs_Lock;
+static int lfs_lock_dir(lua_State *L) {
+ lfs_Lock *lock;
+ size_t pathl;
+ char *ln;
+ const char *lockfile = "/lockfile.lfs";
+ const char *path = luaL_checklstring(L, 1, &pathl);
+ lock = (lfs_Lock*)lua_newuserdata(L, sizeof(lfs_Lock));
+ ln = (char*)malloc(pathl + strlen(lockfile) + 1);
+ if(!ln) {
+ lua_pushnil(L); lua_pushstring(L, strerror(errno)); return 2;
+ }
+ strcpy(ln, path); strcat(ln, lockfile);
+ if(symlink("lock", ln) == -1) {
+ free(ln); lua_pushnil(L);
+ lua_pushstring(L, strerror(errno)); return 2;
+ }
+ lock->ln = ln;
+ luaL_getmetatable (L, LOCK_METATABLE);
+ lua_setmetatable (L, -2);
+ return 1;
+}
+static int lfs_unlock_dir(lua_State *L) {
+ lfs_Lock *lock = (lfs_Lock *)luaL_checkudata(L, 1, LOCK_METATABLE);
+ if(lock->ln) {
+ unlink(lock->ln);
+ free(lock->ln);
+ lock->ln = NULL;
+ }
+ return 0;
+}
+#endif
+
+static int lfs_g_setmode (lua_State *L, FILE *f, int arg) {
+ static const int mode[] = {_O_BINARY, _O_TEXT};
+ static const char *const modenames[] = {"binary", "text", NULL};
+ int op = luaL_checkoption(L, arg, NULL, modenames);
+ int res = lfs_setmode(f, mode[op]);
+ if (res != -1) {
+ int i;
+ lua_pushboolean(L, 1);
+ for (i = 0; modenames[i] != NULL; i++) {
+ if (mode[i] == res) {
+ lua_pushstring(L, modenames[i]);
+ return 2;
+ }
+ }
+ lua_pushnil(L);
+ return 2;
+ } else {
+ return pusherror(L, NULL);
+ }
+}
+
+static int lfs_f_setmode(lua_State *L) {
+ return lfs_g_setmode(L, check_file(L, 1, "setmode"), 2);
+}
+
+/*
+** Locks a file.
+** @param #1 File handle.
+** @param #2 String with lock mode ('w'rite, 'r'ead).
+** @param #3 Number with start position (optional).
+** @param #4 Number with length (optional).
+*/
+static int file_lock (lua_State *L) {
+ FILE *fh = check_file (L, 1, "lock");
+ const char *mode = luaL_checkstring (L, 2);
+ const long start = (long) luaL_optinteger (L, 3, 0);
+ long len = (long) luaL_optinteger (L, 4, 0);
+ if (_file_lock (L, fh, mode, start, len, "lock")) {
+ lua_pushboolean (L, 1);
+ return 1;
+ } else {
+ lua_pushnil (L);
+ lua_pushfstring (L, "%s", strerror(errno));
+ return 2;
+ }
+}
+
+
+/*
+** Unlocks a file.
+** @param #1 File handle.
+** @param #2 Number with start position (optional).
+** @param #3 Number with length (optional).
+*/
+static int file_unlock (lua_State *L) {
+ FILE *fh = check_file (L, 1, "unlock");
+ const long start = (long) luaL_optinteger (L, 2, 0);
+ long len = (long) luaL_optinteger (L, 3, 0);
+ if (_file_lock (L, fh, "u", start, len, "unlock")) {
+ lua_pushboolean (L, 1);
+ return 1;
+ } else {
+ lua_pushnil (L);
+ lua_pushfstring (L, "%s", strerror(errno));
+ return 2;
+ }
+}
+
+
+/*
+** Creates a link.
+** @param #1 Object to link to.
+** @param #2 Name of link.
+** @param #3 True if link is symbolic (optional).
+*/
+static int make_link (lua_State *L) {
+#ifndef _WIN32
+ const char *oldpath = luaL_checkstring(L, 1);
+ const char *newpath = luaL_checkstring(L, 2);
+ int res = (lua_toboolean(L,3) ? symlink : link)(oldpath, newpath);
+ if (res == -1) {
+ return pusherror(L, NULL);
+ } else {
+ lua_pushinteger(L, 0);
+ return 1;
+ }
+#else
+ errno = ENOSYS; /* = "Function not implemented" */
+ return pushresult(L, -1, "make_link is not supported on Windows");
+#endif
+}
+
+
+/*
+** Creates a directory.
+** @param #1 Directory path.
+*/
+static int make_dir (lua_State *L) {
+ const char *path = luaL_checkstring(L, 1);
+ return pushresult(L, lfs_mkdir(path), NULL);
+}
+
+
+/*
+** Removes a directory.
+** @param #1 Directory path.
+*/
+static int remove_dir (lua_State *L) {
+ const char *path = luaL_checkstring(L, 1);
+ return pushresult(L, rmdir(path), NULL);
+}
+
+
+/*
+** Directory iterator
+*/
+static int dir_iter (lua_State *L) {
+#ifdef _WIN32
+ struct _finddata_t c_file;
+#else
+ struct dirent *entry;
+#endif
+ dir_data *d = (dir_data *)luaL_checkudata (L, 1, DIR_METATABLE);
+ luaL_argcheck (L, d->closed == 0, 1, "closed directory");
+#ifdef _WIN32
+ if (d->hFile == 0L) { /* first entry */
+ if ((d->hFile = _findfirst (d->pattern, &c_file)) == -1L) {
+ lua_pushnil (L);
+ lua_pushstring (L, strerror (errno));
+ d->closed = 1;
+ return 2;
+ } else {
+ lua_pushstring (L, c_file.name);
+ return 1;
+ }
+ } else { /* next entry */
+ if (_findnext (d->hFile, &c_file) == -1L) {
+ /* no more entries => close directory */
+ _findclose (d->hFile);
+ d->closed = 1;
+ return 0;
+ } else {
+ lua_pushstring (L, c_file.name);
+ return 1;
+ }
+ }
+#else
+ if ((entry = readdir (d->dir)) != NULL) {
+ lua_pushstring (L, entry->d_name);
+ return 1;
+ } else {
+ /* no more entries => close directory */
+ closedir (d->dir);
+ d->closed = 1;
+ return 0;
+ }
+#endif
+}
+
+
+/*
+** Closes directory iterators
+*/
+static int dir_close (lua_State *L) {
+ dir_data *d = (dir_data *)lua_touserdata (L, 1);
+#ifdef _WIN32
+ if (!d->closed && d->hFile) {
+ _findclose (d->hFile);
+ }
+#else
+ if (!d->closed && d->dir) {
+ closedir (d->dir);
+ }
+#endif
+ d->closed = 1;
+ return 0;
+}
+
+
+/*
+** Factory of directory iterators
+*/
+static int dir_iter_factory (lua_State *L) {
+ const char *path = luaL_checkstring (L, 1);
+ dir_data *d;
+ lua_pushcfunction (L, dir_iter);
+ d = (dir_data *) lua_newuserdata (L, sizeof(dir_data));
+ luaL_getmetatable (L, DIR_METATABLE);
+ lua_setmetatable (L, -2);
+ d->closed = 0;
+#ifdef _WIN32
+ d->hFile = 0L;
+ if (strlen(path) > MAX_PATH-2)
+ luaL_error (L, "path too long: %s", path);
+ else
+ sprintf (d->pattern, "%s/*", path);
+#else
+ d->dir = opendir (path);
+ if (d->dir == NULL)
+ luaL_error (L, "cannot open %s: %s", path, strerror (errno));
+#endif
+ return 2;
+}
+
+
+/*
+** Creates directory metatable.
+*/
+static int dir_create_meta (lua_State *L) {
+ luaL_newmetatable (L, DIR_METATABLE);
+
+ /* Method table */
+ lua_newtable(L);
+ lua_pushcfunction (L, dir_iter);
+ lua_setfield(L, -2, "next");
+ lua_pushcfunction (L, dir_close);
+ lua_setfield(L, -2, "close");
+
+ /* Metamethods */
+ lua_setfield(L, -2, "__index");
+ lua_pushcfunction (L, dir_close);
+ lua_setfield (L, -2, "__gc");
+ return 1;
+}
+
+
+/*
+** Creates lock metatable.
+*/
+static int lock_create_meta (lua_State *L) {
+ luaL_newmetatable (L, LOCK_METATABLE);
+
+ /* Method table */
+ lua_newtable(L);
+ lua_pushcfunction(L, lfs_unlock_dir);
+ lua_setfield(L, -2, "free");
+
+ /* Metamethods */
+ lua_setfield(L, -2, "__index");
+ lua_pushcfunction(L, lfs_unlock_dir);
+ lua_setfield(L, -2, "__gc");
+ return 1;
+}
+
+
+#ifdef _WIN32
+ #ifndef S_ISDIR
+ #define S_ISDIR(mode) (mode&_S_IFDIR)
+ #endif
+ #ifndef S_ISREG
+ #define S_ISREG(mode) (mode&_S_IFREG)
+ #endif
+ #ifndef S_ISLNK
+ #define S_ISLNK(mode) (0)
+ #endif
+ #ifndef S_ISSOCK
+ #define S_ISSOCK(mode) (0)
+ #endif
+ #ifndef S_ISFIFO
+ #define S_ISFIFO(mode) (0)
+ #endif
+ #ifndef S_ISCHR
+ #define S_ISCHR(mode) (mode&_S_IFCHR)
+ #endif
+ #ifndef S_ISBLK
+ #define S_ISBLK(mode) (0)
+ #endif
+#endif
+/*
+** Convert the inode protection mode to a string.
+*/
+#ifdef _WIN32
+static const char *mode2string (unsigned short mode) {
+#else
+static const char *mode2string (mode_t mode) {
+#endif
+ if ( S_ISREG(mode) )
+ return "file";
+ else if ( S_ISDIR(mode) )
+ return "directory";
+ else if ( S_ISLNK(mode) )
+ return "link";
+ else if ( S_ISSOCK(mode) )
+ return "socket";
+ else if ( S_ISFIFO(mode) )
+ return "named pipe";
+ else if ( S_ISCHR(mode) )
+ return "char device";
+ else if ( S_ISBLK(mode) )
+ return "block device";
+ else
+ return "other";
+}
+
+
+/*
+** Set access time and modification values for a file.
+** @param #1 File path.
+** @param #2 Access time in seconds, current time is used if missing.
+** @param #3 Modification time in seconds, access time is used if missing.
+*/
+static int file_utime (lua_State *L) {
+ const char *file = luaL_checkstring(L, 1);
+ struct utimbuf utb, *buf;
+
+ if (lua_gettop (L) == 1) /* set to current date/time */
+ buf = NULL;
+ else {
+ utb.actime = (time_t) luaL_optnumber(L, 2, 0);
+ utb.modtime = (time_t) luaL_optinteger(L, 3, utb.actime);
+ buf = &utb;
+ }
+
+ return pushresult(L, utime(file, buf), NULL);
+}
+
+
+/* inode protection mode */
+static void push_st_mode (lua_State *L, STAT_STRUCT *info) {
+ lua_pushstring (L, mode2string (info->st_mode));
+}
+/* device inode resides on */
+static void push_st_dev (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer) info->st_dev);
+}
+/* inode's number */
+static void push_st_ino (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer) info->st_ino);
+}
+/* number of hard links to the file */
+static void push_st_nlink (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer)info->st_nlink);
+}
+/* user-id of owner */
+static void push_st_uid (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer)info->st_uid);
+}
+/* group-id of owner */
+static void push_st_gid (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer)info->st_gid);
+}
+/* device type, for special file inode */
+static void push_st_rdev (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer) info->st_rdev);
+}
+/* time of last access */
+static void push_st_atime (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer) info->st_atime);
+}
+/* time of last data modification */
+static void push_st_mtime (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer) info->st_mtime);
+}
+/* time of last file status change */
+static void push_st_ctime (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer) info->st_ctime);
+}
+/* file size, in bytes */
+static void push_st_size (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer)info->st_size);
+}
+#ifndef _WIN32
+/* blocks allocated for file */
+static void push_st_blocks (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer)info->st_blocks);
+}
+/* optimal file system I/O blocksize */
+static void push_st_blksize (lua_State *L, STAT_STRUCT *info) {
+ lua_pushinteger (L, (lua_Integer)info->st_blksize);
+}
+#endif
+
+ /*
+** Convert the inode protection mode to a permission list.
+*/
+
+#ifdef _WIN32
+static const char *perm2string (unsigned short mode) {
+ static char perms[10] = "---------";
+ int i;
+ for (i=0;i<9;i++) perms[i]='-';
+ if (mode & _S_IREAD)
+ { perms[0] = 'r'; perms[3] = 'r'; perms[6] = 'r'; }
+ if (mode & _S_IWRITE)
+ { perms[1] = 'w'; perms[4] = 'w'; perms[7] = 'w'; }
+ if (mode & _S_IEXEC)
+ { perms[2] = 'x'; perms[5] = 'x'; perms[8] = 'x'; }
+ return perms;
+}
+#else
+static const char *perm2string (mode_t mode) {
+ static char perms[10] = "---------";
+ int i;
+ for (i=0;i<9;i++) perms[i]='-';
+ if (mode & S_IRUSR) perms[0] = 'r';
+ if (mode & S_IWUSR) perms[1] = 'w';
+ if (mode & S_IXUSR) perms[2] = 'x';
+ if (mode & S_IRGRP) perms[3] = 'r';
+ if (mode & S_IWGRP) perms[4] = 'w';
+ if (mode & S_IXGRP) perms[5] = 'x';
+ if (mode & S_IROTH) perms[6] = 'r';
+ if (mode & S_IWOTH) perms[7] = 'w';
+ if (mode & S_IXOTH) perms[8] = 'x';
+ return perms;
+}
+#endif
+
+/* permssions string */
+static void push_st_perm (lua_State *L, STAT_STRUCT *info) {
+ lua_pushstring (L, perm2string (info->st_mode));
+}
+
+typedef void (*_push_function) (lua_State *L, STAT_STRUCT *info);
+
+struct _stat_members {
+ const char *name;
+ _push_function push;
+};
+
+struct _stat_members members[] = {
+ { "mode", push_st_mode },
+ { "dev", push_st_dev },
+ { "ino", push_st_ino },
+ { "nlink", push_st_nlink },
+ { "uid", push_st_uid },
+ { "gid", push_st_gid },
+ { "rdev", push_st_rdev },
+ { "access", push_st_atime },
+ { "modification", push_st_mtime },
+ { "change", push_st_ctime },
+ { "size", push_st_size },
+ { "permissions", push_st_perm },
+#ifndef _WIN32
+ { "blocks", push_st_blocks },
+ { "blksize", push_st_blksize },
+#endif
+ { NULL, NULL }
+};
+
+/*
+** Get file or symbolic link information
+*/
+static int _file_info_ (lua_State *L, int (*st)(const char*, STAT_STRUCT*)) {
+ STAT_STRUCT info;
+ const char *file = luaL_checkstring (L, 1);
+ int i;
+
+ if (st(file, &info)) {
+ lua_pushnil(L);
+ lua_pushfstring(L, "cannot obtain information from file '%s': %s", file, strerror(errno));
+ lua_pushinteger(L, errno);
+ return 3;
+ }
+ if (lua_isstring (L, 2)) {
+ const char *member = lua_tostring (L, 2);
+ for (i = 0; members[i].name; i++) {
+ if (strcmp(members[i].name, member) == 0) {
+ /* push member value and return */
+ members[i].push (L, &info);
+ return 1;
+ }
+ }
+ /* member not found */
+ return luaL_error(L, "invalid attribute name '%s'", member);
+ }
+ /* creates a table if none is given, removes extra arguments */
+ lua_settop(L, 2);
+ if (!lua_istable (L, 2)) {
+ lua_newtable (L);
+ }
+ /* stores all members in table on top of the stack */
+ for (i = 0; members[i].name; i++) {
+ lua_pushstring (L, members[i].name);
+ members[i].push (L, &info);
+ lua_rawset (L, -3);
+ }
+ return 1;
+}
+
+
+/*
+** Get file information using stat.
+*/
+static int file_info (lua_State *L) {
+ return _file_info_ (L, STAT_FUNC);
+}
+
+
+/*
+** Push the symlink target to the top of the stack.
+** Assumes the file name is at position 1 of the stack.
+** Returns 1 if successful (with the target on top of the stack),
+** 0 on failure (with stack unchanged, and errno set).
+*/
+static int push_link_target(lua_State *L) {
+#ifdef _WIN32
+ errno = ENOSYS;
+ return 0;
+#else
+ const char *file = luaL_checkstring(L, 1);
+ char *target = NULL;
+ int tsize, size = 256; /* size = initial buffer capacity */
+ while (1) {
+ target = realloc(target, size);
+ if (!target) /* failed to allocate */
+ return 0;
+ tsize = readlink(file, target, size);
+ if (tsize < 0) { /* a readlink() error occurred */
+ free(target);
+ return 0;
+ }
+ if (tsize < size)
+ break;
+ /* possibly truncated readlink() result, double size and retry */
+ size *= 2;
+ }
+ target[tsize] = '\0';
+ lua_pushlstring(L, target, tsize);
+ free(target);
+ return 1;
+#endif
+}
+
+/*
+** Get symbolic link information using lstat.
+*/
+static int link_info (lua_State *L) {
+ int ret;
+ if (lua_isstring (L, 2) && (strcmp(lua_tostring(L, 2), "target") == 0)) {
+ int ok = push_link_target(L);
+ return ok ? 1 : pusherror(L, "could not obtain link target");
+ }
+ ret = _file_info_ (L, LSTAT_FUNC);
+ if (ret == 1 && lua_type(L, -1) == LUA_TTABLE) {
+ int ok = push_link_target(L);
+ if (ok) {
+ lua_setfield(L, -2, "target");
+ }
+ }
+ return ret;
+}
+
+
+/*
+** Assumes the table is on top of the stack.
+*/
+static void set_info (lua_State *L) {
+ lua_pushliteral(L, "Copyright (C) 2003-2017 Kepler Project");
+ lua_setfield(L, -2, "_COPYRIGHT");
+ lua_pushliteral(L, "LuaFileSystem is a Lua library developed to complement the set of functions related to file systems offered by the standard Lua distribution");
+ lua_setfield(L, -2, "_DESCRIPTION");
+ lua_pushliteral(L, "LuaFileSystem " LFS_VERSION);
+ lua_setfield(L, -2, "_VERSION");
+}
+
+
+static const struct luaL_Reg fslib[] = {
+ {"attributes", file_info},
+ {"chdir", change_dir},
+ {"currentdir", get_dir},
+ {"dir", dir_iter_factory},
+ {"link", make_link},
+ {"lock", file_lock},
+ {"mkdir", make_dir},
+ {"rmdir", remove_dir},
+ {"symlinkattributes", link_info},
+ {"setmode", lfs_f_setmode},
+ {"touch", file_utime},
+ {"unlock", file_unlock},
+ {"lock_dir", lfs_lock_dir},
+ {NULL, NULL},
+};
+
+LFS_EXPORT int luaopen_lfs (lua_State *L) {
+ dir_create_meta (L);
+ lock_create_meta (L);
+ new_lib (L, fslib);
+ lua_pushvalue(L, -1);
+ lua_setglobal(L, LFS_LIBNAME);
+ set_info (L);
+ return 1;
+}
diff --git a/libs/lfs/lfs.h b/libs/lfs/lfs.h
@@ -0,0 +1,34 @@
+/*
+** LuaFileSystem
+** Copyright Kepler Project 2003 - 2017 (http://keplerproject.github.io/luafilesystem)
+*/
+
+/* Define 'chdir' for systems that do not implement it */
+#ifdef NO_CHDIR
+ #define chdir(p) (-1)
+ #define chdir_error "Function 'chdir' not provided by system"
+#else
+ #define chdir_error strerror(errno)
+#endif
+
+#ifdef _WIN32
+ #define chdir(p) (_chdir(p))
+ #define getcwd(d, s) (_getcwd(d, s))
+ #define rmdir(p) (_rmdir(p))
+ #define LFS_EXPORT __declspec (dllexport)
+ #ifndef fileno
+ #define fileno(f) (_fileno(f))
+ #endif
+#else
+ #define LFS_EXPORT
+#endif
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+LFS_EXPORT int luaopen_lfs (lua_State *L);
+
+#ifdef __cplusplus
+}
+#endif
diff --git a/libs/lpeg.lua b/libs/lpeg.lua
@@ -5,4 +5,5 @@ lib( "lpeg", {
"libs/lpeg/lptree",
"libs/lpeg/lpvm",
} )
-obj_replace_cxxflags( "libs/lpeg/%", "-c -O2 -x c" )
+-- obj_replace_cxxflags( "libs/lpeg/%", "-c -O2 -x c" )
+obj_cxxflags( "libs/lpeg/%", "/c /TC /I libs/lua" )
diff --git a/libs/lua.lua b/libs/lua.lua
@@ -33,4 +33,5 @@ lib( "lua", {
"libs/lua/loadlib",
"libs/lua/linit",
} )
-obj_replace_cxxflags( "libs/lua/%", "-c -O2 -x c -DLUA_COMPAT_5_2 -DLUA_USE_LINUX" )
+-- obj_replace_cxxflags( "libs/lua/%", "-c -O2 -x c -DLUA_COMPAT_5_2 -DLUA_USE_LINUX" )
+obj_cxxflags( "libs/lua/%", "/c /TC /DLUA_COMPAT_5_2" )
diff --git a/make.lua b/make.lua
@@ -1,9 +1,14 @@
require( "scripts.gen_makefile" )
+-- bin( "wintest", { "win", "platform_network", "ggformat", "strlcpy", "strlcat", "patterns", "strtonum" } )
+-- msvc_bin_ldflags( "wintest", "gdi32.lib Ws2_32.lib" )
+
-- it only really makes sense to vendor lua on windows/osx
-- but we can't actually build for them yet so do nothing for now
--- require( "libs.lua" )
--- require( "libs.lpeg" )
+require( "libs.lua" )
+require( "libs.lpeg" )
+require( "libs.lfs" )
-bin( "mudgangster", { "src/main", "src/script", "src/textbox", "src/input", "src/x11", "src/platform_network" } )
+bin( "mudgangster", { "src/win32", "src/script", "src/textbox", "src/input", "src/platform_network" }, { "lua", "lpeg", "lfs" } )
+msvc_bin_ldflags( "mudgangster", "gdi32.lib Ws2_32.lib" )
gcc_bin_ldflags( "mudgangster", "-lm -lX11 -llua" ) -- -Wl,-E" ) need to export symbols when vendoring
diff --git a/scripts/gen_makefile.lua b/scripts/gen_makefile.lua
@@ -14,7 +14,7 @@ local configs = {
toolchain = "msvc",
- cxxflags = "/I src /c /Oi /Gm- /GR- /EHa- /EHsc /nologo /DNOMINMAX /DWIN32_LEAN_AND_MEAN",
+ cxxflags = "/I . /I src /c /Oi /Gm- /GR- /EHa- /EHsc /nologo /DNOMINMAX /DWIN32_LEAN_AND_MEAN",
ldflags = "user32.lib shell32.lib advapi32.lib dbghelp.lib /nologo",
warnings = "/W4 /wd4100 /wd4146 /wd4189 /wd4201 /wd4324 /wd4351 /wd4127 /wd4505 /wd4530 /wd4702 /D_CRT_SECURE_NO_WARNINGS",
},
diff --git a/scripts/merge.lua b/scripts/merge.lua
@@ -5,7 +5,7 @@ end
local lfs = require( "lfs" )
-local merged = { "#! /usr/bin/env lua" .. ( arg[ 3 ] or "" ) }
+local merged = { }
local root = arg[ 1 ]
local main = arg[ 2 ]
diff --git a/scripts/pack_lua.sh b/scripts/pack_lua.sh
@@ -4,4 +4,5 @@ set -o pipefail
mkdir -p build
lua scripts/merge.lua src/lua main.lua > build/lua_combined.lua
-exec lua scripts/merge.lua src/lua main.lua | luac -o - - | lua scripts/bin2arr.lua > build/lua_bytecode.h
+#exec lua scripts/merge.lua src/lua main.lua | luac -o - - | lua scripts/bin2arr.lua > build/lua_bytecode.h
+exec lua scripts/merge.lua src/lua main.lua | lua scripts/bin2arr.lua > build/lua_bytecode.h
diff --git a/src/common.h b/src/common.h
@@ -13,6 +13,8 @@ typedef uint8_t u8;
typedef uint16_t u16;
typedef uint32_t u32;
+typedef int64_t s64;
+
#define FATAL( form, ... ) \
do { \
printf( "[FATAL] " form, ##__VA_ARGS__ ); \
diff --git a/src/lua/script.lua b/src/lua/script.lua
@@ -2,6 +2,7 @@ local lfs = require( "lfs" )
local serialize = require( "serialize" )
local ScriptsDir = os.getenv( "HOME" ) .. "/.mudgangster/scripts"
+local ScriptsDir = os.getenv( "APPDATA" ) .. "\\Mud Gangster\\scripts"
package.path = package.path .. ";" .. ScriptsDir .. "/?.lua"
diff --git a/src/lua/utils.lua b/src/lua/utils.lua
@@ -41,15 +41,17 @@ function math.avg( a, b )
end
function io.readable( path )
- local file, err = io.open( path, "r" )
-
- if not file then
- return false, err
- end
-
- io.close( file )
-
+ -- TODO: this gives a no permissions error on windows
return true
+ -- local file, err = io.open( path, "r" )
+ --
+ -- if not file then
+ -- return false, err
+ -- end
+ --
+ -- io.close( file )
+ --
+ -- return true
end
function enforce( var, name, ... )
diff --git a/src/platform_network.h b/src/platform_network.h
@@ -1,6 +1,7 @@
#pragma once
#include "platform.h"
+#include "common.h"
#if PLATFORM_WINDOWS
typedef s64 OSSocket;
diff --git a/src/script.cc b/src/script.cc
@@ -4,7 +4,11 @@
#include "platform_time.h"
+#if PLATFORM_WINDOWS
+#include "libs/lua/lua.hpp"
+#else
#include <lua.hpp>
+#endif
#if LUA_VERSION_NUM < 502
#define luaL_len lua_objlen
@@ -235,6 +239,7 @@ extern "C" int mud_now( lua_State * L ) {
#if PLATFORM_WINDOWS
extern "C" int luaopen_lpeg( lua_State * L );
+extern "C" int luaopen_lfs( lua_State * L );
#endif
void script_init() {
@@ -246,13 +251,16 @@ void script_init() {
#if PLATFORM_WINDOWS
luaL_requiref( lua, "lpeg", luaopen_lpeg, 0 );
lua_pop( lua, 1 );
+
+ luaL_requiref( lua, "lfs", luaopen_lfs, 0 );
+ lua_pop( lua, 1 );
#endif
lua_getglobal( lua, "debug" );
lua_getfield( lua, -1, "traceback" );
lua_remove( lua, -2 );
- if( luaL_loadbufferx( lua, ( const char * ) lua_bytecode, sizeof( lua_bytecode ), "main", "b" ) != LUA_OK ) {
+ if( luaL_loadbufferx( lua, ( const char * ) lua_bytecode, sizeof( lua_bytecode ), "main", "t" ) != LUA_OK ) {
printf( "Error reading main.lua: %s\n", lua_tostring( lua, -1 ) );
exit( 1 );
}
diff --git a/src/win32.cc b/src/win32.cc
@@ -0,0 +1,688 @@
+#include <windows.h>
+#include <windowsx.h>
+#include <Winsock2.h>
+#include <stdio.h>
+
+#include "common.h"
+#include "input.h"
+#include "script.h"
+#include "textbox.h"
+
+#include "platform_network.h"
+
+#define WINDOW_CLASSNAME "MudGangsterClass"
+
+struct {
+ HWND hwnd;
+ HDC hdc;
+
+ TextBox main_text;
+ TextBox chat_text;
+
+ int width, height;
+
+ bool has_focus;
+} UI;
+
+struct Socket {
+ TCPSocket sock;
+ bool in_use;
+};
+
+static Socket sockets[ 128 ];
+
+void * platform_connect( const char ** err, const char * host, int port ) {
+ size_t idx;
+ {
+ bool ok = false;
+ for( size_t i = 0; i < ARRAY_COUNT( sockets ); i++ ) {
+ if( !sockets[ i ].in_use ) {
+ idx = i;
+ ok = true;
+ break;
+ }
+ }
+
+ if( !ok ) {
+ *err = "too many connections";
+ return NULL;
+ }
+ }
+
+ NetAddress addr;
+ {
+ bool ok = dns_first( host, &addr );
+ if( !ok ) {
+ *err = "couldn't resolve hostname"; // TODO: error from dns_first
+ return NULL;
+ }
+ }
+ addr.port = checked_cast< u16 >( port );
+
+ TCPSocket sock;
+ bool ok = net_new_tcp( &sock, addr, NET_BLOCKING );
+ if( !ok ) {
+ *err = "net_new_tcp";
+ return NULL;
+ }
+
+ sockets[ idx ].sock = sock;
+ sockets[ idx ].in_use = true;
+
+ WSAAsyncSelect( sock.fd, UI.hwnd, 12345, FD_READ | FD_CLOSE );
+
+ return &sockets[ idx ];
+}
+
+void platform_send( void * vsock, const char * data, size_t len ) {
+ Socket * sock = ( Socket * ) vsock;
+ net_send( sock->sock, data, len );
+}
+
+void platform_close( void * vsock ) {
+ Socket * sock = ( Socket * ) vsock;
+ net_destroy( &sock->sock );
+ sock->in_use = false;
+}
+
+static Socket * socket_from_fd( int fd ) {
+ for( Socket & sock : sockets ) {
+ if( sock.in_use && sock.sock.fd == fd ) {
+ return &sock;
+ }
+ }
+
+ return NULL;
+}
+
+struct MudFont {
+ int ascent;
+ int width, height;
+ HFONT font;
+};
+
+struct {
+ COLORREF bg;
+ COLORREF status_bg;
+ COLORREF cursor;
+
+ MudFont font;
+ MudFont bold_font;
+
+ union {
+ struct {
+ COLORREF black;
+ COLORREF red;
+ COLORREF green;
+ COLORREF yellow;
+ COLORREF blue;
+ COLORREF magenta;
+ COLORREF cyan;
+ COLORREF white;
+
+ COLORREF lblack;
+ COLORREF lred;
+ COLORREF lgreen;
+ COLORREF lyellow;
+ COLORREF lblue;
+ COLORREF lmagenta;
+ COLORREF lcyan;
+ COLORREF lwhite;
+
+ COLORREF system;
+ } Colours;
+
+ COLORREF colours[ 2 ][ 8 ];
+ };
+} Style;
+
+static COLORREF get_colour( Colour colour, bool bold ) {
+ switch( colour ) {
+ case SYSTEM:
+ return Style.Colours.system;
+ case COLOUR_BG:
+ return Style.bg;
+ case COLOUR_STATUSBG:
+ return Style.status_bg;
+ case COLOUR_CURSOR:
+ return Style.cursor;
+ }
+
+ return Style.colours[ bold ][ colour ];
+}
+
+void ui_fill_rect( int left, int top, int width, int height, Colour colour, bool bold ) {
+ HBRUSH brush = CreateSolidBrush( get_colour( colour, bold ) );
+ RECT r = { left, top, left + width, top + height };
+ FillRect( UI.hdc, &r, brush );
+ DeleteObject( brush );
+}
+
+void ui_draw_char( int left, int top, char c, Colour colour, bool bold, bool force_bold_font ) {
+ int left_spacing = Style.font.width / 2;
+ int right_spacing = Style.font.width - left_spacing;
+ int line_height = Style.font.height + SPACING;
+ int top_spacing = line_height / 2;
+ int bot_spacing = line_height - top_spacing;
+
+ // TODO: not the right char...
+ // if( uint8_t( c ) == 155 ) { // fill
+ // ui_fill_rect( left, top, Style.font.width, Style.font.height, colour, bold );
+ // return;
+ // }
+
+ // TODO: this has a vertical seam. using textbox-space coordinates would help
+ if( uint8_t( c ) == 176 ) { // light shade
+ for( int y = 0; y < Style.font.height; y += 3 ) {
+ for( int x = y % 6 == 0 ? 0 : 1; x < Style.font.width; x += 2 ) {
+ ui_fill_rect( left + x, top + y, 1, 1, colour, bold );
+ }
+ }
+ return;
+ }
+
+ // TODO: this has a horizontal seam but so does mm2k
+ if( uint8_t( c ) == 177 ) { // medium shade
+ for( int y = 0; y < Style.font.height; y += 2 ) {
+ for( int x = y % 4 == 0 ? 1 : 0; x < Style.font.width; x += 2 ) {
+ ui_fill_rect( left + x, top + y, 1, 1, colour, bold );
+ }
+ }
+ return;
+ }
+
+ // TODO: this probably has a horizontal seam
+ if( uint8_t( c ) == 178 ) { // heavy shade
+ for( int y = 0; y < Style.font.height + SPACING; y++ ) {
+ for( int x = y % 2 == 0 ? 1 : 0; x < Style.font.width; x += 2 ) {
+ ui_fill_rect( left + x, top + y, 1, 1, colour, bold );
+ }
+ }
+ return;
+ }
+
+ if( uint8_t( c ) == 179 ) { // vertical
+ ui_fill_rect( left + left_spacing, top, 1, line_height, colour, bold );
+ return;
+ // set_fg( colour, bold );
+ // const char asdf[] = "│";
+ // Xutf8DrawString( UI.display, UI.back_buffer, ( bold ? Style.fontBold : Style.font ).font, UI.gc, left, top + Style.font.ascent + SPACING, asdf, sizeof( asdf ) - 1 );
+ }
+
+ if( uint8_t( c ) == 180 ) { // right stopper
+ ui_fill_rect( left, top + top_spacing, left_spacing, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top, 1, line_height, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 186 ) { // double vertical
+ ui_fill_rect( left + left_spacing - 1, top, 1, line_height, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top, 1, line_height, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 187 ) { // double top right
+ ui_fill_rect( left, top + top_spacing - 1, right_spacing + 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top + top_spacing, 1, bot_spacing + 1, colour, bold );
+ ui_fill_rect( left, top + top_spacing + 1, right_spacing - 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing - 1, top + top_spacing + 1, 1, bot_spacing - 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 188 ) { // double bottom right
+ ui_fill_rect( left, top + top_spacing + 1, right_spacing + 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top, 1, top_spacing + 1, colour, bold );
+ ui_fill_rect( left, top + top_spacing - 1, right_spacing - 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing - 1, top, 1, top_spacing - 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 191 ) { // top right
+ ui_fill_rect( left, top + top_spacing, left_spacing, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top + top_spacing, 1, bot_spacing, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 192 ) { // bottom left
+ ui_fill_rect( left + left_spacing, top + top_spacing, right_spacing, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top, 1, top_spacing, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 193 ) { // bottom stopper
+ ui_fill_rect( left + left_spacing, top, 1, top_spacing, colour, bold );
+ ui_fill_rect( left, top + top_spacing, Style.font.width, 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 194 ) { // top stopper
+ ui_fill_rect( left + left_spacing, top + top_spacing, 1, bot_spacing, colour, bold );
+ ui_fill_rect( left, top + top_spacing, Style.font.width, 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 195 ) { // left stopper
+ ui_fill_rect( left + left_spacing, top + top_spacing, right_spacing, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top, 1, line_height, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 196 ) { // horizontal
+ ui_fill_rect( left, top + top_spacing, Style.font.width, 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 197 ) { // cross
+ ui_fill_rect( left, top + top_spacing, Style.font.width, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top, 1, line_height, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 200 ) { // double bottom left
+ ui_fill_rect( left + left_spacing - 1, top + top_spacing + 1, right_spacing + 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing - 1, top, 1, top_spacing + 1, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top + top_spacing - 1, right_spacing - 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top, 1, top_spacing - 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 201 ) { // double top left
+ ui_fill_rect( left + left_spacing - 1, top + top_spacing - 1, right_spacing + 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing - 1, top + top_spacing - 1, 1, bot_spacing + 1, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top + top_spacing + 1, right_spacing - 1, 1, colour, bold );
+ ui_fill_rect( left + left_spacing + 1, top + top_spacing + 1, 1, bot_spacing - 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 205 ) { // double horizontal
+ ui_fill_rect( left, top + top_spacing - 1, Style.font.width, 1, colour, bold );
+ ui_fill_rect( left, top + top_spacing + 1, Style.font.width, 1, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 217 ) { // bottom right
+ ui_fill_rect( left, top + top_spacing, right_spacing, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top, 1, top_spacing, colour, bold );
+ return;
+ }
+
+ if( uint8_t( c ) == 218 ) { // top left
+ ui_fill_rect( left + left_spacing, top + top_spacing, right_spacing, 1, colour, bold );
+ ui_fill_rect( left + left_spacing, top + top_spacing, 1, bot_spacing, colour, bold );
+ return;
+ }
+
+ SelectObject( UI.hdc, ( bold || force_bold_font ? Style.bold_font : Style.font ).font );
+ SetTextColor( UI.hdc, get_colour( colour, bold ) );
+ TextOutA( UI.hdc, left, top + SPACING, &c, 1 );
+}
+
+void ui_dirty( int left, int top, int width, int height ) {
+}
+
+typedef struct {
+ char c;
+
+ Colour fg;
+ bool bold;
+} StatusChar;
+
+static StatusChar * statusContents = NULL;
+static size_t statusCapacity = 256;
+static size_t statusLen = 0;
+
+void ui_clear_status() {
+ statusLen = 0;
+}
+
+void ui_statusAdd( const char c, const Colour fg, const bool bold ) {
+ if( ( statusLen + 1 ) * sizeof( StatusChar ) > statusCapacity ) {
+ size_t newcapacity = statusCapacity * 2;
+ StatusChar * newcontents = ( StatusChar * ) realloc( statusContents, newcapacity );
+ if( !newcontents )
+ FATAL( "realloc" );
+
+ statusContents = newcontents;
+ statusCapacity = newcapacity;
+ }
+
+ statusContents[ statusLen ] = { c, fg, bold };
+ statusLen++;
+}
+
+void ui_draw_status() {
+ ui_fill_rect( 0, UI.height - PADDING * 4 - Style.font.height * 2, UI.width, Style.font.height + PADDING * 2, COLOUR_STATUSBG, false );
+
+ for( size_t i = 0; i < statusLen; i++ ) {
+ StatusChar sc = statusContents[ i ];
+
+ int x = PADDING + i * Style.font.width;
+ int y = UI.height - ( PADDING * 3 ) - Style.font.height * 2 - SPACING;
+ ui_draw_char( x, y, sc.c, sc.fg, sc.bold );
+ }
+
+ ui_dirty( 0, UI.height - ( PADDING * 4 ) - ( Style.font.height * 2 ), UI.width, Style.font.height + ( PADDING * 2 ) );
+}
+
+void draw_input() {
+ InputBuffer input = input_get_buffer();
+
+ int top = UI.height - PADDING - Style.font.height;
+ ui_fill_rect( PADDING, top, UI.width - PADDING * 2, Style.font.height, COLOUR_BG, false );
+
+ for( size_t i = 0; i < input.len; i++ )
+ ui_draw_char( PADDING + i * Style.font.width, top - SPACING, input.buf[ i ], WHITE, false );
+
+ ui_fill_rect( PADDING + input.cursor_pos * Style.font.width, top, Style.font.width, Style.font.height, COLOUR_CURSOR, false );
+
+ if( input.cursor_pos < input.len ) {
+ ui_draw_char( PADDING + input.cursor_pos * Style.font.width, top - SPACING, input.buf[ input.cursor_pos ], COLOUR_BG, false );
+ }
+
+ ui_dirty( PADDING, UI.height - ( PADDING + Style.font.height ), UI.width - PADDING * 2, Style.font.height );
+}
+
+void ui_draw() {
+ ui_fill_rect( 0, 0, UI.width, UI.height, COLOUR_BG, false );
+
+ draw_input();
+ ui_draw_status();
+
+ textbox_draw( &UI.chat_text );
+ textbox_draw( &UI.main_text );
+
+ int spacerY = ( 2 * PADDING ) + ( Style.font.height + SPACING ) * CHAT_ROWS;
+ ui_fill_rect( 0, spacerY, UI.width, 1, COLOUR_STATUSBG, false );
+
+ ui_dirty( 0, 0, UI.width, UI.height );
+}
+
+void ui_handleXEvents() { }
+
+void ui_main_draw() {
+ textbox_draw( &UI.main_text );
+}
+
+void ui_main_newline() {
+ textbox_newline( &UI.main_text );
+}
+
+void ui_main_print( const char * str, size_t len, Colour fg, Colour bg, bool bold ) {
+ textbox_add( &UI.main_text, str, len, fg, bg, bold );
+}
+
+void ui_chat_draw() {
+ textbox_draw( &UI.chat_text );
+}
+
+void ui_chat_newline() {
+ textbox_newline( &UI.chat_text );
+}
+
+void ui_chat_print( const char * str, size_t len, Colour fg, Colour bg, bool bold ) {
+ textbox_add( &UI.chat_text, str, len, fg, bg, bold );
+}
+
+void ui_get_font_size( int * fw, int * fh ) {
+ *fw = Style.font.width;
+ *fh = Style.font.height;
+}
+
+void ui_urgent() {
+ FlashWindow( UI.hwnd, FALSE );
+}
+
+LRESULT CALLBACK WndProc( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam ) {
+ switch( msg ) {
+ case WM_CREATE: {
+ UI.hdc = GetDC( hwnd );
+ SetBkMode( UI.hdc, TRANSPARENT );
+
+ Style.font.font = CreateFont( 14, 0, 0, 0, FW_REGULAR,
+ FALSE, FALSE, FALSE, DEFAULT_CHARSET,
+ OUT_TT_PRECIS, CLIP_DEFAULT_PRECIS, CLEARTYPE_QUALITY, DEFAULT_PITCH,
+ "Dina" );
+ Style.bold_font.font = CreateFont( 14, 0, 0, 0, FW_BOLD,
+ FALSE, FALSE, FALSE, DEFAULT_CHARSET,
+ OUT_TT_PRECIS, CLIP_DEFAULT_PRECIS, CLEARTYPE_QUALITY, DEFAULT_PITCH,
+ "Dina" );
+
+ SelectObject( UI.hdc, Style.font.font );
+
+ TEXTMETRIC metrics;
+ GetTextMetrics( UI.hdc, &metrics );
+
+ Style.font.height = metrics.tmHeight;
+ Style.font.width = metrics.tmMaxCharWidth;
+ Style.font.ascent = metrics.tmAscent;
+
+ Style.bg = RGB( 0x1a, 0x1a, 0x1a );
+ Style.status_bg = RGB( 0x33, 0x33, 0x33 );
+ Style.cursor = RGB( 0x00, 0xff, 0x00 );
+ Style.Colours.system = RGB( 0xff, 0xff, 0xff );
+
+ Style.Colours.black = RGB( 0x1a, 0x1a, 0x1a );
+ Style.Colours.red = RGB( 0xca, 0x44, 0x33 );
+ Style.Colours.green = RGB( 0x17, 0x8a, 0x3a );
+ Style.Colours.yellow = RGB( 0xdc, 0x7c, 0x2a );
+ Style.Colours.blue = RGB( 0x41, 0x5e, 0x87 );
+ Style.Colours.magenta = RGB( 0x5e, 0x46, 0x8c );
+ Style.Colours.cyan = RGB( 0x35, 0x78, 0x9b );
+ Style.Colours.white = RGB( 0xb6, 0xc2, 0xc4 );
+
+ Style.Colours.lblack = RGB( 0x66, 0x66, 0x66 );
+ Style.Colours.lred = RGB( 0xff, 0x29, 0x54 );
+ Style.Colours.lgreen = RGB( 0x5d, 0xd0, 0x30 );
+ Style.Colours.lyellow = RGB( 0xfa, 0xfc, 0x4f );
+ Style.Colours.lblue = RGB( 0x35, 0x81, 0xe1 );
+ Style.Colours.lmagenta = RGB( 0x87, 0x5f, 0xff );
+ Style.Colours.lcyan = RGB( 0x29, 0xfb, 0xff );
+ Style.Colours.lwhite = RGB( 0xce, 0xdb, 0xde );
+ } break;
+
+ case WM_SIZE: {
+ int old_width = UI.width;
+ int old_height = UI.height;
+
+ UI.width = LOWORD( lParam );
+ UI.height = HIWORD( lParam );
+
+ if( UI.width == old_width && UI.height == old_height )
+ break;
+
+ textbox_set_pos( &UI.chat_text, PADDING, PADDING );
+ textbox_set_size( &UI.chat_text, UI.width - ( 2 * PADDING ), ( Style.font.height + SPACING ) * CHAT_ROWS );
+
+ textbox_set_pos( &UI.main_text, PADDING, ( PADDING * 2 ) + CHAT_ROWS * ( Style.font.height + SPACING ) + 1 );
+ textbox_set_size( &UI.main_text, UI.width - ( 2 * PADDING ), UI.height
+ - ( ( ( Style.font.height + SPACING ) * CHAT_ROWS ) + ( PADDING * 2 ) )
+ - ( ( Style.font.height * 2 ) + ( PADDING * 5 ) ) - 1
+ );
+
+ ui_draw();
+ } break;
+
+ case WM_LBUTTONDOWN: {
+ // char szFileName[MAX_PATH];
+ // HINSTANCE hInstance = GetModuleHandle(NULL);
+ //
+ // GetModuleFileName(hInstance, szFileName, MAX_PATH);
+ // MessageBox(hwnd, szFileName, "This program is:", MB_OK | MB_ICONINFORMATION);
+ } break;
+
+ case WM_MOUSEMOVE: {
+ // char buf[ 128 ];
+ // int x = GET_X_LPARAM( lParam );
+ // int y = GET_Y_LPARAM( lParam );
+ // int l = snprintf( buf, sizeof( buf ), "what the fuck son %d %d", x, y );
+ // TextOutA( dc, 10, 10, buf, l );
+ } break;
+
+ case WM_CLOSE:
+ DestroyWindow( hwnd );
+ break;
+
+ case WM_DESTROY:
+ PostQuitMessage( 0 );
+ break;
+
+ case WM_TIMER: {
+ // SelectObject( dc, c % 2 == 0 ? font : bold_font );
+ // char buf[ 64 ];
+ // int l = snprintf( buf, sizeof( buf ), "timer %d", c );
+ // TextOutA( dc, 10, 40, buf, l );
+ // c++;
+ } break;
+
+ case WM_CHAR: {
+ if( wParam >= ' ' && wParam < 128 ) {
+ char c = wParam;
+ input_add( &c, 1 );
+ ui_draw();
+ }
+ } break;
+
+ case WM_KEYDOWN: {
+ switch( wParam ) {
+ case VK_BACK:
+ input_backspace();
+ break;
+
+ case VK_DELETE:
+ input_delete();
+ break;
+
+ case VK_RETURN:
+ input_return();
+ break;
+
+ case VK_LEFT:
+ input_left();
+ break;
+
+ case VK_RIGHT:
+ input_right();
+ break;
+
+ case VK_UP:
+ input_up();
+ break;
+
+ case VK_DOWN:
+ input_down();
+ break;
+ }
+ ui_draw();
+ } break;
+
+ case 12345: {
+ if( WSAGETSELECTERROR( lParam ) ) {
+ printf( "bye\n" );
+ break;
+ }
+
+ SOCKET fd = ( SOCKET ) wParam;
+ Socket * sock = socket_from_fd( fd );
+ if( sock == NULL )
+ break;
+
+ assert( WSAGETSELECTEVENT( lParam ) == FD_CLOSE || WSAGETSELECTEVENT( lParam ) == FD_READ );
+
+ while( true ) {
+ char buf[ 2048 ];
+ int n = recv( fd, buf, sizeof( buf ), 0 );
+ if( n >= 0 ) {
+ script_socketData( sock, n > 0 ? buf : NULL, n );
+ }
+ else {
+ int err = WSAGetLastError();
+ if( err != WSAEWOULDBLOCK )
+ FATAL( "shit %d", err );
+ break;
+ }
+ }
+ } break;
+
+ default:
+ return DefWindowProc(hwnd, msg, wParam, lParam);
+ }
+ return 0;
+}
+
+int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow ) {
+ net_init();
+
+ for( Socket & s : sockets ) {
+ s.in_use = false;
+ }
+
+ UI = { };
+
+ textbox_init( &UI.main_text, SCROLLBACK_SIZE );
+ textbox_init( &UI.chat_text, CHAT_ROWS );
+
+ statusContents = ( StatusChar * ) malloc( statusCapacity * sizeof( StatusChar ) );
+ if( statusContents == NULL )
+ FATAL( "malloc" );
+
+ WNDCLASSEX wc = { };
+ wc.cbSize = sizeof( WNDCLASSEX );
+ wc.lpfnWndProc = WndProc;
+ wc.hInstance = hInstance;
+ wc.hCursor = LoadCursor( NULL, IDC_ARROW );
+ wc.hbrBackground = ( HBRUSH ) COLOR_WINDOW;
+ wc.lpszClassName = WINDOW_CLASSNAME;
+ wc.hbrBackground = CreateSolidBrush( RGB( 0x1a, 0x1a, 0x1a ) );
+
+ if( !RegisterClassEx( &wc ) ) {
+ MessageBox(NULL, "Window Registration Failed!", "Error!",
+ MB_ICONEXCLAMATION | MB_OK);
+ return 0;
+ }
+
+ UI.hwnd = CreateWindowExA(
+ NULL,
+ WINDOW_CLASSNAME,
+ "Mud Gangster",
+ WS_OVERLAPPEDWINDOW | WS_MAXIMIZE | WS_VISIBLE,
+ CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
+ NULL, NULL, hInstance, NULL);
+
+ if( UI.hwnd == NULL ) {
+ MessageBox( NULL, "Window Creation Failed!", "Error!", MB_ICONEXCLAMATION | MB_OK );
+ return 0;
+ }
+
+ ShowWindow( UI.hwnd, SW_MAXIMIZE );
+ UpdateWindow( UI.hwnd );
+ SetTimer( UI.hwnd, 1, 500, NULL );
+
+ input_init();
+ script_init();
+
+ // NetAddress addr;
+ // dns_first( "mikejsavage.co.uk", &addr );
+ // TCPSocket sock;
+ // bool ok = net_new_tcp( &sock, addr, NET_BLOCKING );
+ // if( !ok ) return 1;
+ // printf( "%lld\n", sock.fd );
+ //
+ // WSAAsyncSelect( sock.fd, UI.hwnd, 12345, FD_READ | FD_CLOSE );
+ //
+ // const char buf[] = "GET / HTTP/1.1\r\n"
+ // "Host: mikejsavage.co.uk\r\n"
+ // "Connection: close\r\n"
+ // "\r\n\r\n";
+ // if( !net_send( sock, buf, strlen( buf ) ) )
+ // FATAL( "net_send" );
+
+ MSG msg;
+ while( GetMessage( &msg, NULL, 0, 0 ) > 0 ) {
+ TranslateMessage( &msg );
+ DispatchMessage( &msg );
+ }
+
+ net_term();
+
+ return 0;
+}
diff --git a/src/win32_network.cc b/src/win32_network.cc
@@ -1,6 +1,8 @@
#include <winsock2.h>
#include <ws2tcpip.h>
+#define ssize_t int
+
static int NET_SEND_FLAGS = 0;
void net_init() {
