shithub: masto9

ref: 45930f4521a779916b87d051065a8d2b751a7c12
dir: /util.c/

#include <u.h>
#include <libc.h>
#include <bio.h>
#include <thread.h>
#include <regexp.h>

#include "masto9.h"

void *
emalloc(ulong n)
{
	void *v;

	v = mallocz(n, 1);
	if(v == nil)
		sysfatal("malloc: %r");
	setmalloctag(v, getcallerpc(&n));
	return v;
}

void *
erealloc(void *p, ulong n)
{
	void *v;

	v = realloc(p, n);
	if(v == nil)
		sysfatal("realloc: %r");
	setmalloctag(v, getcallerpc(&p));
	return v;
}

char*
estrdup(char *s)
{
	s = strdup(s);
	if(s == nil)
		sysfatal("strdup: %r");
	setmalloctag(s, getcallerpc(&s));
	return s;
}

/* char* */
/* estrjoin(char *s, ...) */
/* { */
/* 	va_list ap; */
/* 	char *r, *t, *p, *e; */
/* 	int n; */

/* 	va_start(ap, s); */
/* 	n = strlen(s) + 1; */
/* 	while((p = va_arg(ap, char*)) != nil) */
/* 		n += strlen(p); */
/* 	va_end(ap); */

/* 	r = emalloc(n); */
/* 	e = r + n; */
/* 	va_start(ap, s); */
/* 	t = strecpy(r, e, s); */
/* 	while((p = va_arg(ap, char*)) != nil) */
/* 		t = strecpy(t, e, p); */
/* 	va_end(ap); */
/* 	return r; */
/* } */

char *
concat(char *s1, char *s2)
{
  char *result;
  result = emalloc(strlen(s1) + strlen(s2) + 1); // +1 for the null-terminator

  strcpy(result, s1);
  strcat(result, s2);
  return result;
}

char*
esmprint(char *fmt, ...)
{
	char *s;
	va_list ap;

	va_start(ap, fmt);
	s = vsmprint(fmt, ap);
	va_end(ap);
	if(s == nil)
		sysfatal("smprint: %r");
	setmalloctag(s, getcallerpc(&fmt));
	return s;
}

char*
fslurp(int fd, int *nbuf)
{
	int n, sz, r;
	char *buf;

	n = 0;
	sz = 128;
	buf = emalloc(sz);
	while(1){
		r = read(fd, buf + n, sz - n);
		if(r == 0)
			break;
		if(r == -1)
			goto error;
		n += r;
		if(n == sz){
			sz += sz/2;
			buf = erealloc(buf, sz);
		}
	}
	buf[n] = 0;
	if(nbuf)
		*nbuf = n;
	return buf;
error:
	free(buf);
	return nil;
}

u32int
strhash(char *s)
{
	u32int h, c;

	h = 5381;
	while(c = *s++ & 0xff)
		h = ((h << 5) + h) + c;
	return h;
}

void
remove_substring(char *str, char *sub)
{
  int len = strlen(sub);

  while ((str = strstr(str, sub))) {
    memmove(str, str + len, strlen(str + len) + 1);
  }
}

void
remove_tag(char *str, char *tag)
{
  char *start = strstr(str, tag);

  while (start) {
    char *end = strchr(start, '>');

    if (end) {
      memmove(start, end + 1, strlen(end + 1) + 1);
    } else {
      *start = '\0';
      break;
    }

    start = strstr(start, tag);
  }
}