Annotation of embedaddon/readline/histfile.c, revision 1.1

1.1     ! misho       1: /* histfile.c - functions to manipulate the history file. */
        !             2: 
        !             3: /* Copyright (C) 1989-2010 Free Software Foundation, Inc.
        !             4: 
        !             5:    This file contains the GNU History Library (History), a set of
        !             6:    routines for managing the text of previously typed lines.
        !             7: 
        !             8:    History is free software: you can redistribute it and/or modify
        !             9:    it under the terms of the GNU General Public License as published by
        !            10:    the Free Software Foundation, either version 3 of the License, or
        !            11:    (at your option) any later version.
        !            12: 
        !            13:    History is distributed in the hope that it will be useful,
        !            14:    but WITHOUT ANY WARRANTY; without even the implied warranty of
        !            15:    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        !            16:    GNU General Public License for more details.
        !            17: 
        !            18:    You should have received a copy of the GNU General Public License
        !            19:    along with History.  If not, see <http://www.gnu.org/licenses/>.
        !            20: */
        !            21: 
        !            22: /* The goal is to make the implementation transparent, so that you
        !            23:    don't have to know what data types are used, just what functions
        !            24:    you can call.  I think I have done that. */
        !            25: 
        !            26: #define READLINE_LIBRARY
        !            27: 
        !            28: #if defined (__TANDEM)
        !            29: #  include <floss.h>
        !            30: #endif
        !            31: 
        !            32: #if defined (HAVE_CONFIG_H)
        !            33: #  include <config.h>
        !            34: #endif
        !            35: 
        !            36: #include <stdio.h>
        !            37: 
        !            38: #include <sys/types.h>
        !            39: #if ! defined (_MINIX) && defined (HAVE_SYS_FILE_H)
        !            40: #  include <sys/file.h>
        !            41: #endif
        !            42: #include "posixstat.h"
        !            43: #include <fcntl.h>
        !            44: 
        !            45: #if defined (HAVE_STDLIB_H)
        !            46: #  include <stdlib.h>
        !            47: #else
        !            48: #  include "ansi_stdlib.h"
        !            49: #endif /* HAVE_STDLIB_H */
        !            50: 
        !            51: #if defined (HAVE_UNISTD_H)
        !            52: #  include <unistd.h>
        !            53: #endif
        !            54: 
        !            55: #include <ctype.h>
        !            56: 
        !            57: #if defined (__EMX__)
        !            58: #  undef HAVE_MMAP
        !            59: #endif
        !            60: 
        !            61: #ifdef HISTORY_USE_MMAP
        !            62: #  include <sys/mman.h>
        !            63: 
        !            64: #  ifdef MAP_FILE
        !            65: #    define MAP_RFLAGS (MAP_FILE|MAP_PRIVATE)
        !            66: #    define MAP_WFLAGS (MAP_FILE|MAP_SHARED)
        !            67: #  else
        !            68: #    define MAP_RFLAGS MAP_PRIVATE
        !            69: #    define MAP_WFLAGS MAP_SHARED
        !            70: #  endif
        !            71: 
        !            72: #  ifndef MAP_FAILED
        !            73: #    define MAP_FAILED ((void *)-1)
        !            74: #  endif
        !            75: 
        !            76: #endif /* HISTORY_USE_MMAP */
        !            77: 
        !            78: /* If we're compiling for __EMX__ (OS/2) or __CYGWIN__ (cygwin32 environment
        !            79:    on win 95/98/nt), we want to open files with O_BINARY mode so that there
        !            80:    is no \n -> \r\n conversion performed.  On other systems, we don't want to
        !            81:    mess around with O_BINARY at all, so we ensure that it's defined to 0. */
        !            82: #if defined (__EMX__) || defined (__CYGWIN__)
        !            83: #  ifndef O_BINARY
        !            84: #    define O_BINARY 0
        !            85: #  endif
        !            86: #else /* !__EMX__ && !__CYGWIN__ */
        !            87: #  undef O_BINARY
        !            88: #  define O_BINARY 0
        !            89: #endif /* !__EMX__ && !__CYGWIN__ */
        !            90: 
        !            91: #include <errno.h>
        !            92: #if !defined (errno)
        !            93: extern int errno;
        !            94: #endif /* !errno */
        !            95: 
        !            96: #include "history.h"
        !            97: #include "histlib.h"
        !            98: 
        !            99: #include "rlshell.h"
        !           100: #include "xmalloc.h"
        !           101: 
        !           102: /* If non-zero, we write timestamps to the history file in history_do_write() */
        !           103: int history_write_timestamps = 0;
        !           104: 
        !           105: /* Does S look like the beginning of a history timestamp entry?  Placeholder
        !           106:    for more extensive tests. */
        !           107: #define HIST_TIMESTAMP_START(s)                (*(s) == history_comment_char && isdigit ((s)[1]) )
        !           108: 
        !           109: /* Return the string that should be used in the place of this
        !           110:    filename.  This only matters when you don't specify the
        !           111:    filename to read_history (), or write_history (). */
        !           112: static char *
        !           113: history_filename (filename)
        !           114:      const char *filename;
        !           115: {
        !           116:   char *return_val;
        !           117:   const char *home;
        !           118:   int home_len;
        !           119: 
        !           120:   return_val = filename ? savestring (filename) : (char *)NULL;
        !           121: 
        !           122:   if (return_val)
        !           123:     return (return_val);
        !           124:   
        !           125:   home = sh_get_env_value ("HOME");
        !           126: 
        !           127:   if (home == 0)
        !           128:     return (NULL);
        !           129:   else
        !           130:     home_len = strlen (home);
        !           131: 
        !           132:   return_val = (char *)xmalloc (2 + home_len + 8); /* strlen(".history") == 8 */
        !           133:   strcpy (return_val, home);
        !           134:   return_val[home_len] = '/';
        !           135: #if defined (__MSDOS__)
        !           136:   strcpy (return_val + home_len + 1, "_history");
        !           137: #else
        !           138:   strcpy (return_val + home_len + 1, ".history");
        !           139: #endif
        !           140: 
        !           141:   return (return_val);
        !           142: }
        !           143: 
        !           144: static char *
        !           145: history_backupfile (filename)
        !           146:      const char *filename;
        !           147: {
        !           148:   char *ret;
        !           149:   size_t len;
        !           150: 
        !           151:   len = strlen (filename);
        !           152:   ret = xmalloc (len + 2);
        !           153:   strcpy (ret, filename);
        !           154:   ret[len] = '-';
        !           155:   ret[len+1] = '\0';
        !           156:   return ret;
        !           157: }
        !           158:   
        !           159: /* Add the contents of FILENAME to the history list, a line at a time.
        !           160:    If FILENAME is NULL, then read from ~/.history.  Returns 0 if
        !           161:    successful, or errno if not. */
        !           162: int
        !           163: read_history (filename)
        !           164:      const char *filename;
        !           165: {
        !           166:   return (read_history_range (filename, 0, -1));
        !           167: }
        !           168: 
        !           169: /* Read a range of lines from FILENAME, adding them to the history list.
        !           170:    Start reading at the FROM'th line and end at the TO'th.  If FROM
        !           171:    is zero, start at the beginning.  If TO is less than FROM, read
        !           172:    until the end of the file.  If FILENAME is NULL, then read from
        !           173:    ~/.history.  Returns 0 if successful, or errno if not. */
        !           174: int
        !           175: read_history_range (filename, from, to)
        !           176:      const char *filename;
        !           177:      int from, to;
        !           178: {
        !           179:   register char *line_start, *line_end, *p;
        !           180:   char *input, *buffer, *bufend, *last_ts;
        !           181:   int file, current_line, chars_read;
        !           182:   struct stat finfo;
        !           183:   size_t file_size;
        !           184: #if defined (EFBIG)
        !           185:   int overflow_errno = EFBIG;
        !           186: #elif defined (EOVERFLOW)
        !           187:   int overflow_errno = EOVERFLOW;
        !           188: #else
        !           189:   int overflow_errno = EIO;
        !           190: #endif
        !           191: 
        !           192:   buffer = last_ts = (char *)NULL;
        !           193:   input = history_filename (filename);
        !           194:   file = input ? open (input, O_RDONLY|O_BINARY, 0666) : -1;
        !           195: 
        !           196:   if ((file < 0) || (fstat (file, &finfo) == -1))
        !           197:     goto error_and_exit;
        !           198: 
        !           199:   file_size = (size_t)finfo.st_size;
        !           200: 
        !           201:   /* check for overflow on very large files */
        !           202:   if (file_size != finfo.st_size || file_size + 1 < file_size)
        !           203:     {
        !           204:       errno = overflow_errno;
        !           205:       goto error_and_exit;
        !           206:     }
        !           207: 
        !           208: #ifdef HISTORY_USE_MMAP
        !           209:   /* We map read/write and private so we can change newlines to NULs without
        !           210:      affecting the underlying object. */
        !           211:   buffer = (char *)mmap (0, file_size, PROT_READ|PROT_WRITE, MAP_RFLAGS, file, 0);
        !           212:   if ((void *)buffer == MAP_FAILED)
        !           213:     {
        !           214:       errno = overflow_errno;
        !           215:       goto error_and_exit;
        !           216:     }
        !           217:   chars_read = file_size;
        !           218: #else
        !           219:   buffer = (char *)malloc (file_size + 1);
        !           220:   if (buffer == 0)
        !           221:     {
        !           222:       errno = overflow_errno;
        !           223:       goto error_and_exit;
        !           224:     }
        !           225: 
        !           226:   chars_read = read (file, buffer, file_size);
        !           227: #endif
        !           228:   if (chars_read < 0)
        !           229:     {
        !           230:   error_and_exit:
        !           231:       if (errno != 0)
        !           232:        chars_read = errno;
        !           233:       else
        !           234:        chars_read = EIO;
        !           235:       if (file >= 0)
        !           236:        close (file);
        !           237: 
        !           238:       FREE (input);
        !           239: #ifndef HISTORY_USE_MMAP
        !           240:       FREE (buffer);
        !           241: #endif
        !           242: 
        !           243:       return (chars_read);
        !           244:     }
        !           245: 
        !           246:   close (file);
        !           247: 
        !           248:   /* Set TO to larger than end of file if negative. */
        !           249:   if (to < 0)
        !           250:     to = chars_read;
        !           251: 
        !           252:   /* Start at beginning of file, work to end. */
        !           253:   bufend = buffer + chars_read;
        !           254:   current_line = 0;
        !           255: 
        !           256:   /* Skip lines until we are at FROM. */
        !           257:   for (line_start = line_end = buffer; line_end < bufend && current_line < from; line_end++)
        !           258:     if (*line_end == '\n')
        !           259:       {
        !           260:        p = line_end + 1;
        !           261:        /* If we see something we think is a timestamp, continue with this
        !           262:           line.  We should check more extensively here... */
        !           263:        if (HIST_TIMESTAMP_START(p) == 0)
        !           264:          current_line++;
        !           265:        line_start = p;
        !           266:       }
        !           267: 
        !           268:   /* If there are lines left to gobble, then gobble them now. */
        !           269:   for (line_end = line_start; line_end < bufend; line_end++)
        !           270:     if (*line_end == '\n')
        !           271:       {
        !           272:        /* Change to allow Windows-like \r\n end of line delimiter. */
        !           273:        if (line_end > line_start && line_end[-1] == '\r')
        !           274:          line_end[-1] = '\0';
        !           275:        else
        !           276:          *line_end = '\0';
        !           277: 
        !           278:        if (*line_start)
        !           279:          {
        !           280:            if (HIST_TIMESTAMP_START(line_start) == 0)
        !           281:              {
        !           282:                add_history (line_start);
        !           283:                if (last_ts)
        !           284:                  {
        !           285:                    add_history_time (last_ts);
        !           286:                    last_ts = NULL;
        !           287:                  }
        !           288:              }
        !           289:            else
        !           290:              {
        !           291:                last_ts = line_start;
        !           292:                current_line--;
        !           293:              }
        !           294:          }
        !           295: 
        !           296:        current_line++;
        !           297: 
        !           298:        if (current_line >= to)
        !           299:          break;
        !           300: 
        !           301:        line_start = line_end + 1;
        !           302:       }
        !           303: 
        !           304:   FREE (input);
        !           305: #ifndef HISTORY_USE_MMAP
        !           306:   FREE (buffer);
        !           307: #else
        !           308:   munmap (buffer, file_size);
        !           309: #endif
        !           310: 
        !           311:   return (0);
        !           312: }
        !           313: 
        !           314: /* Truncate the history file FNAME, leaving only LINES trailing lines.
        !           315:    If FNAME is NULL, then use ~/.history.  Returns 0 on success, errno
        !           316:    on failure. */
        !           317: int
        !           318: history_truncate_file (fname, lines)
        !           319:      const char *fname;
        !           320:      int lines;
        !           321: {
        !           322:   char *buffer, *filename, *bp, *bp1;          /* bp1 == bp+1 */
        !           323:   int file, chars_read, rv;
        !           324:   struct stat finfo;
        !           325:   size_t file_size;
        !           326: 
        !           327:   buffer = (char *)NULL;
        !           328:   filename = history_filename (fname);
        !           329:   file = filename ? open (filename, O_RDONLY|O_BINARY, 0666) : -1;
        !           330:   rv = 0;
        !           331: 
        !           332:   /* Don't try to truncate non-regular files. */
        !           333:   if (file == -1 || fstat (file, &finfo) == -1)
        !           334:     {
        !           335:       rv = errno;
        !           336:       if (file != -1)
        !           337:        close (file);
        !           338:       goto truncate_exit;
        !           339:     }
        !           340: 
        !           341:   if (S_ISREG (finfo.st_mode) == 0)
        !           342:     {
        !           343:       close (file);
        !           344: #ifdef EFTYPE
        !           345:       rv = EFTYPE;
        !           346: #else
        !           347:       rv = EINVAL;
        !           348: #endif
        !           349:       goto truncate_exit;
        !           350:     }
        !           351: 
        !           352:   file_size = (size_t)finfo.st_size;
        !           353: 
        !           354:   /* check for overflow on very large files */
        !           355:   if (file_size != finfo.st_size || file_size + 1 < file_size)
        !           356:     {
        !           357:       close (file);
        !           358: #if defined (EFBIG)
        !           359:       rv = errno = EFBIG;
        !           360: #elif defined (EOVERFLOW)
        !           361:       rv = errno = EOVERFLOW;
        !           362: #else
        !           363:       rv = errno = EINVAL;
        !           364: #endif
        !           365:       goto truncate_exit;
        !           366:     }
        !           367: 
        !           368:   buffer = (char *)malloc (file_size + 1);
        !           369:   if (buffer == 0)
        !           370:     {
        !           371:       close (file);
        !           372:       goto truncate_exit;
        !           373:     }
        !           374: 
        !           375:   chars_read = read (file, buffer, file_size);
        !           376:   close (file);
        !           377: 
        !           378:   if (chars_read <= 0)
        !           379:     {
        !           380:       rv = (chars_read < 0) ? errno : 0;
        !           381:       goto truncate_exit;
        !           382:     }
        !           383: 
        !           384:   /* Count backwards from the end of buffer until we have passed
        !           385:      LINES lines.  bp1 is set funny initially.  But since bp[1] can't
        !           386:      be a comment character (since it's off the end) and *bp can't be
        !           387:      both a newline and the history comment character, it should be OK. */
        !           388:   for (bp1 = bp = buffer + chars_read - 1; lines && bp > buffer; bp--)
        !           389:     {
        !           390:       if (*bp == '\n' && HIST_TIMESTAMP_START(bp1) == 0)
        !           391:        lines--;
        !           392:       bp1 = bp;
        !           393:     }
        !           394: 
        !           395:   /* If this is the first line, then the file contains exactly the
        !           396:      number of lines we want to truncate to, so we don't need to do
        !           397:      anything.  It's the first line if we don't find a newline between
        !           398:      the current value of i and 0.  Otherwise, write from the start of
        !           399:      this line until the end of the buffer. */
        !           400:   for ( ; bp > buffer; bp--)
        !           401:     {
        !           402:       if (*bp == '\n' && HIST_TIMESTAMP_START(bp1) == 0)
        !           403:         {
        !           404:          bp++;
        !           405:          break;
        !           406:         }
        !           407:       bp1 = bp;
        !           408:     }
        !           409: 
        !           410:   /* Write only if there are more lines in the file than we want to
        !           411:      truncate to. */
        !           412:   if (bp > buffer && ((file = open (filename, O_WRONLY|O_TRUNC|O_BINARY, 0600)) != -1))
        !           413:     {
        !           414:       if (write (file, bp, chars_read - (bp - buffer)) < 0)
        !           415:        rv = errno;
        !           416: 
        !           417: #if defined (__BEOS__)
        !           418:       /* BeOS ignores O_TRUNC. */
        !           419:       ftruncate (file, chars_read - (bp - buffer));
        !           420: #endif
        !           421: 
        !           422:       if (close (file) < 0 && rv == 0)
        !           423:        rv = errno;
        !           424:     }
        !           425: 
        !           426:  truncate_exit:
        !           427: 
        !           428:   FREE (buffer);
        !           429: 
        !           430:   xfree (filename);
        !           431:   return rv;
        !           432: }
        !           433: 
        !           434: /* Workhorse function for writing history.  Writes NELEMENT entries
        !           435:    from the history list to FILENAME.  OVERWRITE is non-zero if you
        !           436:    wish to replace FILENAME with the entries. */
        !           437: static int
        !           438: history_do_write (filename, nelements, overwrite)
        !           439:      const char *filename;
        !           440:      int nelements, overwrite;
        !           441: {
        !           442:   register int i;
        !           443:   char *output, *bakname;
        !           444:   int file, mode, rv;
        !           445: #ifdef HISTORY_USE_MMAP
        !           446:   size_t cursize;
        !           447: 
        !           448:   mode = overwrite ? O_RDWR|O_CREAT|O_TRUNC|O_BINARY : O_RDWR|O_APPEND|O_BINARY;
        !           449: #else
        !           450:   mode = overwrite ? O_WRONLY|O_CREAT|O_TRUNC|O_BINARY : O_WRONLY|O_APPEND|O_BINARY;
        !           451: #endif
        !           452:   output = history_filename (filename);
        !           453:   bakname = (overwrite && output) ? history_backupfile (output) : 0;
        !           454: 
        !           455:   if (output && bakname)
        !           456:     rename (output, bakname);
        !           457: 
        !           458:   file = output ? open (output, mode, 0600) : -1;
        !           459:   rv = 0;
        !           460: 
        !           461:   if (file == -1)
        !           462:     {
        !           463:       rv = errno;
        !           464:       if (output && bakname)
        !           465:         rename (bakname, output);
        !           466:       FREE (output);
        !           467:       FREE (bakname);
        !           468:       return (rv);
        !           469:     }
        !           470: 
        !           471: #ifdef HISTORY_USE_MMAP
        !           472:   cursize = overwrite ? 0 : lseek (file, 0, SEEK_END);
        !           473: #endif
        !           474: 
        !           475:   if (nelements > history_length)
        !           476:     nelements = history_length;
        !           477: 
        !           478:   /* Build a buffer of all the lines to write, and write them in one syscall.
        !           479:      Suggested by Peter Ho (peter@robosts.oxford.ac.uk). */
        !           480:   {
        !           481:     HIST_ENTRY **the_history;  /* local */
        !           482:     register int j;
        !           483:     int buffer_size;
        !           484:     char *buffer;
        !           485: 
        !           486:     the_history = history_list ();
        !           487:     /* Calculate the total number of bytes to write. */
        !           488:     for (buffer_size = 0, i = history_length - nelements; i < history_length; i++)
        !           489: #if 0
        !           490:       buffer_size += 2 + HISTENT_BYTES (the_history[i]);
        !           491: #else
        !           492:       {
        !           493:        if (history_write_timestamps && the_history[i]->timestamp && the_history[i]->timestamp[0])
        !           494:          buffer_size += strlen (the_history[i]->timestamp) + 1;
        !           495:        buffer_size += strlen (the_history[i]->line) + 1;
        !           496:       }
        !           497: #endif
        !           498: 
        !           499:     /* Allocate the buffer, and fill it. */
        !           500: #ifdef HISTORY_USE_MMAP
        !           501:     if (ftruncate (file, buffer_size+cursize) == -1)
        !           502:       goto mmap_error;
        !           503:     buffer = (char *)mmap (0, buffer_size, PROT_READ|PROT_WRITE, MAP_WFLAGS, file, cursize);
        !           504:     if ((void *)buffer == MAP_FAILED)
        !           505:       {
        !           506: mmap_error:
        !           507:        rv = errno;
        !           508:        close (file);
        !           509:        if (output && bakname)
        !           510:          rename (bakname, output);
        !           511:        FREE (output);
        !           512:        FREE (bakname);
        !           513:        return rv;
        !           514:       }
        !           515: #else    
        !           516:     buffer = (char *)malloc (buffer_size);
        !           517:     if (buffer == 0)
        !           518:       {
        !           519:        rv = errno;
        !           520:        close (file);
        !           521:        if (output && bakname)
        !           522:          rename (bakname, output);
        !           523:        FREE (output);
        !           524:        FREE (bakname);
        !           525:        return rv;
        !           526:       }
        !           527: #endif
        !           528: 
        !           529:     for (j = 0, i = history_length - nelements; i < history_length; i++)
        !           530:       {
        !           531:        if (history_write_timestamps && the_history[i]->timestamp && the_history[i]->timestamp[0])
        !           532:          {
        !           533:            strcpy (buffer + j, the_history[i]->timestamp);
        !           534:            j += strlen (the_history[i]->timestamp);
        !           535:            buffer[j++] = '\n';
        !           536:          }
        !           537:        strcpy (buffer + j, the_history[i]->line);
        !           538:        j += strlen (the_history[i]->line);
        !           539:        buffer[j++] = '\n';
        !           540:       }
        !           541: 
        !           542: #ifdef HISTORY_USE_MMAP
        !           543:     if (msync (buffer, buffer_size, 0) != 0 || munmap (buffer, buffer_size) != 0)
        !           544:       rv = errno;
        !           545: #else
        !           546:     if (write (file, buffer, buffer_size) < 0)
        !           547:       rv = errno;
        !           548:     xfree (buffer);
        !           549: #endif
        !           550:   }
        !           551: 
        !           552:   if (close (file) < 0 && rv == 0)
        !           553:     rv = errno;
        !           554: 
        !           555:   if (rv != 0 && output && bakname)
        !           556:     rename (bakname, output);
        !           557:   else if (rv == 0 && bakname)
        !           558:     unlink (bakname);
        !           559: 
        !           560:   FREE (output);
        !           561:   FREE (bakname);
        !           562: 
        !           563:   return (rv);
        !           564: }
        !           565: 
        !           566: /* Append NELEMENT entries to FILENAME.  The entries appended are from
        !           567:    the end of the list minus NELEMENTs up to the end of the list. */
        !           568: int
        !           569: append_history (nelements, filename)
        !           570:      int nelements;
        !           571:      const char *filename;
        !           572: {
        !           573:   return (history_do_write (filename, nelements, HISTORY_APPEND));
        !           574: }
        !           575: 
        !           576: /* Overwrite FILENAME with the current history.  If FILENAME is NULL,
        !           577:    then write the history list to ~/.history.  Values returned
        !           578:    are as in read_history ().*/
        !           579: int
        !           580: write_history (filename)
        !           581:      const char *filename;
        !           582: {
        !           583:   return (history_do_write (filename, history_length, HISTORY_OVERWRITE));
        !           584: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>