Implemented a readline() function that will read an entire line into a
authorIvo Timmermans <ivo@lychnis.net>
Wed, 29 Nov 2000 14:24:40 +0000 (14:24 +0000)
committerIvo Timmermans <ivo@lychnis.net>
Wed, 29 Nov 2000 14:24:40 +0000 (14:24 +0000)
dynamically allocated buffer;

Ask for a file name in ask_and_safe_open().

src/conf.c
src/conf.h
src/tincd.c

index 37bfe9469639d3fd423ff57566ec97e2ccc96e99..7399c823a7dbe537e23b2dfeed4e0840092db23d 100644 (file)
@@ -19,7 +19,7 @@
     along with this program; if not, write to the Free Software
     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 
-    $Id: conf.c,v 1.9.4.23 2000/11/28 23:12:56 zarq Exp $
+    $Id: conf.c,v 1.9.4.24 2000/11/29 14:24:40 zarq Exp $
 */
 
 #include "config.h"
@@ -38,7 +38,6 @@
 #include "conf.h"
 #include "netutl.h" /* for strtoip */
 
-#include "config.h"
 #include "system.h"
 
 config_t *config = NULL;
@@ -130,6 +129,66 @@ cp
     }
 }
 
+/*
+  Read exactly one line and strip the trailing newline if any.  If the
+  file was on EOF, return NULL. Otherwise, return all the data in a
+  dynamically allocated buffer.
+*/
+char *readline(FILE *fp)
+{
+  char *newline = NULL;
+  char *p;
+  char *line; /* The array that contains everything that has been read
+                 so far */
+  char *idx; /* Read into this pointer, which points to an offset
+                within line */
+  size_t size, newsize; /* The size of the current array pointed to by
+                           line */
+  size_t maxlen; /* Maximum number of characters that may be read with
+                    fgets.  This is newsize - oldsize. */
+
+  if(feof(fp))
+    return NULL;
+  
+  size = 100;
+  maxlen = size;
+  line = xmalloc(size);
+  idx = line;
+  for(;;)
+    {
+      errno = 0;
+      p = fgets(idx, maxlen, fp);
+      if(p == NULL)  /* EOF or error */
+       {
+         if(feof(fp))
+           break;
+
+         /* otherwise: error; let the calling function print an error
+             message if applicable */
+         free(line);
+         return NULL;
+       }
+
+      newline = strchr(p, '\n');
+      if(newline == NULL)
+       /* We haven't yet read everything to the end of the line */
+       {
+         newsize = size << 1;
+         line = xrealloc(line, newsize);
+         idx = &line[size - 1];
+         maxlen = newsize - size + 1;
+         size = newsize;
+       }
+      else
+       {
+         *newline = '\0'; /* kill newline */
+         break;  /* yay */
+       }
+    }
+
+  return line;
+}
+
 /*
   Parse a configuration file and put the results in the configuration tree
   starting at *base.
@@ -258,28 +317,49 @@ cp
 
 #define is_safe_file(p) 1
 
-FILE *ask_and_safe_open(const char* filename)
+FILE *ask_and_safe_open(const char* filename, const char* what)
 {
   FILE *r;
   char *directory;
   char *fn;
   int len;
 
-  if(!isatty(0))
+  /* Check stdin and stdout */
+  if(!isatty(0) || !isatty(1))
     {
       /* Argh, they are running us from a script or something.  Write
          the files to the current directory and let them burn in hell
          for ever. */
-      directory = "."; /* get_current_directory */
+      fn = xstrdup(filename);
     }
   else
     {
-      directory = ".";
+      /* Ask for a file and/or directory name. */
+      fprintf(stdout, _("Please enter a file to save %s to [%s]: "),
+             what, filename);
+      fflush(stdout);  /* Don't wait for a newline */
+      if((fn = readline(stdin)) == NULL)
+       {
+         fprintf(stderr, _("Error while reading stdin: %m\n"));
+         return NULL;
+       }
+      if(strlen(fn) == 0)
+       /* User just pressed enter. */
+       fn = xstrdup(filename);
     }
 
-  len = strlen(filename) + strlen(directory) + 2; /* 1 for the / */
-  fn = xmalloc(len);
-  snprintf(fn, len, "%s/%s", directory, filename);
+  if((strchr(fn, '/') == NULL) || (fn[0] != '/'))
+    {
+      /* The directory is a relative path or a filename. */
+      char *p;
+      
+      directory = get_current_dir_name();
+      len = strlen(fn) + strlen(directory) + 2; /* 1 for the / */
+      p = xmalloc(len);
+      snprintf(p, len, "%s/%s", directory, fn);
+      free(fn);
+      fn = p;
+    }
 
   if(!is_safe_file(fn))
     {
@@ -291,11 +371,12 @@ FILE *ask_and_safe_open(const char* filename)
 
   if((r = fopen(fn, "w")) == NULL)
     {
-      fprintf(stderr, _("Error opening file `%s': %m"),
+      fprintf(stderr, _("Error opening file `%s': %m\n"),
              fn);
     }
 
   free(fn);
+  free(directory);
   
   return r;
 }
index 9ab0add63884ee8f6efe5b72007f365ae249b9be..297dc42b2838d931ac0db742c2964df1421911d1 100644 (file)
@@ -17,7 +17,7 @@
     along with this program; if not, write to the Free Software
     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 
-    $Id: conf.h,v 1.6.4.16 2000/11/04 22:57:30 guus Exp $
+    $Id: conf.h,v 1.6.4.17 2000/11/29 14:24:40 zarq Exp $
 */
 
 #ifndef __TINC_CONF_H__
@@ -96,5 +96,6 @@ extern int read_config_file(config_t **, const char *);
 extern const config_t *get_config_val(config_t *, which_t type);
 extern void clear_config();
 extern int read_server_config(void);
+extern FILE *ask_and_safe_open(const char*, const char*);
 
 #endif /* __TINC_CONF_H__ */
index 7a65aadae51658f88595ea5f982055ccd61739d9..77e52c07a94df08e9f8a84e2aadc9daddc86597f 100644 (file)
@@ -17,7 +17,7 @@
     along with this program; if not, write to the Free Software
     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 
-    $Id: tincd.c,v 1.10.4.36 2000/11/28 23:12:57 zarq Exp $
+    $Id: tincd.c,v 1.10.4.37 2000/11/29 14:24:40 zarq Exp $
 */
 
 #include "config.h"
@@ -236,12 +236,12 @@ int keygen(int bits)
   else
     fprintf(stderr, _("Done.\n"));
 
-  if((f = ask_and_safe_open("rsa_key.pub")) == NULL)
+  if((f = ask_and_safe_open("rsa_key.pub", _("public RSA key"))) == NULL)
     return -1;
   PEM_write_RSAPublicKey(f, rsa_key);
   fclose(f);
   
-  if((f = ask_and_safe_open("rsa_key.priv")) == NULL)
+  if((f = ask_and_safe_open("rsa_key.priv", _("private RSA key"))) == NULL)
     return -1;
   PEM_write_RSAPrivateKey(f, rsa_key, NULL, NULL, 0, NULL, NULL);
   fclose(f);