git.mutantstargoat.com Git - visor/blob - libvisor/src/visor.c

   1 /*
   2 visor - lightweight system-independent embeddable text editor framework
   3 Copyright (C)  2019 John Tsiombikas <nuclear@member.fsf.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU Lesser General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU Lesser General Public License for more details.
  14
  15 You should have received a copy of the GNU Lesser General Public License
  16 along with this program.  If not, see <https://www.gnu.org/licenses/>.
  17 */
  18
  19 #include "vilibc.h"
  20 #include "visor.h"
  21 #include "vimpl.h"
  22
  23 #define vi_malloc       vi->mm.malloc
  24 #define vi_free         vi->mm.free
  25 #define vi_realloc      vi->mm.realloc
  26
  27 #define vi_open         vi->fop.open
  28 #define vi_size         vi->fop.size
  29 #define vi_close        vi->fop.close
  30 #define vi_map          vi->fop.map
  31 #define vi_unmap        vi->fop.unmap
  32 #define vi_read         vi->fop.read
  33 #define vi_write        vi->fop.write
  34 #define vi_seek         vi->fop.seek
  35
  36 #ifdef HAVE_LIBC
  37 static const struct vi_alloc stdalloc = { malloc, free, realloc };
  38 #endif
  39
  40 struct visor *vi_create(struct vi_alloc *mm)
  41 {
  42         struct visor *vi;
  43
  44 #ifdef HAVE_LIBC
  45         if(!mm) mm = &stdalloc;
  46 #else
  47         if(!mm) return 0;
  48 #endif
  49
  50         if(!(vi = mm->malloc(sizeof *vi))) {
  51                 return 0;
  52         }
  53         memset(vi, 0, sizeof *vi);
  54         vi->mm = *mm;
  55
  56         return vi;
  57 }
  58
  59 void vi_destroy(struct visor *vi)
  60 {
  61         while(vi->buflist) {
  62                 vi_delete_buf(vi, vi->buflist);
  63         }
  64         vi_free(vi);
  65 }
  66
  67 void vi_set_fileops(struct visor *vi, struct vi_fileops *fop)
  68 {
  69         vi->fop = *fop;
  70 }
  71
  72 void vi_set_ttyops(struct visor *vi, struct vi_ttyops *tty)
  73 {
  74         vi->tty = *tty;
  75 }
  76
  77 struct vi_buffer *vi_new_buf(struct visor *vi, const char *path)
  78 {
  79         struct vi_buffer *nb;
  80
  81         if(!(nb = vi_malloc(sizeof *nb))) {
  82                 vi_error(vi, "failed to allocate new buffer\n");
  83                 return 0;
  84         }
  85         memset(nb, 0, sizeof *nb);
  86         nb->vi = vi;
  87
  88         if(path) {
  89                 if(vi_buf_read(nb, path) == -1) {
  90                         vi_free(nb);
  91                         return 0;
  92                 }
  93         }
  94
  95         if(vi->buflist) {
  96                 struct vi_buffer *last = vi->buflist->prev;
  97                 nb->prev = last;
  98                 nb->next = vi->buflist;
  99                 last->next = nb;
 100                 vi->buflist->prev = nb;
 101         } else {
 102                 nb->next = nb->prev = nb;
 103                 vi->buflist = nb;
 104         }
 105         return nb;
 106 }
 107
 108 static int remove_buf(struct visor *vi, struct vi_buffer *vb)
 109 {
 110         if(!vi->buflist) {
 111                 vi_error(vi, "failed to remove a buffer which doesn't exist\n");
 112                 return -1;
 113         }
 114
 115         if(vb->next == vb) {
 116                 if(vi->buflist != vb) {
 117                         vi_error(vi, "failed to remove buffer, buffer list inconsistency\n");
 118                         return -1;
 119                 }
 120                 vi->buflist = 0;
 121                 return 0;
 122         }
 123
 124         if(vi->buflist == vb) {
 125                 vi->buflist = vb->next;
 126         }
 127         vb->prev->next = vb->next;
 128         vb->next->prev = vb->prev;
 129         vb->next = vb->prev = vb;
 130         return 0;
 131 }
 132
 133 int vi_delete_buf(struct visor *vi, struct vi_buffer *vb)
 134 {
 135         if(remove_buf(vi, vb) == -1) {
 136                 return -1;
 137         }
 138
 139         vi_free(vb->path);
 140         vi_free(vb->orig);
 141         vi_free(vb->add);
 142         vi_free(vb->spans);
 143         return 0;
 144 }
 145
 146 int vi_num_buf(struct visor *vi)
 147 {
 148         int count;
 149         struct vi_buffer *vb;
 150
 151         if(!vi->buflist) return 0;
 152
 153         count = 1;
 154         vb = vi->buflist->next;
 155         while(vb != vi->buflist) {
 156                 count++;
 157                 vb = vb->next;
 158         }
 159         return count;
 160 }
 161
 162 struct vi_buffer *vi_getcur_buf(struct visor *vi)
 163 {
 164         return vi->buflist;
 165 }
 166
 167 void vi_setcur_buf(struct visor *vi, struct vi_buffer *vb)
 168 {
 169         vi->buflist = vb;
 170 }
 171
 172 struct vi_buffer *vi_next_buf(struct visor *vi)
 173 {
 174         return vi->buflist ? vi->buflist->next : 0;
 175 }
 176
 177 struct vi_buffer *vi_prev_buf(struct visor *vi)
 178 {
 179         return vi->buflist ? vi->buflist->prev : 0;
 180 }
 181
 182 /* split_span splits the span sp. if size > 0 it moves the second part to sp+2,
 183  * leaving an empty place at sp+1 for the new span. The start point of the
 184  * second part is adjusted by size.
 185  *
 186  * It can't fail, because it's always called with the span array having at
 187  * least two empty slots (see: add_span).
 188  */
 189 void split_span(struct vi_buffer *vb, struct vi_span *sp, vi_addr spoffs, unsigned long size)
 190 {
 191         int newseg = size > 0 ? 1 : 0;
 192         struct vi_span *tail = sp + newseg + 1;
 193         int num_move = vb->spans + vb->num_spans - sp - 1;
 194
 195         memmove(tail + 1, sp + 1, num_move * sizeof *sp);
 196         vb->num_spans += tail - sp;
 197
 198         *tail = *sp;
 199         sp->size = spoffs;
 200         tail->start += spoffs;
 201         tail->size -= spoffs;
 202
 203         sp = tail;
 204         for(;;) {
 205                 if(size <= tail->size) {
 206                         tail->size -= size;
 207                         break;
 208                 }
 209                 size -= tail->size;
 210                 tail->size = 0;
 211                 tail++;
 212         }
 213
 214         if(tail > sp) {
 215                 /* we produced one or more zero-sized spans, drop them */
 216                 num_move = vb->spans + vb->num_spans - (tail - sp);
 217                 memmove(sp, tail, num_move * sizeof *sp);
 218                 vb->num_spans -= num_move;
 219         }
 220 }
 221
 222 static int add_span(struct vi_buffer *vb, vi_addr at, int src, vi_addr start, unsigned long size)
 223 {
 224         struct visor *vi = vb->vi;
 225         struct vi_span *sp;
 226
 227         /* make sure we have space for at least two new spans (split + add) */
 228         if(vb->num_spans + 1 >= vb->max_spans) {
 229                 int newmax = vb->max_spans > 0 ? (vb->max_spans << 1) : 16;
 230                 struct vi_span *tmp = vi_realloc(vb->spans, newmax * sizeof *tmp);
 231                 if(!tmp) return -1;
 232                 vb->spans = tmp;
 233                 vb->max_spans = newmax;
 234         }
 235
 236         if((sp = vi_buf_find_span(vb, at, &spoffs))) {
 237                 if(spoffs > 0) {
 238                         split_span(vb, sp++, spoffs, 1);
 239                 } else {
 240                         split_span(vb, sp++, 0, 0);
 241                 }
 242         } else {
 243                 sp = vb->spans + vb->num_spans;
 244         }
 245
 246         sp->src = src;
 247         sp->start = start;
 248         sp->size = size;
 249         vb->num_spans++;
 250         return 0;
 251 }
 252
 253 void vi_buf_reset(struct vi_buffer *vb)
 254 {
 255         struct visor *vi = vb->vi;
 256         struct vi_buffer *prev, *next;
 257
 258         vi_free(vb->path);
 259
 260         if(vb->fp) {
 261                 if(vb->file_mapped) vi_unmap(vb->fp);
 262                 vi_close(vb->fp);
 263         }
 264         vi_free(vb->orig);
 265         vi_free(vb->add);
 266         vi_free(vb->spans);
 267
 268         prev = vb->prev;
 269         next = vb->next;
 270         memset(vb, 0, sizeof *vb);
 271         vb->prev = prev;
 272         vb->next = next;
 273 }
 274
 275 int vi_buf_read(struct vi_buffer *vb, const char *path)
 276 {
 277         struct visor *vi = vb->vi;
 278         vi_file *fp;
 279         unsigned long fsz;
 280         int plen;
 281
 282         vi_buf_reset(vb);
 283
 284         if(!(fp = vi_open(path, VI_RDONLY | VI_CREAT))) {
 285                 return -1;
 286         }
 287         plen = strlen(path);
 288         if(!(vb->path = vi_malloc(plen + 1))) {
 289                 vi_error(vi, "failed to allocate path name buffer\n");
 290                 vi_buf_reset(vb);
 291                 return -1;
 292         }
 293         memcpy(vb->path, path, plen + 1);
 294
 295         vb->num_spans = 0;
 296
 297         if((fsz = vi_size(fp))) {
 298                 /* existing file, map it into memory, or failing that read it */
 299                 if(!vi->fop.map || !(vb->orig = vi_map(fp))) {
 300                         if(!(vb->orig = vi_malloc(fsz))) {
 301                                 vi_buf_reset(vb);
 302                                 return -1;
 303                         }
 304                 } else {
 305                         vb->file_mapped = 1;
 306                 }
 307
 308                 if(add_span(vb, 0, SPAN_ORIG, 0, fsz) == -1) {
 309                         vi_error(vi, "failed to allocate span\n");
 310                         vi_buf_reset(vb);
 311                         return -1;
 312                 }
 313         }
 314         vb->orig_size = fsz;
 315         return 0;
 316 }
 317
 318 int vi_buf_write(struct vi_buffer *vb, const char *path)
 319 {
 320         int i, wbuf_count;
 321         struct visor *vi = vb->vi;
 322         vi_file *fp;
 323         static char wbuf[512];
 324
 325         if(!path) path = vb->path;
 326         if(!path) {
 327                 vi_error(vi, "failed to write buffer, unknown path\n");
 328                 return -1;
 329         }
 330
 331         if(!(fp = vi_open(path, VI_WRONLY | VI_CREAT))) {
 332                 vi_error(vi, "failed to open %s for writing\n", path);
 333                 return -1;
 334         }
 335
 336         wbuf_count = 0;
 337         for(i=0; i<vb->num_spans; i++) {
 338                 struct vi_span *sp = vb->spans + i;
 339                 const char *sptxt = vi_buf_span_text(vb, sp);
 340                 int n, count = 0;
 341                 while(count < sp->size) {
 342                         n = sp->size - count;
 343                         if(n > sizeof wbuf - wbuf_count) {
 344                                 n = sizeof wbuf - wbuf_count;
 345                         }
 346                         memcpy(wbuf + wbuf_count, sptxt + count, n);
 347                         count += n;
 348                         wbuf_count += n;
 349                 }
 350
 351                 if(wbuf_count >= sizeof wbuf) {
 352                         vi_write(fp, wbuf, wbuf_count);
 353                 }
 354         }
 355
 356         if(wbuf_count > 0) {
 357                 vi_write(fp, wbuf, wbuf_count);
 358         }
 359         vi_close(fp);
 360         return 0;
 361 }
 362
 363 long vi_buf_size(struct vi_buffer *vb)
 364 {
 365         int i;
 366         long sz = 0;
 367
 368         for(i=0; i<vb->num_spans; i++) {
 369                 sz += vb->spans[i].size;
 370         }
 371         return sz;
 372 }
 373
 374 struct vi_span *vi_buf_find_span(struct vi_buffer *vb, vi_addr at)
 375 {
 376         int i;
 377         long sz = 0;
 378
 379         for(i=0; i<vb->num_spans; i++) {
 380                 sz += vb->spans[i].size;
 381                 if(sz > at) {
 382                         return vb->spans + i;
 383                 }
 384         }
 385         return 0;
 386 }
 387
 388 const char *vi_buf_span_text(struct vi_buffer *vb, struct vi_span *sp)
 389 {
 390         const char *buf = sp->src == SPAN_ORIG ? vb->orig : vb->add;
 391         return buf + sp->start;
 392 }