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->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         vi_addr spoffs;
 227
 228         /* make sure we have space for at least two new spans (split + add) */
 229         if(vb->num_spans + 1 >= vb->max_spans) {
 230                 int newmax = vb->max_spans > 0 ? (vb->max_spans << 1) : 16;
 231                 struct vi_span *tmp = vi_realloc(vb->spans, newmax * sizeof *tmp);
 232                 if(!tmp) return -1;
 233                 vb->spans = tmp;
 234                 vb->max_spans = newmax;
 235         }
 236
 237         if((sp = vi_buf_find_span(vb, at, &spoffs))) {
 238                 if(spoffs > 0) {
 239                         split_span(vb, sp++, spoffs, 1);
 240                 } else {
 241                         split_span(vb, sp++, 0, 0);
 242                 }
 243         } else {
 244                 sp = vb->spans + vb->num_spans;
 245         }
 246
 247         sp->src = src;
 248         sp->start = start;
 249         sp->size = size;
 250         vb->num_spans++;
 251         return 0;
 252 }
 253
 254 void vi_buf_reset(struct vi_buffer *vb)
 255 {
 256         struct visor *vi = vb->vi;
 257         struct vi_buffer *prev, *next;
 258
 259         vi_free(vb->path);
 260
 261         if(vb->fp) {
 262                 if(vb->file_mapped) vi_unmap(vb->fp);
 263                 vi_close(vb->fp);
 264         }
 265         vi_free(vb->orig);
 266         vi_free(vb->add);
 267         vi_free(vb->spans);
 268
 269         prev = vb->prev;
 270         next = vb->next;
 271         memset(vb, 0, sizeof *vb);
 272         vb->prev = prev;
 273         vb->next = next;
 274         vb->vi = vi;
 275 }
 276
 277 int vi_buf_read(struct vi_buffer *vb, const char *path)
 278 {
 279         struct visor *vi = vb->vi;
 280         vi_file *fp;
 281         unsigned long fsz;
 282         int plen;
 283
 284         vi_buf_reset(vb);
 285
 286         if(!(fp = vi_open(path, VI_RDONLY | VI_CREAT))) {
 287                 return -1;
 288         }
 289         plen = strlen(path);
 290         if(!(vb->path = vi_malloc(plen + 1))) {
 291                 vi_error(vi, "failed to allocate path name buffer\n");
 292                 vi_buf_reset(vb);
 293                 return -1;
 294         }
 295         memcpy(vb->path, path, plen + 1);
 296
 297         vb->num_spans = 0;
 298
 299         if((fsz = vi_size(fp))) {
 300                 /* existing file, map it into memory, or failing that read it */
 301                 if(!vi->fop.map || !(vb->orig = vi_map(fp))) {
 302                         if(!(vb->orig = vi_malloc(fsz))) {
 303                                 vi_buf_reset(vb);
 304                                 return -1;
 305                         }
 306                 } else {
 307                         vb->file_mapped = 1;
 308                 }
 309
 310                 if(add_span(vb, 0, SPAN_ORIG, 0, fsz) == -1) {
 311                         vi_error(vi, "failed to allocate span\n");
 312                         vi_buf_reset(vb);
 313                         return -1;
 314                 }
 315         }
 316         vb->orig_size = fsz;
 317         return 0;
 318 }
 319
 320 int vi_buf_write(struct vi_buffer *vb, const char *path)
 321 {
 322         int i, wbuf_count;
 323         struct visor *vi = vb->vi;
 324         vi_file *fp;
 325         static char wbuf[512];
 326
 327         if(!path) path = vb->path;
 328         if(!path) {
 329                 vi_error(vi, "failed to write buffer, unknown path\n");
 330                 return -1;
 331         }
 332
 333         if(!(fp = vi_open(path, VI_WRONLY | VI_CREAT))) {
 334                 vi_error(vi, "failed to open %s for writing\n", path);
 335                 return -1;
 336         }
 337
 338         wbuf_count = 0;
 339         for(i=0; i<vb->num_spans; i++) {
 340                 struct vi_span *sp = vb->spans + i;
 341                 const char *sptxt = vi_buf_span_text(vb, sp);
 342                 int n, count = 0;
 343                 while(count < sp->size) {
 344                         n = sp->size - count;
 345                         if(n > sizeof wbuf - wbuf_count) {
 346                                 n = sizeof wbuf - wbuf_count;
 347                         }
 348                         memcpy(wbuf + wbuf_count, sptxt + count, n);
 349                         count += n;
 350                         wbuf_count += n;
 351                 }
 352
 353                 if(wbuf_count >= sizeof wbuf) {
 354                         vi_write(fp, wbuf, wbuf_count);
 355                 }
 356         }
 357
 358         if(wbuf_count > 0) {
 359                 vi_write(fp, wbuf, wbuf_count);
 360         }
 361         vi_close(fp);
 362         return 0;
 363 }
 364
 365 long vi_buf_size(struct vi_buffer *vb)
 366 {
 367         int i;
 368         long sz = 0;
 369
 370         for(i=0; i<vb->num_spans; i++) {
 371                 sz += vb->spans[i].size;
 372         }
 373         return sz;
 374 }
 375
 376 struct vi_span *vi_buf_find_span(struct vi_buffer *vb, vi_addr at, vi_addr *soffs)
 377 {
 378         int i;
 379         long sz = 0, prev_sz;
 380
 381         for(i=0; i<vb->num_spans; i++) {
 382                 prev_sz = sz;
 383                 sz += vb->spans[i].size;
 384                 if(sz > at) {
 385                         if(soffs) *soffs = at - prev_sz;
 386                         return vb->spans + i;
 387                 }
 388         }
 389         return 0;
 390 }
 391
 392 const char *vi_buf_span_text(struct vi_buffer *vb, struct vi_span *sp)
 393 {
 394         const char *buf = sp->src == SPAN_ORIG ? vb->orig : vb->add;
 395         return buf + sp->start;
 396 }