From 1e19b24ea0081cde0cbccfecd9698a0b55a4450c Mon Sep 17 00:00:00 2001
From: Paul Sokolovsky <pfalcon@users.sourceforge.net>
Date: Sat, 8 Feb 2014 21:15:46 +0200
Subject: [PATCH] io.File, socket types: Add fileno() method.

Implementation is duplicated, but tolerate this for now, because there's
no clear idea how to de-dup it.
---
 tests/io/file-stdio.py | 4 ++++
 unix/file.c            | 7 +++++++
 unix/socket.c          | 7 +++++++
 3 files changed, 18 insertions(+)
 create mode 100644 tests/io/file-stdio.py

diff --git a/tests/io/file-stdio.py b/tests/io/file-stdio.py
new file mode 100644
index 000000000..cbdb07016
--- /dev/null
+++ b/tests/io/file-stdio.py
@@ -0,0 +1,4 @@
+import sys
+
+print(sys.stdin.fileno())
+print(sys.stdout.fileno())
diff --git a/unix/file.c b/unix/file.c
index 5249b5bba..7bbe3bfd0 100644
--- a/unix/file.c
+++ b/unix/file.c
@@ -48,6 +48,12 @@ static mp_obj_t fdfile_close(mp_obj_t self_in) {
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(fdfile_close_obj, fdfile_close);
 
+static mp_obj_t fdfile_fileno(mp_obj_t self_in) {
+    mp_obj_fdfile_t *self = self_in;
+    return MP_OBJ_NEW_SMALL_INT(self->fd);
+}
+static MP_DEFINE_CONST_FUN_OBJ_1(fdfile_fileno_obj, fdfile_fileno);
+
 static mp_obj_fdfile_t *fdfile_new(int fd) {
     mp_obj_fdfile_t *o = m_new_obj(mp_obj_fdfile_t);
     o->base.type = &rawfile_type;
@@ -99,6 +105,7 @@ static mp_obj_t fdfile_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const
 }
 
 static const mp_method_t rawfile_type_methods[] = {
+        { "fileno", &fdfile_fileno_obj },
         { "read", &mp_stream_read_obj },
         { "readall", &mp_stream_readall_obj },
         { "readline", &mp_stream_unbuffered_readline_obj},
diff --git a/unix/socket.c b/unix/socket.c
index c38e38a0a..aaa15cbee 100644
--- a/unix/socket.c
+++ b/unix/socket.c
@@ -84,6 +84,12 @@ static mp_obj_t socket_close(mp_obj_t self_in) {
 }
 static MP_DEFINE_CONST_FUN_OBJ_1(socket_close_obj, socket_close);
 
+static mp_obj_t socket_fileno(mp_obj_t self_in) {
+    mp_obj_socket_t *self = self_in;
+    return MP_OBJ_NEW_SMALL_INT(self->fd);
+}
+static MP_DEFINE_CONST_FUN_OBJ_1(socket_fileno_obj, socket_fileno);
+
 static mp_obj_t socket_connect(mp_obj_t self_in, mp_obj_t addr_in) {
     mp_obj_socket_t *self = self_in;
     buffer_info_t bufinfo;
@@ -209,6 +215,7 @@ static mp_obj_t socket_make_new(mp_obj_t type_in, uint n_args, uint n_kw, const
 }
 
 static const mp_method_t microsocket_type_methods[] = {
+        { "fileno", &socket_fileno_obj },
         { "read", &mp_stream_read_obj },
         { "readall", &mp_stream_readall_obj },
         { "readline", &mp_stream_unbuffered_readline_obj},
-- 
GitLab