summaryrefslogtreecommitdiffstats
path: root/Python
diff options
context:
space:
mode:
Diffstat (limited to 'Python')
-rw-r--r--Python/import.c210
1 files changed, 112 insertions, 98 deletions
diff --git a/Python/import.c b/Python/import.c
index b532651..284d881 100644
--- a/Python/import.c
+++ b/Python/import.c
@@ -40,6 +40,8 @@ OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
extern int verbose; /* Defined in pythonmain.c */
+extern long getmtime(); /* Defined in posixmodule.c */
+
#ifdef DEBUG
#define D(x) x
#else
@@ -47,7 +49,12 @@ extern int verbose; /* Defined in pythonmain.c */
#endif
#ifdef USE_DL
+#ifdef SUN_SHLIB
+#include <dlfcn.h>
+typedef void (*dl_funcptr)();
+#else
#include "dl.h"
+#endif /* SUN_SHLIB */
extern char *argv0;
#endif
@@ -94,44 +101,46 @@ add_module(name)
return m;
}
-/* Suffixes used by open_module: */
+/* Suffixes used by find_module: */
#define PY_SUFFIX ".py"
+#define PYC_SUFFIX ".pyc"
#ifdef USE_DL
+#ifdef SUN_SHLIB
+#define O_SUFFIX "module.so"
+#else
#define O_SUFFIX "module.o"
+#endif /* SUN_SHLIB */
#endif
-/* Find and open a module file, using sys.path.
- Return a NULL pointer if no module file is found.
- When dynamic loading is enabled, the contents of namebuf
- is important when NULL is returned: if namebuf[0] != '\0'
- a dl-able object file was found and namebuf is its pathname. */
-
+/* This will search for a module named 'name' with the extension 'ext'
+ and return it in 'namebuf' and return the mtime of each in 'mtime'.
+ It returns a file pointer opened for 'mode' if successful, NULL if
+ unsuccessful.
+ */
static FILE *
-open_module(name, namebuf)
+find_module(name, ext, mode, namebuf, mtime)
char *name;
- char *namebuf; /* XXX No buffer overflow checks! */
+ char *ext;
+ char *mode;
+ char *namebuf;
+ long *mtime;
{
object *path;
FILE *fp;
-
+
path = sysget("path");
if (path == NULL || !is_listobject(path)) {
/* No path -- at least try current directory */
-#ifdef USE_DL
strcpy(namebuf, name);
- strcat(namebuf, O_SUFFIX);
- if (getmtime(namebuf) > 0)
+ strcat(namebuf, ext);
+ if ((fp = fopen(namebuf, mode)) == NULL)
return NULL;
-#endif
- strcpy(namebuf, name);
- strcat(namebuf, PY_SUFFIX);
- fp = fopen(namebuf, "r");
- }
- else {
+ *mtime = getmtime(namebuf);
+ return fp;
+ } else {
int npath = getlistsize(path);
int i;
- fp = NULL;
for (i = 0; i < npath; i++) {
object *v = getlistitem(path, i);
int len;
@@ -141,22 +150,16 @@ open_module(name, namebuf)
len = getstringsize(v);
if (len > 0 && namebuf[len-1] != SEP)
namebuf[len++] = SEP;
-#ifdef USE_DL
- strcpy(namebuf+len, name);
- strcat(namebuf, O_SUFFIX);
- if (getmtime(namebuf) > 0)
- return NULL;
-#endif
strcpy(namebuf+len, name);
- strcat(namebuf, PY_SUFFIX);
- fp = fopen(namebuf, "r");
- if (fp != NULL)
- break;
+ strcat(namebuf, ext);
+ if ((fp = fopen(namebuf, mode)) == NULL)
+ continue;
+ *mtime = getmtime(namebuf);
+ return fp;
}
}
- if (fp == NULL)
- namebuf[0] = '\0';
- return fp;
+ namebuf[0] = '\0';
+ return NULL;
}
static object *
@@ -173,66 +176,63 @@ get_module(m, name, m_ret)
char namebuf[MAXPATHLEN+1];
int namelen;
long mtime;
- extern long getmtime();
-
- fp = open_module(name, namebuf);
- if (fp == NULL) {
+
#ifdef USE_DL
- if (namebuf[0] != '\0') {
- char funcname[258];
- dl_funcptr p;
- D(fprintf(stderr, "Found %s\n", namebuf));
- sprintf(funcname, "init%s", name);
- p = dl_loadmod(argv0, namebuf, funcname);
- if (p == NULL) {
- D(fprintf(stderr, "dl_loadmod failed\n"));
- }
- else {
- if (verbose)
- fprintf(stderr,
- "import %s # dynamically loaded from \"%s\"\n",
- name, namebuf);
- (*p)();
- *m_ret = m = dictlookup(modules, name);
- if (m == NULL) {
- err_setstr(SystemError,
- "dynamic module missing");
- return NULL;
- }
- else {
- D(fprintf(stderr,
- "module %s loaded!\n", name));
- INCREF(None);
- return None;
- }
- }
- }
-#endif
- if (m == NULL) {
- sprintf(namebuf, "no module named %.200s", name);
- err_setstr(ImportError, namebuf);
+ if ((fpc = find_module(name, O_SUFFIX, "rb",
+ namebuf, &mtime)) != NULL) {
+ char funcname[258];
+ dl_funcptr p;
+ D(fprintf(stderr, "Found %s\n", namebuf));
+ fclose(fpc);
+ sprintf(funcname, "init%s", name);
+#ifdef SUN_SHLIB
+ {
+ void *handle = dlopen (namebuf, 1);
+ p = (dl_funcptr) dlsym(handle, funcname);
}
- else {
- sprintf(namebuf, "no source for module %.200s", name);
- err_setstr(ImportError, namebuf);
+#else
+ p = dl_loadmod(argv0, namebuf, funcname);
+#endif /* SUN_SHLIB */
+ if (p == NULL) {
+ D(fprintf(stderr, "dl_loadmod failed\n"));
+ } else {
+ if (verbose)
+ fprintf(stderr,
+ "import %s # dynamically loaded from \"%s\"\n",
+ name, namebuf);
+ (*p)();
+ *m_ret = m = dictlookup(modules, name);
+ if (m == NULL) {
+ err_setstr(SystemError,
+ "dynamic module missing");
+ return NULL;
+ } else {
+ D(fprintf(stderr,
+ "module %s loaded!\n", name));
+ INCREF(None);
+ return None;
+ }
}
- return NULL;
}
- /* Get mtime -- always useful */
- mtime = getmtime(namebuf);
- /* Check ".pyc" file first */
- namelen = strlen(namebuf);
- namebuf[namelen] = 'c';
- namebuf[namelen+1] = '\0';
- fpc = fopen(namebuf, "rb");
- if (fpc != NULL) {
+ else
+#endif
+ if ((fpc = find_module(name, PYC_SUFFIX, "rb",
+ namebuf, &mtime)) != NULL) {
long pyc_mtime;
long magic;
+ namebuf[(strlen(namebuf)-1)] = '\0';
+ mtime = getmtime(namebuf);
magic = rd_long(fpc);
pyc_mtime = rd_long(fpc);
- if (magic == MAGIC && pyc_mtime == mtime && mtime != 0 && mtime != -1) {
+ if (mtime != -1 && mtime > pyc_mtime) {
+ fclose(fpc);
+ fp = fopen(namebuf, "rb");
+ goto read_py;
+ }
+ if (magic == MAGIC) {
v = rd_object(fpc);
- if (v == NULL || err_occurred() || !is_codeobject(v)) {
+ if (v == NULL || err_occurred() ||
+ !is_codeobject(v)) {
err_clear();
XDECREF(v);
}
@@ -243,27 +243,41 @@ get_module(m, name, m_ret)
if (verbose) {
if (co != NULL)
fprintf(stderr,
- "import %s # precompiled from \"%s\"\n",
+ "import %s # precompiled from \"%s\"\n",
name, namebuf);
else
fprintf(stderr,
- "# invalid precompiled file \"%s\"\n",
+ "# invalid precompiled file \"%s\"\n",
namebuf);
}
}
- namebuf[namelen] = '\0';
- if (co == NULL) {
- if (verbose)
- fprintf(stderr,
- "import %s # from \"%s\"\n",
- name, namebuf);
- err = parse_file(fp, namebuf, file_input, &n);
+ else if ((fp = find_module(name, PY_SUFFIX, "r",
+ namebuf, &mtime)) != NULL) {
+read_py:
+ namelen = strlen(namebuf);
+ if (co == NULL) {
+ if (verbose)
+ fprintf(stderr,
+ "import %s # from \"%s\"\n",
+ name, namebuf);
+ err = parse_file(fp, namebuf, file_input, &n);
+ } else
+ err = E_DONE;
+ fclose(fp);
+ if (err != E_DONE) {
+ err_input(err);
+ return NULL;
+ }
}
- else
- err = E_DONE;
- fclose(fp);
- if (err != E_DONE) {
- err_input(err);
+ else {
+ if (m == NULL) {
+ sprintf(namebuf, "no module named %.200s", name);
+ err_setstr(ImportError, namebuf);
+ }
+ else {
+ sprintf(namebuf, "no source for module %.200s", name);
+ err_setstr(ImportError, namebuf);
+ }
return NULL;
}
if (m == NULL) {