[BACK]Return to ssh-keyscan.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / ssh

Diff for /src/usr.bin/ssh/ssh-keyscan.c between version 1.81 and 1.82

version 1.81, 2010/01/09 23:04:13 version 1.82, 2010/06/22 04:54:30
Line 99 
Line 99 
 TAILQ_HEAD(conlist, Connection) tq;     /* Timeout Queue */  TAILQ_HEAD(conlist, Connection) tq;     /* Timeout Queue */
 con *fdcon;  con *fdcon;
   
 /*  
  *  This is just a wrapper around fgets() to make it usable.  
  */  
   
 /* Stress-test.  Increase this later. */  
 #define LINEBUF_SIZE 16  
   
 typedef struct {  
         char *buf;  
         u_int size;  
         int lineno;  
         const char *filename;  
         FILE *stream;  
         void (*errfun) (const char *,...);  
 } Linebuf;  
   
 static Linebuf *  
 Linebuf_alloc(const char *filename, void (*errfun) (const char *,...))  
 {  
         Linebuf *lb;  
   
         if (!(lb = malloc(sizeof(*lb)))) {  
                 if (errfun)  
                         (*errfun) ("linebuf (%s): malloc failed\n",  
                             filename ? filename : "(stdin)");  
                 return (NULL);  
         }  
         if (filename) {  
                 lb->filename = filename;  
                 if (!(lb->stream = fopen(filename, "r"))) {  
                         xfree(lb);  
                         if (errfun)  
                                 (*errfun) ("%s: %s\n", filename, strerror(errno));  
                         return (NULL);  
                 }  
         } else {  
                 lb->filename = "(stdin)";  
                 lb->stream = stdin;  
         }  
   
         if (!(lb->buf = malloc((lb->size = LINEBUF_SIZE)))) {  
                 if (errfun)  
                         (*errfun) ("linebuf (%s): malloc failed\n", lb->filename);  
                 xfree(lb);  
                 return (NULL);  
         }  
         lb->errfun = errfun;  
         lb->lineno = 0;  
         return (lb);  
 }  
   
 static void  
 Linebuf_free(Linebuf * lb)  
 {  
         fclose(lb->stream);  
         xfree(lb->buf);  
         xfree(lb);  
 }  
   
 #if 0  
 static void  
 Linebuf_restart(Linebuf * lb)  
 {  
         clearerr(lb->stream);  
         rewind(lb->stream);  
         lb->lineno = 0;  
 }  
   
 static int  static int
 Linebuf_lineno(Linebuf * lb)  
 {  
         return (lb->lineno);  
 }  
 #endif  
   
 static char *  
 Linebuf_getline(Linebuf * lb)  
 {  
         size_t n = 0;  
         void *p;  
   
         lb->lineno++;  
         for (;;) {  
                 /* Read a line */  
                 if (!fgets(&lb->buf[n], lb->size - n, lb->stream)) {  
                         if (ferror(lb->stream) && lb->errfun)  
                                 (*lb->errfun)("%s: %s\n", lb->filename,  
                                     strerror(errno));  
                         return (NULL);  
                 }  
                 n = strlen(lb->buf);  
   
                 /* Return it or an error if it fits */  
                 if (n > 0 && lb->buf[n - 1] == '\n') {  
                         lb->buf[n - 1] = '\0';  
                         return (lb->buf);  
                 }  
                 if (n != lb->size - 1) {  
                         if (lb->errfun)  
                                 (*lb->errfun)("%s: skipping incomplete last line\n",  
                                     lb->filename);  
                         return (NULL);  
                 }  
                 /* Double the buffer if we need more space */  
                 lb->size *= 2;  
                 if ((p = realloc(lb->buf, lb->size)) == NULL) {  
                         lb->size /= 2;  
                         if (lb->errfun)  
                                 (*lb->errfun)("linebuf (%s): realloc failed\n",  
                                     lb->filename);  
                         return (NULL);  
                 }  
                 lb->buf = p;  
         }  
 }  
   
 static int  
 fdlim_get(int hard)  fdlim_get(int hard)
 {  {
         struct rlimit rlfd;          struct rlimit rlfd;
Line 709 
Line 593 
 main(int argc, char **argv)  main(int argc, char **argv)
 {  {
         int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO;          int debug_flag = 0, log_level = SYSLOG_LEVEL_INFO;
         int opt, fopt_count = 0;          int opt, fopt_count = 0, j;
         char *tname;          char *tname, *cp, line[NI_MAXHOST];
           FILE *fp;
           u_long linenum;
   
         extern int optind;          extern int optind;
         extern char *optarg;          extern char *optarg;
Line 808 
Line 694 
         read_wait_nfdset = howmany(maxfd, NFDBITS);          read_wait_nfdset = howmany(maxfd, NFDBITS);
         read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask));          read_wait = xcalloc(read_wait_nfdset, sizeof(fd_mask));
   
         if (fopt_count) {          for (j = 0; j < fopt_count; j++) {
                 Linebuf *lb;                  if (argv[j] == NULL)
                 char *line;                          fp = stdin;
                 int j;                  else if ((fp = fopen(argv[j], "r")) == NULL)
                           fatal("%s: %s: %s", __progname, argv[j],
                               strerror(errno));
                   linenum = 0;
   
                 for (j = 0; j < fopt_count; j++) {                  while (read_keyfile_line(fp,
                         lb = Linebuf_alloc(argv[j], error);                      argv[j] == NULL ? "(stdin)" : argv[j], line, sizeof(line),
                         if (!lb)                      &linenum) != -1) {
                           /* Chomp off trailing whitespace and comments */
                           if ((cp = strchr(line, '#')) == NULL)
                                   cp = line + strlen(line) - 1;
                           while (cp >= line) {
                                   if (*cp == ' ' || *cp == '\t' ||
                                       *cp == '\n' || *cp == '#')
                                           *cp-- = '\0';
                                   else
                                           break;
                           }
   
                           /* Skip empty lines */
                           if (*line == '\0')
                                 continue;                                  continue;
                         while ((line = Linebuf_getline(lb)) != NULL)  
                                 do_host(line);                          do_host(line);
                         Linebuf_free(lb);  
                 }                  }
   
                   if (ferror(fp))
                           fatal("%s: %s: %s", __progname, argv[j],
                               strerror(errno));
   
                   fclose(fp);
         }          }
   
         while (optind < argc)          while (optind < argc)
Legend:
Removed from v.1.81  
changed lines
  Added in v.1.82