import _socket

"import socket" does not quite work yet

Makefile

@@ -291,7 +291,7 @@ STDLIB_OBJS := stdlib.bc.o stdlib.stripped.bc.o
 STDLIB_RELEASE_OBJS := stdlib.release.bc.o
 ASM_SRCS := $(wildcard src/runtime/*.S)
 
-STDMODULE_SRCS := errnomodule.c shamodule.c sha256module.c sha512module.c _math.c mathmodule.c md5.c md5module.c _randommodule.c _sre.c operator.c binascii.c pwdmodule.c posixmodule.c _struct.c datetimemodule.c _functoolsmodule.c _collectionsmodule.c itertoolsmodule.c resource.c signalmodule.c selectmodule.c fcntlmodule.c timemodule.c arraymodule.c zlibmodule.c _codecsmodule.c $(EXTRA_STDMODULE_SRCS)
+STDMODULE_SRCS := errnomodule.c shamodule.c sha256module.c sha512module.c _math.c mathmodule.c md5.c md5module.c _randommodule.c _sre.c operator.c binascii.c pwdmodule.c posixmodule.c _struct.c datetimemodule.c _functoolsmodule.c _collectionsmodule.c itertoolsmodule.c resource.c signalmodule.c selectmodule.c fcntlmodule.c timemodule.c arraymodule.c zlibmodule.c _codecsmodule.c socketmodule.c $(EXTRA_STDMODULE_SRCS)
 STDOBJECT_SRCS := structseq.c capsule.c stringobject.c $(EXTRA_STDOBJECT_SRCS)
 STDPYTHON_SRCS := pyctype.c getargs.c formatter_string.c pystrtod.c dtoa.c $(EXTRA_STDPYTHON_SRCS)
 FROM_CPYTHON_SRCS := $(addprefix from_cpython/Modules/,$(STDMODULE_SRCS)) $(addprefix from_cpython/Objects/,$(STDOBJECT_SRCS)) $(addprefix from_cpython/Python/,$(STDPYTHON_SRCS))

from_cpython/CMakeLists.txt

@@ -15,7 +15,7 @@ endforeach(STDLIB_FILE)
 add_custom_target(copy_stdlib ALL DEPENDS ${STDLIB_TARGETS})
 
 # compile specified files in from_cpython/Modules
-file(GLOB_RECURSE STDMODULE_SRCS Modules errnomodule.c shamodule.c sha256module.c sha512module.c _math.c mathmodule.c md5.c md5module.c _randommodule.c _sre.c operator.c binascii.c pwdmodule.c posixmodule.c _struct.c datetimemodule.c _functoolsmodule.c _collectionsmodule.c itertoolsmodule.c resource.c signalmodule.c selectmodule.c fcntlmodule.c timemodule.c arraymodule.c zlibmodule.c _codecsmodule.c)
+file(GLOB_RECURSE STDMODULE_SRCS Modules errnomodule.c shamodule.c sha256module.c sha512module.c _math.c mathmodule.c md5.c md5module.c _randommodule.c _sre.c operator.c binascii.c pwdmodule.c posixmodule.c _struct.c datetimemodule.c _functoolsmodule.c _collectionsmodule.c itertoolsmodule.c resource.c signalmodule.c selectmodule.c fcntlmodule.c timemodule.c arraymodule.c zlibmodule.c _codecsmodule.c socketmodule.c)
 
 # compile specified files in from_cpython/Objects
 file(GLOB_RECURSE STDOBJECT_SRCS Objects structseq.c capsule.c stringobject.c)

from_cpython/Include/pyconfig.h

@@ -49,6 +49,9 @@
 #define HAVE_SELECT 1
 #define HAVE_ALARM 1
 #define HAVE_SYMLINK 1
+#define HAVE_ADDRINFO 1
+#define HAVE_SOCKADDR_STORAGE 1
+#define HAVE_SOCKETPAIR 1
 
 #define PY_FORMAT_LONG_LONG "ll"
 #define PY_FORMAT_SIZE_T "z"

from_cpython/Include/pyport.h

@@ -12,6 +12,8 @@
 #define PYSTON_NOEXCEPT
 #endif
 
+#define Py_PROTO(x) x
+
 // Pyston change: these are just hard-coded for now:
 typedef ssize_t         Py_ssize_t;
 #define Py_FORMAT_PARSETUPLE(func,p1,p2)

from_cpython/Modules/socketmodule.c

@@ -4578,6 +4578,9 @@ init_socket(void)
     if (!os_init())
         return;
 
+    // Pyston change: we require calling PyType_Ready for now:
+    PyType_Ready(&sock_type);
+
     Py_TYPE(&sock_type) = &PyType_Type;
     m = Py_InitModule3(PySocket_MODULE_NAME,
                        socket_methods,

src/core/types.h

@@ -439,12 +439,13 @@ public:
 
     void setattr(const std::string& attr, Box* val, SetattrRewriteArgs* rewrite_args);
     void giveAttr(const std::string& attr, Box* val) {
-        assert(this->getattr(attr) == NULL);
+        assert(!this->hasattr(attr));
         this->setattr(attr, val, NULL);
     }
 
     Box* getattr(const std::string& attr, GetattrRewriteArgs* rewrite_args);
     Box* getattr(const std::string& attr) { return getattr(attr, NULL); }
+    bool hasattr(const std::string& attr) { return getattr(attr) != NULL; }
     void delattr(const std::string& attr, DelattrRewriteArgs* rewrite_args);
 
     Box* reprIC();

src/runtime/builtin_modules/builtins.cpp

@@ -674,8 +674,10 @@ static BoxedClass* makeBuiltinException(BoxedClass* base, const char* name, int
 }
 
 extern "C" PyObject* PyErr_NewException(char* name, PyObject* _base, PyObject* dict) noexcept {
-    RELEASE_ASSERT(_base == NULL, "unimplemented");
-    RELEASE_ASSERT(dict == NULL, "unimplemented");
+    if (_base == NULL)
+        _base = Exception;
+    if (dict == NULL)
+        dict = new BoxedDict();
 
     try {
         char* dot_pos = strchr(name, '.');
@@ -683,13 +685,16 @@ extern "C" PyObject* PyErr_NewException(char* name, PyObject* _base, PyObject* d
         int n = strlen(name);
         BoxedString* boxedName = boxStrConstantSize(dot_pos + 1, n - (dot_pos - name) - 1);
 
-        BoxedClass* base = Exception;
-        BoxedClass* cls
-            = new BoxedHeapClass(base, NULL, offsetof(BoxedException, attrs), sizeof(BoxedException), true, boxedName);
+        // It can also be a tuple of bases
+        RELEASE_ASSERT(isSubclass(_base->cls, type_cls), "");
+        BoxedClass* base = static_cast<BoxedClass*>(_base);
 
-        cls->giveAttr("__module__", boxStrConstantSize(name, dot_pos - name));
-        // TODO Not sure if this should be called here
-        fixup_slot_dispatchers(cls);
+        if (PyDict_GetItemString(dict, "__module__") == NULL) {
+            PyDict_SetItemString(dict, "__module__", boxStrConstantSize(name, dot_pos - name));
+        }
+        checkAndThrowCAPIException();
+
+        Box* cls = runtimeCall(type_cls, ArgPassSpec(3), boxedName, new BoxedTuple({ base }), dict, NULL, NULL);
         return cls;
     } catch (ExcInfo e) {
         abort();

src/runtime/objmodel.cpp

@@ -3543,14 +3543,17 @@ Box* typeNew(Box* _cls, Box* arg1, Box* arg2, Box** _args) {
     }
 
     made->tp_dictoffset = base->tp_dictoffset;
-    made->giveAttr("__module__", boxString(getCurrentModule()->name()));
-    made->giveAttr("__doc__", None);
 
     for (const auto& p : attr_dict->d) {
         assert(p.first->cls == str_cls);
         made->setattr(static_cast<BoxedString*>(p.first)->s, p.second, NULL);
     }
 
+    if (!made->hasattr("__module__"))
+        made->giveAttr("__module__", boxString(getCurrentModule()->name()));
+    if (!made->hasattr("__doc__"))
+        made->giveAttr("__doc__", None);
+
     made->tp_new = base->tp_new;
 
     PystonType_Ready(made);

src/runtime/types.cpp

@@ -63,6 +63,7 @@ extern "C" void inittime();
 extern "C" void initarray();
 extern "C" void initzlib();
 extern "C" void init_codecs();
+extern "C" void init_socket();
 
 namespace pyston {
 
@@ -1371,6 +1372,7 @@ void setupRuntime() {
     initarray();
     initzlib();
     init_codecs();
+    init_socket();
 
     setupSysEnd();
 

test/tests/socket_test.py

+import socket