Ticket #4242: filetransfer.diff

src/jarabe/model/filetransfer.py

@@ -20 +20 @@
 
 from gi.repository import GObject
 from gi.repository import Gio
+from gi.repository import GLib
 import dbus
 from telepathy.interfaces import CONNECTION_INTERFACE_REQUESTS, CHANNEL
 from telepathy.constants import CONNECTION_HANDLE_TYPE_CONTACT,     \
@@ -33 +34 @@
 from jarabe.util.telepathy import connection_watcher
 from jarabe.model import neighborhood
 
+# Avoid "Fatal Python error: GC object already tracked"
+# http://stackoverflow.com/questions/7496629/gstreamer-appsrc-causes-random-crashes
+GObject.threads_init()
 
 FT_STATE_NONE = 0
 FT_STATE_PENDING = 1
@@ -73 +77 @@
         self._pending_buffers = []
 
     def start(self):
-        self._input_stream.read_async(self._CHUNK_SIZE, self.__read_async_cb,
-                                      GObject.PRIORITY_LOW)
+        self._input_stream.read_async(
+            self._pending_buffers, self._CHUNK_SIZE, GLib.PRIORITY_LOW,
+            None, self.__read_async_cb, 'user_data arg')
 
-    def __read_async_cb(self, input_stream, result):
+    def __read_async_cb(self, input_stream, result, user_data):
+        print user_data
         data = input_stream.read_finish(result)
 
         if not data:
             logging.debug('closing input stream')
-            self._input_stream.close()
+            self._input_stream.close(None)
         else:
             self._pending_buffers.append(data)
-            self._input_stream.read_async(self._CHUNK_SIZE,
-                                            self.__read_async_cb,
-                                            GObject.PRIORITY_LOW)
+            self._input_stream.read_async(
+                self._pending_buffers, self._CHUNK_SIZE, GLib.PRIORITY_LOW,
+                None, self.__read_async_cb, 'user_data_arg2')
         self._write_next_buffer()
 
     def __write_async_cb(self, output_stream, result, user_data):
+        print user_data
         count_ = output_stream.write_finish(result)
 
         if not self._pending_buffers and \
                 not self._output_stream.has_pending() and \
                 not self._input_stream.has_pending():
             logging.debug('closing output stream')
-            output_stream.close()
+            output_stream.close(None)
             self.emit('finished')
         else:
             self._write_next_buffer()
@@ -106 +113 @@
             data = self._pending_buffers.pop(0)
             # TODO: we pass the buffer as user_data because of
             # http://bugzilla.gnome.org/show_bug.cgi?id=564102
-            self._output_stream.write_async(data, self.__write_async_cb,
-                                            GObject.PRIORITY_LOW,
-                                            user_data=data)
+            self._output_stream.write_async(
+                data, self._CHUNK_SIZE, GLib.PRIORITY_LOW, None,
+                self.__write_async_cb, 'user_data_arg3')
 
 
 class BaseFileTransfer(GObject.GObject):
@@ -215 +222 @@
             # close the fd when it goes out of scope
             self._socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
             self._socket.connect(self._socket_address)
-            input_stream = Gio.unix.InputStream(self._socket.fileno(), True)
+            input_stream = Gio.UnixInputStream.new(self._socket.fileno(), True)
 
-            destination_file = Gio.File(self.destination_path)
+            destination_file = Gio.File.new_for_path(self.destination_path)
             if self.initial_offset == 0:
-                output_stream = destination_file.create()
+                output_stream = destination_file.create(
+                    Gio.FileCreateFlags.PRIVATE, None)
             else:
                 output_stream = destination_file.append_to()
 
@@ -277 +285 @@
             # closes the fd when it goes out of scope
             self._socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
             self._socket.connect(self._socket_address)
-            output_stream = Gio.unix.OutputStream(self._socket.fileno(), True)
+            output_stream = Gio.UnixOutputStream.new(
+                self._socket.fileno(), True)
 
             logging.debug('opening %s for reading', self._file_name)
-            input_stream = Gio.File(self._file_name).read()
+            input_stream = Gio.File.new_for_path(self._file_name).read(None)
             if self.initial_offset > 0:
                 input_stream.skip(self.initial_offset)
 
@@ -356 +365 @@
 if __name__ == '__main__':
     import tempfile
 
-    test_file_name = '/home/tomeu/isos/Soas2-200904031934.iso'
-    test_input_stream = Gio.File(test_file_name).read()
-    test_output_stream = Gio.File(tempfile.mkstemp()[1]).append_to()
+    test_file_name = '/home/humitos/test.py'
+    test_input_stream = Gio.File.new_for_path(test_file_name).read(None)
+    test_output_stream = Gio.File.new_for_path(
+        tempfile.mkstemp()[1]).append_to(Gio.FileCreateFlags.PRIVATE, None)
 
     # TODO: Use splice_async when it gets implemented
     splicer = StreamSplicer(test_input_stream, test_output_stream)