Annotation of embedaddon/tmux/paste.c, revision 1.1.1.1

1.1       misho       1: /* $OpenBSD$ */
                      2: 
                      3: /*
                      4:  * Copyright (c) 2007 Nicholas Marriott <nicholas.marriott@gmail.com>
                      5:  *
                      6:  * Permission to use, copy, modify, and distribute this software for any
                      7:  * purpose with or without fee is hereby granted, provided that the above
                      8:  * copyright notice and this permission notice appear in all copies.
                      9:  *
                     10:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     11:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     12:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     13:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     14:  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
                     15:  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
                     16:  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     17:  */
                     18: 
                     19: #include <sys/types.h>
                     20: 
                     21: #include <stdlib.h>
                     22: #include <string.h>
                     23: #include <time.h>
                     24: 
                     25: #include "tmux.h"
                     26: 
                     27: /*
                     28:  * Set of paste buffers. Note that paste buffer data is not necessarily a C
                     29:  * string!
                     30:  */
                     31: 
                     32: struct paste_buffer {
                     33:        char            *data;
                     34:        size_t           size;
                     35: 
                     36:        char            *name;
                     37:        time_t           created;
                     38:        int              automatic;
                     39:        u_int            order;
                     40: 
                     41:        RB_ENTRY(paste_buffer) name_entry;
                     42:        RB_ENTRY(paste_buffer) time_entry;
                     43: };
                     44: 
                     45: static u_int   paste_next_index;
                     46: static u_int   paste_next_order;
                     47: static u_int   paste_num_automatic;
                     48: static RB_HEAD(paste_name_tree, paste_buffer) paste_by_name;
                     49: static RB_HEAD(paste_time_tree, paste_buffer) paste_by_time;
                     50: 
                     51: static int     paste_cmp_names(const struct paste_buffer *,
                     52:                    const struct paste_buffer *);
                     53: RB_GENERATE_STATIC(paste_name_tree, paste_buffer, name_entry, paste_cmp_names);
                     54: 
                     55: static int     paste_cmp_times(const struct paste_buffer *,
                     56:                    const struct paste_buffer *);
                     57: RB_GENERATE_STATIC(paste_time_tree, paste_buffer, time_entry, paste_cmp_times);
                     58: 
                     59: static int
                     60: paste_cmp_names(const struct paste_buffer *a, const struct paste_buffer *b)
                     61: {
                     62:        return (strcmp(a->name, b->name));
                     63: }
                     64: 
                     65: static int
                     66: paste_cmp_times(const struct paste_buffer *a, const struct paste_buffer *b)
                     67: {
                     68:        if (a->order > b->order)
                     69:                return (-1);
                     70:        if (a->order < b->order)
                     71:                return (1);
                     72:        return (0);
                     73: }
                     74: 
                     75: /* Get paste buffer name. */
                     76: const char *
                     77: paste_buffer_name(struct paste_buffer *pb)
                     78: {
                     79:        return (pb->name);
                     80: }
                     81: 
                     82: /* Get paste buffer order. */
                     83: u_int
                     84: paste_buffer_order(struct paste_buffer *pb)
                     85: {
                     86:        return (pb->order);
                     87: }
                     88: 
                     89: /* Get paste buffer created. */
                     90: time_t
                     91: paste_buffer_created(struct paste_buffer *pb)
                     92: {
                     93:        return (pb->created);
                     94: }
                     95: 
                     96: /* Get paste buffer data. */
                     97: const char *
                     98: paste_buffer_data(struct paste_buffer *pb, size_t *size)
                     99: {
                    100:        if (size != NULL)
                    101:                *size = pb->size;
                    102:        return (pb->data);
                    103: }
                    104: 
                    105: /* Walk paste buffers by time. */
                    106: struct paste_buffer *
                    107: paste_walk(struct paste_buffer *pb)
                    108: {
                    109:        if (pb == NULL)
                    110:                return (RB_MIN(paste_time_tree, &paste_by_time));
                    111:        return (RB_NEXT(paste_time_tree, &paste_by_time, pb));
                    112: }
                    113: 
                    114: /* Get the most recent automatic buffer. */
                    115: struct paste_buffer *
                    116: paste_get_top(const char **name)
                    117: {
                    118:        struct paste_buffer     *pb;
                    119: 
                    120:        pb = RB_MIN(paste_time_tree, &paste_by_time);
                    121:        if (pb == NULL)
                    122:                return (NULL);
                    123:        if (name != NULL)
                    124:                *name = pb->name;
                    125:        return (pb);
                    126: }
                    127: 
                    128: /* Get a paste buffer by name. */
                    129: struct paste_buffer *
                    130: paste_get_name(const char *name)
                    131: {
                    132:        struct paste_buffer     pbfind;
                    133: 
                    134:        if (name == NULL || *name == '\0')
                    135:                return (NULL);
                    136: 
                    137:        pbfind.name = (char *)name;
                    138:        return (RB_FIND(paste_name_tree, &paste_by_name, &pbfind));
                    139: }
                    140: 
                    141: /* Free a paste buffer. */
                    142: void
                    143: paste_free(struct paste_buffer *pb)
                    144: {
                    145:        RB_REMOVE(paste_name_tree, &paste_by_name, pb);
                    146:        RB_REMOVE(paste_time_tree, &paste_by_time, pb);
                    147:        if (pb->automatic)
                    148:                paste_num_automatic--;
                    149: 
                    150:        free(pb->data);
                    151:        free(pb->name);
                    152:        free(pb);
                    153: }
                    154: 
                    155: /*
                    156:  * Add an automatic buffer, freeing the oldest automatic item if at limit. Note
                    157:  * that the caller is responsible for allocating data.
                    158:  */
                    159: void
                    160: paste_add(char *data, size_t size)
                    161: {
                    162:        struct paste_buffer     *pb, *pb1;
                    163:        u_int                    limit;
                    164: 
                    165:        if (size == 0) {
                    166:                free(data);
                    167:                return;
                    168:        }
                    169: 
                    170:        limit = options_get_number(global_options, "buffer-limit");
                    171:        RB_FOREACH_REVERSE_SAFE(pb, paste_time_tree, &paste_by_time, pb1) {
                    172:                if (paste_num_automatic < limit)
                    173:                        break;
                    174:                if (pb->automatic)
                    175:                        paste_free(pb);
                    176:        }
                    177: 
                    178:        pb = xmalloc(sizeof *pb);
                    179: 
                    180:        pb->name = NULL;
                    181:        do {
                    182:                free(pb->name);
                    183:                xasprintf(&pb->name, "buffer%04u", paste_next_index);
                    184:                paste_next_index++;
                    185:        } while (paste_get_name(pb->name) != NULL);
                    186: 
                    187:        pb->data = data;
                    188:        pb->size = size;
                    189: 
                    190:        pb->automatic = 1;
                    191:        paste_num_automatic++;
                    192: 
                    193:        pb->created = time(NULL);
                    194: 
                    195:        pb->order = paste_next_order++;
                    196:        RB_INSERT(paste_name_tree, &paste_by_name, pb);
                    197:        RB_INSERT(paste_time_tree, &paste_by_time, pb);
                    198: }
                    199: 
                    200: /* Rename a paste buffer. */
                    201: int
                    202: paste_rename(const char *oldname, const char *newname, char **cause)
                    203: {
                    204:        struct paste_buffer     *pb, *pb_new;
                    205: 
                    206:        if (cause != NULL)
                    207:                *cause = NULL;
                    208: 
                    209:        if (oldname == NULL || *oldname == '\0') {
                    210:                if (cause != NULL)
                    211:                        *cause = xstrdup("no buffer");
                    212:                return (-1);
                    213:        }
                    214:        if (newname == NULL || *newname == '\0') {
                    215:                if (cause != NULL)
                    216:                        *cause = xstrdup("new name is empty");
                    217:                return (-1);
                    218:        }
                    219: 
                    220:        pb = paste_get_name(oldname);
                    221:        if (pb == NULL) {
                    222:                if (cause != NULL)
                    223:                        xasprintf(cause, "no buffer %s", oldname);
                    224:                return (-1);
                    225:        }
                    226: 
                    227:        pb_new = paste_get_name(newname);
                    228:        if (pb_new != NULL) {
                    229:                if (cause != NULL)
                    230:                        xasprintf(cause, "buffer %s already exists", newname);
                    231:                return (-1);
                    232:        }
                    233: 
                    234:        RB_REMOVE(paste_name_tree, &paste_by_name, pb);
                    235: 
                    236:        free(pb->name);
                    237:        pb->name = xstrdup(newname);
                    238: 
                    239:        if (pb->automatic)
                    240:                paste_num_automatic--;
                    241:        pb->automatic = 0;
                    242: 
                    243:        RB_INSERT(paste_name_tree, &paste_by_name, pb);
                    244: 
                    245:        return (0);
                    246: }
                    247: 
                    248: /*
                    249:  * Add or replace an item in the store. Note that the caller is responsible for
                    250:  * allocating data.
                    251:  */
                    252: int
                    253: paste_set(char *data, size_t size, const char *name, char **cause)
                    254: {
                    255:        struct paste_buffer     *pb, *old;
                    256: 
                    257:        if (cause != NULL)
                    258:                *cause = NULL;
                    259: 
                    260:        if (size == 0) {
                    261:                free(data);
                    262:                return (0);
                    263:        }
                    264:        if (name == NULL) {
                    265:                paste_add(data, size);
                    266:                return (0);
                    267:        }
                    268: 
                    269:        if (*name == '\0') {
                    270:                if (cause != NULL)
                    271:                        *cause = xstrdup("empty buffer name");
                    272:                return (-1);
                    273:        }
                    274: 
                    275:        pb = xmalloc(sizeof *pb);
                    276: 
                    277:        pb->name = xstrdup(name);
                    278: 
                    279:        pb->data = data;
                    280:        pb->size = size;
                    281: 
                    282:        pb->automatic = 0;
                    283:        pb->order = paste_next_order++;
                    284: 
                    285:        pb->created = time(NULL);
                    286: 
                    287:        if ((old = paste_get_name(name)) != NULL)
                    288:                paste_free(old);
                    289: 
                    290:        RB_INSERT(paste_name_tree, &paste_by_name, pb);
                    291:        RB_INSERT(paste_time_tree, &paste_by_time, pb);
                    292: 
                    293:        return (0);
                    294: }
                    295: 
                    296: /* Convert start of buffer into a nice string. */
                    297: char *
                    298: paste_make_sample(struct paste_buffer *pb)
                    299: {
                    300:        char            *buf;
                    301:        size_t           len, used;
                    302:        const int        flags = VIS_OCTAL|VIS_TAB|VIS_NL;
                    303:        const size_t     width = 200;
                    304: 
                    305:        len = pb->size;
                    306:        if (len > width)
                    307:                len = width;
                    308:        buf = xreallocarray(NULL, len, 4 + 4);
                    309: 
                    310:        used = utf8_strvis(buf, pb->data, len, flags);
                    311:        if (pb->size > width || used > width)
                    312:                strlcpy(buf + width, "...", 4);
                    313:        return (buf);
                    314: }

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