src/core/monojob.c

   1 /*
   2  * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License as
   6  * published by the Free Software Foundation; either version 2 of the
   7  * License, or any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <string.h>
  20 #include <stdio.h>
  21 #include <errno.h>
  22 #include <gpxe/process.h>
  23 #include <console.h>
  24 #include <gpxe/keys.h>
  25 #include <gpxe/job.h>
  26 #include <gpxe/monojob.h>
  27 #include <gpxe/timer.h>
  28
  29 /** @file
  30  *
  31  * Single foreground job
  32  *
  33  */
  34
  35 static int monojob_rc;
  36
  37 static void monojob_done ( struct job_interface *job __unused, int rc ) {
  38         monojob_rc = rc;
  39 }
  40
  41 /** Single foreground job operations */
  42 static struct job_interface_operations monojob_operations = {
  43         .done           = monojob_done,
  44         .kill           = ignore_job_kill,
  45         .progress       = ignore_job_progress,
  46 };
  47
  48 /** Single foreground job */
  49 struct job_interface monojob = {
  50         .intf = {
  51                 .dest = &null_job.intf,
  52                 .refcnt = NULL,
  53         },
  54         .op = &monojob_operations,
  55 };
  56
  57 /**
  58  * Wait for single foreground job to complete
  59  *
  60  * @v string            Job description to display
  61  * @ret rc              Job final status code
  62  */
  63 int monojob_wait ( const char *string ) {
  64         int key;
  65         int rc;
  66         unsigned long last_progress_dot;
  67         unsigned long elapsed;
  68
  69         printf ( "%s.", string );
  70         monojob_rc = -EINPROGRESS;
  71         last_progress_dot = currticks();
  72         while ( monojob_rc == -EINPROGRESS ) {
  73                 step();
  74                 if ( iskey() ) {
  75                         key = getchar();
  76                         switch ( key ) {
  77                         case CTRL_C:
  78                                 job_kill ( &monojob );
  79                                 rc = -ECANCELED;
  80                                 goto done;
  81                         default:
  82                                 break;
  83                         }
  84                 }
  85                 elapsed = ( currticks() - last_progress_dot );
  86                 if ( elapsed >= TICKS_PER_SEC ) {
  87                         printf ( "." );
  88                         last_progress_dot = currticks();
  89                 }
  90         }
  91         rc = monojob_rc;
  92
  93 done:
  94         if ( rc ) {
  95                 printf ( " %s\n", strerror ( rc ) );
  96         } else {
  97                 printf ( " ok\n" );
  98         }
  99         return rc;
 100 }