438 lines
11 KiB
C++
438 lines
11 KiB
C++
/* cygheap.h: Cygwin heap manager.
|
|
|
|
Copyright 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007 Red Hat, Inc.
|
|
|
|
This file is part of Cygwin.
|
|
|
|
This software is a copyrighted work licensed under the terms of the
|
|
Cygwin license. Please consult the file "CYGWIN_LICENSE" for
|
|
details. */
|
|
|
|
#include "hires.h"
|
|
|
|
#undef cfree
|
|
|
|
enum cygheap_types
|
|
{
|
|
HEAP_FHANDLER,
|
|
HEAP_STR,
|
|
HEAP_ARGV,
|
|
HEAP_BUF,
|
|
HEAP_MOUNT,
|
|
HEAP_SIGS,
|
|
HEAP_ARCHETYPES,
|
|
HEAP_TLS,
|
|
HEAP_COMMUNE,
|
|
HEAP_1_START,
|
|
HEAP_1_HOOK,
|
|
HEAP_1_STR,
|
|
HEAP_1_ARGV,
|
|
HEAP_1_BUF,
|
|
HEAP_1_EXEC,
|
|
HEAP_1_MAX = 100,
|
|
HEAP_2_STR,
|
|
HEAP_MMAP = 200
|
|
};
|
|
|
|
#define incygheap(s) (cygheap && ((char *) (s) >= (char *) cygheap) && ((char *) (s) <= ((char *) cygheap_max)))
|
|
|
|
struct _cmalloc_entry
|
|
{
|
|
union
|
|
{
|
|
DWORD b;
|
|
char *ptr;
|
|
};
|
|
struct _cmalloc_entry *prev;
|
|
char data[0];
|
|
};
|
|
|
|
struct cygheap_root_mount_info
|
|
{
|
|
char posix_path[CYG_MAX_PATH];
|
|
unsigned posix_pathlen;
|
|
char native_path[CYG_MAX_PATH];
|
|
unsigned native_pathlen;
|
|
};
|
|
|
|
/* CGF: FIXME This doesn't belong here */
|
|
|
|
class cygheap_root
|
|
{
|
|
/* Root directory information.
|
|
This is used after a chroot is called. */
|
|
struct cygheap_root_mount_info *m;
|
|
|
|
public:
|
|
bool posix_ok (const char *path)
|
|
{
|
|
if (!m)
|
|
return 1;
|
|
return path_prefix_p (m->posix_path, path, m->posix_pathlen);
|
|
}
|
|
bool ischroot_native (const char *path)
|
|
{
|
|
if (!m)
|
|
return 1;
|
|
return strncasematch (m->native_path, path, m->native_pathlen)
|
|
&& (path[m->native_pathlen] == '\\' || !path[m->native_pathlen]);
|
|
}
|
|
const char *unchroot (const char *path)
|
|
{
|
|
if (!m)
|
|
return path;
|
|
const char *p = path + m->posix_pathlen;
|
|
if (!*p)
|
|
p = "/";
|
|
return p;
|
|
}
|
|
bool exists () {return !!m;}
|
|
void set (const char *, const char *);
|
|
size_t posix_length () const { return m->posix_pathlen; }
|
|
const char *posix_path () const { return m->posix_path; }
|
|
size_t native_length () const { return m->native_pathlen; }
|
|
const char *native_path () const { return m->native_path; }
|
|
};
|
|
|
|
enum homebodies
|
|
{
|
|
CH_HOMEDRIVE,
|
|
CH_HOMEPATH,
|
|
CH_HOME
|
|
};
|
|
|
|
class cygheap_user
|
|
{
|
|
/* Extendend user information.
|
|
The information is derived from the internal_getlogin call
|
|
when on a NT system. */
|
|
char *pname; /* user's name */
|
|
char *plogsrv; /* Logon server, may be FQDN */
|
|
char *pdomain; /* Logon domain of the user */
|
|
char *homedrive; /* User's home drive */
|
|
char *homepath; /* User's home path */
|
|
char *psystemroot; /* Value of SYSTEMROOT */
|
|
char *pwinname; /* User's name as far as Windows knows it */
|
|
char *puserprof; /* User profile */
|
|
cygsid effec_cygsid; /* buffer for user's SID */
|
|
cygsid saved_cygsid; /* Remains intact even after impersonation */
|
|
public:
|
|
__uid32_t saved_uid; /* Remains intact even after impersonation */
|
|
__gid32_t saved_gid; /* Ditto */
|
|
__uid32_t real_uid; /* Remains intact on seteuid, replaced by setuid */
|
|
__gid32_t real_gid; /* Ditto */
|
|
user_groups groups; /* Primary and supp SIDs */
|
|
|
|
/* token is needed if set(e)uid should be called. It can be set by a call
|
|
to `set_impersonation_token()'. */
|
|
HANDLE external_token;
|
|
HANDLE internal_token;
|
|
HANDLE curr_primary_token;
|
|
HANDLE curr_imp_token;
|
|
|
|
/* CGF 2002-06-27. I removed the initializaton from this constructor
|
|
since this class is always allocated statically. That means that everything
|
|
is zero anyway so there is no need to initialize it to zero. Since the
|
|
token initialization is always handled during process startup as well,
|
|
I've removed the constructor entirely. Please reinstate this if this
|
|
situation ever changes.
|
|
cygheap_user () : pname (NULL), plogsrv (NULL), pdomain (NULL),
|
|
homedrive (NULL), homepath (NULL),
|
|
token (INVALID_HANDLE_VALUE) {}
|
|
*/
|
|
|
|
~cygheap_user ();
|
|
|
|
void init ();
|
|
void set_name (const char *new_name);
|
|
const char *name () const { return pname; }
|
|
|
|
const char *env_logsrv (const char *, size_t);
|
|
const char *env_homepath (const char *, size_t);
|
|
const char *env_homedrive (const char *, size_t);
|
|
const char *env_userprofile (const char *, size_t);
|
|
const char *env_domain (const char *, size_t);
|
|
const char *env_name (const char *, size_t);
|
|
const char *env_systemroot (const char *, size_t);
|
|
|
|
const char *logsrv ()
|
|
{
|
|
const char *p = env_logsrv ("LOGONSERVER=", sizeof ("LOGONSERVER=") - 1);
|
|
return (p == almost_null) ? NULL : p;
|
|
}
|
|
const char *winname ()
|
|
{
|
|
const char *p = env_name ("USERNAME=", sizeof ("USERNAME=") - 1);
|
|
return (p == almost_null) ? NULL : p;
|
|
}
|
|
const char *domain ()
|
|
{
|
|
const char *p = env_domain ("USERDOMAIN=", sizeof ("USERDOMAIN=") - 1);
|
|
return (p == almost_null) ? NULL : p;
|
|
}
|
|
BOOL set_sid (PSID new_sid) {return (BOOL) (effec_cygsid = new_sid);}
|
|
BOOL set_saved_sid () { return (BOOL) (saved_cygsid = effec_cygsid); }
|
|
PSID sid () { return effec_cygsid; }
|
|
PSID saved_sid () { return saved_cygsid; }
|
|
const char *ontherange (homebodies what, struct passwd * = NULL);
|
|
#define NO_IMPERSONATION NULL
|
|
bool issetuid () const { return curr_imp_token != NO_IMPERSONATION; }
|
|
HANDLE primary_token () { return curr_primary_token; }
|
|
HANDLE imp_token () { return curr_imp_token; }
|
|
void deimpersonate ()
|
|
{
|
|
RevertToSelf ();
|
|
}
|
|
bool reimpersonate ()
|
|
{
|
|
if (issetuid ())
|
|
return ImpersonateLoggedOnUser (primary_token ());
|
|
return true;
|
|
}
|
|
bool has_impersonation_tokens ()
|
|
{ return external_token != NO_IMPERSONATION
|
|
|| internal_token != NO_IMPERSONATION
|
|
|| curr_primary_token != NO_IMPERSONATION; }
|
|
void close_impersonation_tokens ()
|
|
{
|
|
if (curr_imp_token != NO_IMPERSONATION)
|
|
CloseHandle (curr_imp_token);
|
|
if (curr_primary_token != NO_IMPERSONATION
|
|
&& curr_primary_token != external_token
|
|
&& curr_primary_token != internal_token)
|
|
CloseHandle (curr_primary_token);
|
|
if (external_token != NO_IMPERSONATION)
|
|
CloseHandle (external_token);
|
|
if (internal_token != NO_IMPERSONATION)
|
|
CloseHandle (internal_token);
|
|
}
|
|
PWCHAR get_windows_id (PWCHAR buf)
|
|
{
|
|
return effec_cygsid.string (buf);
|
|
}
|
|
char *get_windows_id (char *buf)
|
|
{
|
|
return effec_cygsid.string (buf);
|
|
}
|
|
|
|
const char *test_uid (char *&, const char *, size_t)
|
|
__attribute__ ((regparm (3)));
|
|
};
|
|
|
|
/* cwd cache stuff. */
|
|
|
|
class muto;
|
|
|
|
struct cwdstuff
|
|
{
|
|
char *posix;
|
|
UNICODE_STRING win32;
|
|
HANDLE dir;
|
|
DWORD drive_length;
|
|
static muto cwd_lock;
|
|
char *get (char *, int = 1, int = 0, unsigned = NT_MAX_PATH);
|
|
HANDLE get_handle () { return dir; }
|
|
DWORD get_drive (char * dst)
|
|
{
|
|
cwd_lock.acquire ();
|
|
DWORD ret = sys_wcstombs (dst, NT_MAX_PATH, win32.Buffer, drive_length);
|
|
cwd_lock.release ();
|
|
return ret;
|
|
}
|
|
void init ();
|
|
int set (PUNICODE_STRING, const char *, bool);
|
|
};
|
|
|
|
#ifdef DEBUGGING
|
|
struct cygheap_debug
|
|
{
|
|
handle_list starth;
|
|
handle_list *endh;
|
|
handle_list freeh[500];
|
|
};
|
|
#endif
|
|
|
|
struct user_heap_info
|
|
{
|
|
void *base;
|
|
void *ptr;
|
|
void *top;
|
|
void *max;
|
|
unsigned chunk;
|
|
unsigned slop;
|
|
};
|
|
|
|
struct hook_chain
|
|
{
|
|
void **loc;
|
|
const void *func;
|
|
struct hook_chain *next;
|
|
};
|
|
|
|
struct init_cygheap
|
|
{
|
|
_cmalloc_entry *chain;
|
|
char *buckets[32];
|
|
cygheap_root root;
|
|
cygheap_user user;
|
|
user_heap_info user_heap;
|
|
mode_t umask;
|
|
HANDLE console_h;
|
|
cwdstuff cwd;
|
|
dtable fdtab;
|
|
char *shared_prefix;
|
|
char shared_prefix_buf[8];
|
|
#ifdef DEBUGGING
|
|
cygheap_debug debug;
|
|
#endif
|
|
struct sigaction *sigs;
|
|
|
|
fhandler_tty_slave *ctty; /* Current tty */
|
|
#ifdef NEWVFORK
|
|
fhandler_tty_slave *ctty_on_hold;
|
|
#endif
|
|
struct _cygtls **threadlist;
|
|
size_t sthreads;
|
|
pid_t pid; /* my pid */
|
|
HANDLE pid_handle; /* handle for my pid */
|
|
hook_chain hooks;
|
|
void close_ctty ();
|
|
int manage_console_count (const char *, int, bool = false) __attribute__ ((regparm (3)));
|
|
private:
|
|
int console_count;
|
|
};
|
|
|
|
|
|
#define _CYGHEAPSIZE_SLOP (128 * 1024)
|
|
#define CYGHEAPSIZE (sizeof (init_cygheap) + (20000 * sizeof (fhandler_union)) + _CYGHEAPSIZE_SLOP)
|
|
#define CYGHEAPSIZE_MIN (sizeof (init_cygheap) + (10000 * sizeof (fhandler_union)))
|
|
|
|
extern init_cygheap *cygheap;
|
|
extern void *cygheap_max;
|
|
|
|
class cygheap_fdmanip
|
|
{
|
|
protected:
|
|
int fd;
|
|
fhandler_base **fh;
|
|
bool locked;
|
|
public:
|
|
cygheap_fdmanip (): fh (NULL) {}
|
|
virtual ~cygheap_fdmanip ()
|
|
{
|
|
if (locked)
|
|
cygheap->fdtab.unlock ();
|
|
}
|
|
void release ()
|
|
{
|
|
cygheap->fdtab.release (fd);
|
|
}
|
|
operator int &() {return fd;}
|
|
operator fhandler_base* &() {return *fh;}
|
|
operator fhandler_socket* () const {return reinterpret_cast<fhandler_socket *> (*fh);}
|
|
operator fhandler_pipe* () const {return reinterpret_cast<fhandler_pipe *> (*fh);}
|
|
void operator = (fhandler_base *fh) {*this->fh = fh;}
|
|
fhandler_base *operator -> () const {return *fh;}
|
|
bool isopen () const
|
|
{
|
|
if (*fh)
|
|
return true;
|
|
set_errno (EBADF);
|
|
return false;
|
|
}
|
|
};
|
|
|
|
class cygheap_fdnew : public cygheap_fdmanip
|
|
{
|
|
public:
|
|
cygheap_fdnew (int seed_fd = -1, bool lockit = true)
|
|
{
|
|
if (lockit)
|
|
cygheap->fdtab.lock ();
|
|
if (seed_fd < 0)
|
|
fd = cygheap->fdtab.find_unused_handle ();
|
|
else
|
|
fd = cygheap->fdtab.find_unused_handle (seed_fd + 1);
|
|
if (fd >= 0)
|
|
{
|
|
locked = lockit;
|
|
fh = cygheap->fdtab + fd;
|
|
}
|
|
else
|
|
{
|
|
set_errno (EMFILE);
|
|
if (lockit)
|
|
cygheap->fdtab.unlock ();
|
|
locked = false;
|
|
}
|
|
}
|
|
void operator = (fhandler_base *fh) {*this->fh = fh;}
|
|
};
|
|
|
|
class cygheap_fdget : public cygheap_fdmanip
|
|
{
|
|
public:
|
|
cygheap_fdget (int fd, bool lockit = false, bool do_set_errno = true)
|
|
{
|
|
if (lockit)
|
|
cygheap->fdtab.lock ();
|
|
if (fd >= 0 && fd < (int) cygheap->fdtab.size
|
|
&& *(fh = cygheap->fdtab + fd) != NULL)
|
|
{
|
|
this->fd = fd;
|
|
locked = lockit;
|
|
}
|
|
else
|
|
{
|
|
this->fd = -1;
|
|
if (do_set_errno)
|
|
set_errno (EBADF);
|
|
if (lockit)
|
|
cygheap->fdtab.unlock ();
|
|
locked = false;
|
|
}
|
|
}
|
|
};
|
|
|
|
class cygheap_fdenum : public cygheap_fdmanip
|
|
{
|
|
public:
|
|
cygheap_fdenum (bool lockit = false)
|
|
{
|
|
locked = lockit;
|
|
if (lockit)
|
|
cygheap->fdtab.lock ();
|
|
fd = -1;
|
|
}
|
|
int next ()
|
|
{
|
|
while (++fd < (int) cygheap->fdtab.size)
|
|
if (*(fh = cygheap->fdtab + fd) != NULL)
|
|
return fd;
|
|
return -1;
|
|
}
|
|
void rewind ()
|
|
{
|
|
fd = -1;
|
|
}
|
|
};
|
|
|
|
class child_info;
|
|
void __stdcall cygheap_fixup_in_child (bool);
|
|
extern "C" {
|
|
void __stdcall cfree (void *) __attribute__ ((regparm(1)));
|
|
void *__stdcall cmalloc (cygheap_types, DWORD) __attribute__ ((regparm(2)));
|
|
void *__stdcall crealloc (void *, DWORD) __attribute__ ((regparm(2)));
|
|
void *__stdcall ccalloc (cygheap_types, DWORD, DWORD) __attribute__ ((regparm(3)));
|
|
void *__stdcall cmalloc_abort (cygheap_types, DWORD) __attribute__ ((regparm(2)));
|
|
void *__stdcall crealloc_abort (void *, DWORD) __attribute__ ((regparm(2)));
|
|
void *__stdcall ccalloc_abort (cygheap_types, DWORD, DWORD) __attribute__ ((regparm(3)));
|
|
PWCHAR __stdcall cwcsdup (const PWCHAR) __attribute__ ((regparm(1)));
|
|
PWCHAR __stdcall cwcsdup1 (const PWCHAR) __attribute__ ((regparm(1)));
|
|
char *__stdcall cstrdup (const char *) __attribute__ ((regparm(1)));
|
|
char *__stdcall cstrdup1 (const char *) __attribute__ ((regparm(1)));
|
|
void __stdcall cfree_and_set (char *&, char * = NULL) __attribute__ ((regparm(2)));
|
|
void __stdcall cygheap_init ();
|
|
extern char _cygheap_start[] __attribute__((section(".idata")));
|
|
}
|