~teejee2008/timeshift/trunk

 

/*

 * TimeoutCounter.vala

 *

 * Copyright 2016 Tony George <teejeetech@gmail.com>

 *

 * 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 Street, Fifth Floor, Boston,

 * MA 02110-1301, USA.

 *

 *

 */

 

 

using TeeJee.Logging;

using TeeJee.FileSystem;

using TeeJee.Misc;

 

public class TimeoutCounter : GLib.Object {

 

        public bool active = false;

        public string process_to_kill = "";

        public int seconds_to_wait = 30;

        public bool exit_app = false;

        

        public void kill_process_on_timeout(

                string process_to_kill, int seconds_to_wait = 20, bool exit_app = false){

 

                this.process_to_kill = process_to_kill;

                this.seconds_to_wait = seconds_to_wait;

                this.exit_app = exit_app;

                        

                try {

                        active = true;

                        Thread.create<void> (start_counter_thread, true);

                }

                catch (Error e) {

                        log_error (e.message);

                }

        }

 

        public void exit_on_timeout(int seconds_to_wait = 20){

                this.process_to_kill = "";

                this.seconds_to_wait = seconds_to_wait;

                this.exit_app = true;

                        

                try {

                        active = true;

                        Thread.create<void> (start_counter_thread, true);

                }

                catch (Error e) {

                        log_error (e.message);

                }

        }

 

        public void stop(){

                active = false;

        }

        

        public void start_counter_thread(){

                int secs = 0;

                

                while (active && (secs < seconds_to_wait)){

                        Thread.usleep((ulong) GLib.TimeSpan.MILLISECOND * 1000);

                        secs += 1;

                }

 

                if (active){

                        active = false;

                        stdout.printf("\n");

 

                        if (process_to_kill.length > 0){

                                Posix.system("killall " + process_to_kill);

                                stderr.printf("\n[timeout] Killed process" + ": %s\n".printf(process_to_kill));

                        }

 

                        if (exit_app){

                                stderr.printf("\n[timeout] Exit application\n");

                                exit(0);

                        }

                }

        }

}