First commit

4 files changed, 335 insertions, 0 deletions

diff --git a/lib/.isodumper.log b/lib/.isodumper.log
new file mode 100644
index 0000000..b2821c3
--- /dev/null
+++ b/lib/.isodumper.log
@@ -0,0 +1,3 @@
+Image: /documents/boot-nonfree.iso
+Target Device: V88 V88 (/dev/sdb)
+You have not the rights for writing on the device
diff --git a/lib/find_devices b/lib/find_devices
new file mode 100755
index 0000000..6a56c03
--- /dev/null
+++ b/lib/find_devices
@@ -0,0 +1,20 @@
+#!/usr/bin/python
+
+import dbus
+import sys
+
+bus = dbus.SystemBus()
+proxy = bus.get_object("org.freedesktop.UDisks", "/org/freedesktop/UDisks")
+iface = dbus.Interface(proxy, "org.freedesktop.UDisks")
+
+devs = iface.EnumerateDevices()
+
+for dev in devs:
+	dev_obj = bus.get_object("org.freedesktop.UDisks", dev)
+	dev = dbus.Interface(dev_obj, "org.freedesktop.DBus.Properties")
+
+	if str(dev.Get('', 'DriveConnectionInterface')) == 'usb' and not str(dev.Get('', 'PartitionType')) and str(dev.Get('', 'DeviceIsMediaAvailable')) == '1':
+		path = str(dev.Get('', 'DeviceFile'))
+		vend = str(dev.Get('', 'DriveVendor'))
+		model = str(dev.Get('', 'DriveModel'))
+		print vend + ' ' + model + ', ' +path
diff --git a/lib/find_devices.sh b/lib/find_devices.sh
new file mode 100755
index 0000000..b371be0
--- /dev/null
+++ b/lib/find_devices.sh
@@ -0,0 +1,33 @@
+#!/bin/sh
+#
+# Copyright 2007-2009 Canonical Ltd.
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU General Public License as
+#  published by the Free Software Foundation; either version 2 of the
+#  License, or (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software Foundation,
+#  Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+
+for device in $(hal-find-by-capability --capability storage); do
+    VENDOR=$(hal-get-property --udi $device --key storage.vendor)
+    NAME=$(hal-get-property --udi $device --key storage.model)
+    PLUGGABLE=$(hal-get-property --udi $device --key storage.hotpluggable)
+    TYPE=$(hal-get-property --udi $device --key storage.drive_type)
+    DEVPATH=$(hal-get-property --udi $device --key block.device)
+    AVAIL=$(hal-get-property --udi $device --key storage.removable.media_available)
+
+    if [ "${AVAIL}" = true ] && [ "${PLUGGABLE}" = true ] && \
+        ( [ "${TYPE}" = "disk" ] || [ "${TYPE}" = "sd_mmc" ] ); then
+        VENDOR=${VENDOR:-"Unknown"}
+        NAME=${NAME:-"Unknown"}
+        echo "$VENDOR $NAME, $DEVPATH"
+    fi
+done
diff --git a/lib/isodumper.py b/lib/isodumper.py
new file mode 100755
index 0000000..b891cb2
--- /dev/null
+++ b/lib/isodumper.py
@@ -0,0 +1,279 @@
+#!/usr/bin/python
+#  
+#  Copyright (c) 2007-2009 Canonical Ltd.
+#  
+#  Author: Oliver Grawert <ogra@ubuntu.com>
+# 
+#   Modifications 2013 from papoteur <papoteur@mageialinux-online.org>
+#   and Geiger David <david.david@mageialinux-online.org>
+#  This program is free software; you can redistribute it and/or 
+#  modify it under the terms of the GNU General Public License as 
+#  published by the Free Software Foundation; either version 2 of the
+#  License, or (at your option) any later version.
+# 
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+# 
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software Foundation,
+#  Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
+
+import gtk
+import gtk.glade
+import gobject
+from subprocess import Popen,PIPE,call
+import os
+import signal
+
+import gettext
+from gettext import gettext as _
+
+
+class ImageWriter:
+    def __init__(self):
+        APP="isodumper"
+        DIR="/usr/share/locale"
+
+        gettext.bindtextdomain(APP, DIR)
+        gettext.textdomain(APP)
+        gtk.glade.bindtextdomain(APP, DIR)
+        gtk.glade.textdomain(APP)
+
+        # get glade tree
+        self.gladefile = "/documents/isodumper-0.1.0/share/isodumper/isodumper.glade"
+        self.wTree = gtk.glade.XML(self.gladefile)
+
+        # get globally needed widgets
+        self.window = self.wTree.get_widget("main_dialog")
+        self.devicelist = self.wTree.get_widget("device_combobox")
+        self.logview = self.wTree.get_widget("detail_text")
+        #eventbox = self.wTree.get_widget("eventbox")
+        #bgcol = gtk.gdk.color_parse("white")
+        #eventbox.modify_bg(gtk.STATE_NORMAL, bgcol)
+        self.log = self.logview.get_buffer()
+        self.ddpid = 0
+
+        # set default file filter to *.img
+        # Added for Mageia : *.iso
+        self.chooser = self.wTree.get_widget("filechooserbutton")
+        filt = gtk.FileFilter()
+        filt.add_pattern("*.iso")
+        filt.add_pattern("*.img")
+        self.chooser.set_filter(filt)
+
+        # set callbacks
+        dict = { "on_main_dialog_destroy" : self.close,
+                 "on_cancel_button_clicked" : self.close,
+                 "on_emergency_button_clicked" : self.close,
+                 "on_success_button_clicked" : self.close,
+                 "on_filechooserbutton_file_set" : self.activate_devicelist,
+                 "on_detail_expander_activate" : self.expander_control,
+                 "on_device_combobox_changed" : self.device_selected,
+                 "on_button1_clicked" : self.close,
+                 "on_write_button_clicked" : self.do_write}
+        self.wTree.signal_autoconnect(dict)
+
+        self.window.show_all()
+        # make sure we have a target device
+        self.get_devices()
+
+    def get_devices(self):
+        dialog = self.wTree.get_widget("nodev_dialog")
+        list = Popen(["/usr/lib/isodumper/find_devices"], stdout=PIPE).communicate()[0]
+        if not len(list):
+            dialog.run()
+        list = Popen(["/usr/lib/isodumper/find_devices"], stdout=PIPE).communicate()[0]
+        if not len(list):
+            dialog.destroy()
+            self.close
+            exit(0)
+        self.combo = self.wTree.get_widget("device_combobox")
+        list = list.strip().split('\n')
+        for item in list:
+            name,path = item.split(',')
+            self.combo.append_text(name+' ('+path.lstrip()+')')
+        dialog.destroy()
+
+    def device_selected(self, widget):
+        write_button = self.wTree.get_widget("write_button")
+        write_button.set_sensitive(True)
+        self.dev = self.combo.get_active_text()
+
+    def do_write(self, widget):
+        write_button = self.wTree.get_widget("write_button")
+        write_button.set_sensitive(False)
+        combo = self.wTree.get_widget("device_combobox")
+        combo.set_sensitive(False)
+        self.chooser.set_sensitive(False)
+        source = self.chooser.get_filename()
+        target = self.dev.split('(')[1].rstrip(')')
+        dialog = self.wTree.get_widget("confirm_dialog")
+        self.logger(_('Image: ')+source)
+        self.logger(_('Target Device: ')+self.dev)
+        resp = dialog.run()
+        if resp:
+            self.do_umount(target)
+            dialog.hide()
+            task = self.raw_write(source, target)
+            gobject.idle_add(task.next)
+            while gtk.events_pending():
+                gtk.main_iteration(True)
+            
+        else:
+            self.close('dummy')
+
+    def do_umount(self, target):
+        mounts = self.get_mounted(target)
+        if mounts:
+            self.logger(_('Unmounting all partitions of ')+target+':')
+        for mount in mounts:
+            self.logger(_('Trying to unmount ')+mount[0]+'...')
+            while gtk.events_pending():
+                gtk.main_iteration(True)
+            try:
+                retcode = call('umount '+mount[0], shell=True)
+                if retcode < 0:
+                    self.logger(_('Error, umount ')+mount[0]+_(' was terminated by signal ')+str(retcode))
+                    self.emergency()
+                else:
+                    if retcode == 0:
+                        self.logger(mount[0]+_(' successfully unmounted'))
+                    else:
+                        self.logger(_('Error, umount ')+mount[0]+_(' returned ')+str(retcode))
+                        self.emergency()
+            except OSError, e:
+                self.logger(_('Execution failed: ')+str(e))
+                self.emergency()
+
+    def get_mounted(self, target):
+        try:
+            lines = [line.strip("\n").split(" ") for line in open ("/etc/mtab", "r").readlines()]
+            return [mount for mount in lines if mount[0].startswith(target)]
+        except:
+             self.logger(_('Could not read mtab !'))
+             self.emergency()
+
+    def raw_write(self, source, target):
+        bs=4096*128
+        b = os.path.getsize(source)
+        try:
+            ifc=open(source, "rb",1)
+        except:
+             self.logger(_('Reading error.'))
+             self.emergency()
+        else:
+            try:
+                ofc= open(target, 'wb',0)
+            except:
+                 self.logger(_('You have not the rights for writing on the device'))
+                 self.emergency()
+            else:
+                progress = self.wTree.get_widget("progressbar")
+                progress.set_sensitive(True)
+                progress.set_text(_('Writing ')+source.split('/')[-1]+_(' to ')+self.dev)
+                self.logger(_('Executing copy from ')+source+' to '+target)
+                while gtk.events_pending():
+                   gtk.main_iteration(True)
+                steps=range(0, b+1, b/100)
+                indice=1
+                written=0
+                ncuts=b/bs
+                for i in xrange(0,ncuts):
+                    try:
+                        buf=ifc.read(bs)
+                    except:
+                        self.logger(_("Reading error."))
+                        self.emergency()
+                    try:
+                        ofc.write(buf)
+                    except:
+                        self.logger(_("Writing error."))
+                        self.emergency()
+                    written= written+bs
+                    if written > steps[indice]:
+                        if indice%5==0:
+                            self.logger(_('Wrote: ')+str(indice)+'% '+str(written)+' bytes')
+                            mark = self.log.create_mark("end", self.log.get_end_iter(), False)
+                            self.logview.scroll_to_mark(mark, 0.05, True, 0.0, 1.0)
+                        progress.set_fraction(float(indice)/100)
+                        indice +=1
+                        os.fsync(ofc)
+                        yield True
+                progress.set_fraction(1.0)
+                self.logger(_('Image ')+source.split('/')[-1]+_(' successfully written to')+target)
+                self.logger(_('Bytes written: ')+str(written))
+                ofc.close()
+                self.success()
+            ifc.close()
+            yield False
+
+    def success(self):
+        dialog = self.wTree.get_widget("success_dialog")
+        self.final_unsensitive()
+        resp = dialog.run()
+        if resp:
+            exit(0)
+            dialog.destroy()
+
+    def emergency(self):
+        self.final_unsensitive()
+        dialog = self.wTree.get_widget("emergency_dialog")
+        expander = self.wTree.get_widget("detail_expander")
+        expander.set_expanded(True)
+        mark = self.log.create_mark("end", self.log.get_end_iter(), False)
+        self.logview.scroll_to_mark(mark, 0.05, True, 0.0, 1.0)
+        resp = dialog.run()
+        if resp:
+            dialog.destroy()
+
+    def final_unsensitive(self):
+        self.chooser.set_sensitive(False)
+        self.devicelist.set_sensitive(False)
+        write_button = self.wTree.get_widget("write_button")
+        write_button.set_sensitive(False)
+        progress = self.wTree.get_widget("progressbar")
+        progress.set_sensitive(False)
+
+    def close(self, widget):
+        self.write_logfile()
+        if self.ddpid > 0:
+            try:
+                os.killpg(os.getpgid(self.ddpid), signal.SIGKILL)
+            except:
+                gtk.main_quit()
+        else:
+            gtk.main_quit()
+
+    def write_logfile(self):
+        start = self.log.get_start_iter()
+        end = self.log.get_end_iter()
+        from os.path import expanduser
+        home = expanduser("~")
+        if not(os.path.isdir(home+'/.isodumper')):
+            os.mkdir(home+'/.isodumper')
+        logfile=open(home+'/.isodumper/isodumper.log',"w")
+        logfile.write(self.log.get_text(start, end, False))
+        print self.log.get_text(start, end, False)
+
+    def logger(self, text):
+        self.log.insert_at_cursor(text+"\n")
+
+    def activate_devicelist(self, widget):
+        label = self.wTree.get_widget("to_label")
+        expander = self.wTree.get_widget("detail_expander")
+        self.devicelist.set_sensitive(True)
+        expander.set_sensitive(True)
+        label.set_sensitive(True)
+        self.img_name = self.chooser.get_filename()
+
+    def expander_control(self, widget):
+        # this is darn ugly but still better than the UI behavior of
+        # the unexpanded expander which doesnt reset the window size
+        if widget.get_expanded():
+            gobject.timeout_add(130, lambda: self.window.reshow_with_initial_size())
+
+if __name__ == "__main__":
+    app = ImageWriter()
+    gtk.main()