diff --git a/lib/vty.c b/lib/vty.c
index a3f1a3c..3477f63 100644
--- a/lib/vty.c
+++ b/lib/vty.c
@@ -1703,6 +1703,7 @@
 /* create vty for stdio */
 static struct termios stdio_orig_termios;
 static struct vty *stdio_vty = NULL;
+static void (*stdio_vty_atclose)(void);
 
 static void
 vty_stdio_reset (void)
@@ -1711,11 +1712,15 @@
     {
       tcsetattr (0, TCSANOW, &stdio_orig_termios);
       stdio_vty = NULL;
+
+      if (stdio_vty_atclose)
+        stdio_vty_atclose ();
+      stdio_vty_atclose = NULL;
     }
 }
 
 struct vty *
-vty_stdio (void)
+vty_stdio (void (*atclose)())
 {
   struct vty *vty;
   struct termios termios;
@@ -1725,6 +1730,7 @@
     return NULL;
 
   vty = stdio_vty = vty_new_init (0);
+  stdio_vty_atclose = atclose;
   vty->wfd = 1;
 
   /* always have stdio vty in a known _unchangeable_ state, don't want config
diff --git a/lib/vty.h b/lib/vty.h
index abac3dd..806f2c6 100644
--- a/lib/vty.h
+++ b/lib/vty.h
@@ -237,7 +237,7 @@
 extern void vty_terminate (void);
 extern void vty_reset (void);
 extern struct vty *vty_new (void);
-extern struct vty *vty_stdio (void);
+extern struct vty *vty_stdio (void (*atclose)(void));
 extern int vty_out (struct vty *, const char *, ...) PRINTF_ATTRIBUTE(2, 3);
 extern void vty_read_config (char *, char *);
 extern void vty_time_print (struct vty *, int);