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 at, unsigned long size)
 190 {
 191         struct vi_span *tail = sp + (size ? 2 : 1);
 192         int num_move = vb->spans + vb->num_spans - sp - 1;
 193
 194         memmove(tail + 1, sp + 1, num_move);
 195         vb->num_spans += tail - sp;
 196
 197 }
 198
 199 static int add_span(struct vi_buffer *vb, vi_addr at, int src, vi_addr start, unsigned long size)
 200 {
 201         struct visor *vi = vb->vi;
 202         struct vi_span *sp;
 203
 204         /* make sure we have space for at least two new spans (split + add) */
 205         if(vb->num_spans + 1 >= vb->max_spans) {
 206                 int newmax = vb->max_spans > 0 ? (vb->max_spans << 1) : 16;
 207                 struct vi_span *tmp = vi_realloc(vb->spans, newmax * sizeof *tmp);
 208                 if(!tmp) return -1;
 209                 vb->spans = tmp;
 210                 vb->max_spans = newmax;
 211         }
 212
 213         if((sp = vi_buf_find_span(vb, at))) {
 214                 split_span(vb, sp++, at, size);
 215         } else {
 216                 sp = vb->spans + vb->num_spans;
 217         }
 218
 219         sp->src = src;
 220         sp->start = start;
 221         sp->size = size;
 222         vb->num_spans++;
 223         return 0;
 224 }
 225
 226 void vi_buf_reset(struct vi_buffer *vb)
 227 {
 228         struct visor *vi = vb->vi;
 229         struct vi_buffer *prev, *next;
 230
 231         vi_free(vb->path);
 232
 233         if(vb->fp) {
 234                 if(vb->file_mapped) vi_unmap(vb->fp);
 235                 vi_close(vb->fp);
 236         }
 237         vi_free(vb->orig);
 238         vi_free(vb->add);
 239         vi_free(vb->spans);
 240
 241         prev = vb->prev;
 242         next = vb->next;
 243         memset(vb, 0, sizeof *vb);
 244         vb->prev = prev;
 245         vb->next = next;
 246 }
 247
 248 int vi_buf_read(struct vi_buffer *vb, const char *path)
 249 {
 250         struct visor *vi = vb->vi;
 251         vi_file *fp;
 252         unsigned long fsz;
 253         int plen;
 254
 255         vi_buf_reset(vb);
 256
 257         if(!(fp = vi_open(path, VI_RDONLY | VI_CREAT))) {
 258                 return -1;
 259         }
 260         plen = strlen(path);
 261         if(!(vb->path = vi_malloc(plen + 1))) {
 262                 vi_error(vi, "failed to allocate path name buffer\n");
 263                 vi_buf_reset(vb);
 264                 return -1;
 265         }
 266         memcpy(vb->path, path, plen + 1);
 267
 268         vb->num_spans = 0;
 269
 270         if((fsz = vi_size(fp))) {
 271                 /* existing file, map it into memory, or failing that read it */
 272                 if(!vi->fop.map || !(vb->orig = vi_map(fp))) {
 273                         if(!(vb->orig = vi_malloc(fsz))) {
 274                                 vi_buf_reset(vb);
 275                                 return -1;
 276                         }
 277                 } else {
 278                         vb->file_mapped = 1;
 279                 }
 280
 281                 if(add_span(vb, 0, SPAN_ORIG, 0, fsz) == -1) {
 282                         vi_error(vi, "failed to allocate span\n");
 283                         vi_buf_reset(vb);
 284                         return -1;
 285                 }
 286         }
 287         vb->orig_size = fsz;
 288         return 0;
 289 }
 290
 291 int vi_buf_write(struct vi_buffer *vb, const char *path)
 292 {
 293         int i, wbuf_count;
 294         struct visor *vi = vb->vi;
 295         vi_file *fp;
 296         static char wbuf[512];
 297
 298         if(!path) path = vb->path;
 299         if(!path) {
 300                 vi_error(vi, "failed to write buffer, unknown path\n");
 301                 return -1;
 302         }
 303
 304         if(!(fp = vi_open(path, VI_WRONLY | VI_CREAT))) {
 305                 vi_error(vi, "failed to open %s for writing\n", path);
 306                 return -1;
 307         }
 308
 309         wbuf_count = 0;
 310         for(i=0; i<vb->num_spans; i++) {
 311                 struct vi_span *sp = vb->spans + i;
 312                 const char *sptxt = vi_buf_span_text(vb, sp);
 313                 int n, count = 0;
 314                 while(count < sp->size) {
 315                         n = sp->size - count;
 316                         if(n > sizeof wbuf - wbuf_count) {
 317                                 n = sizeof wbuf - wbuf_count;
 318                         }
 319                         memcpy(wbuf + wbuf_count, sptxt + count, n);
 320                         count += n;
 321                         wbuf_count += n;
 322                 }
 323
 324                 if(wbuf_count >= sizeof wbuf) {
 325                         vi_write(fp, wbuf, wbuf_count);
 326                 }
 327         }
 328
 329         if(wbuf_count > 0) {
 330                 vi_write(fp, wbuf, wbuf_count);
 331         }
 332         vi_close(fp);
 333         return 0;
 334 }
 335
 336 long vi_buf_size(struct vi_buffer *vb)
 337 {
 338         int i;
 339         long sz = 0;
 340
 341         for(i=0; i<vb->num_spans; i++) {
 342                 sz += vb->spans[i].size;
 343         }
 344         return sz;
 345 }
 346
 347 struct vi_span *vi_buf_find_span(struct vi_buffer *vb, vi_addr at)
 348 {
 349         int i;
 350         long sz = 0;
 351
 352         for(i=0; i<vb->num_spans; i++) {
 353                 sz += vb->spans[i].size;
 354                 if(sz > at) {
 355                         return vb->spans + i;
 356                 }
 357         }
 358         return 0;
 359 }
 360
 361 const char *vi_buf_span_text(struct vi_buffer *vb, struct vi_span *sp)
 362 {
 363         const char *buf = sp->src == SPAN_ORIG ? vb->orig : vb->add;
 364         return buf + sp->start;
 365 }