version 1.8, 2015/10/06 18:32:20 |
version 1.15, 2016/11/08 15:57:28 |
|
|
/* $Id$ */ |
/* $Id$ */ |
/* |
/* |
* Copyright (c) 2015 Ingo Schwarze <schwarze@openbsd.org> |
* Copyright (c) 2015, 2016 Ingo Schwarze <schwarze@openbsd.org> |
* |
* |
* Permission to use, copy, modify, and distribute this software for any |
* Permission to use, copy, modify, and distribute this software for any |
* purpose with or without fee is hereby granted, provided that the above |
* purpose with or without fee is hereby granted, provided that the above |
|
|
#include <string.h> |
#include <string.h> |
#include <unistd.h> |
#include <unistd.h> |
|
|
#if HAVE_OHASH |
|
#include <ohash.h> |
|
#else |
|
#include "compat_ohash.h" |
|
#endif |
|
|
|
#include "mandoc_aux.h" |
#include "mandoc_aux.h" |
|
#include "mandoc_ohash.h" |
#include "tag.h" |
#include "tag.h" |
|
|
struct tag_entry { |
struct tag_entry { |
size_t line; |
size_t *lines; |
|
size_t maxlines; |
|
size_t nlines; |
int prio; |
int prio; |
char s[]; |
char s[]; |
}; |
}; |
|
|
static void tag_signal(int); |
static void tag_signal(int) __attribute__((noreturn)); |
static void *tag_alloc(size_t, void *); |
|
static void tag_free(void *, void *); |
|
static void *tag_calloc(size_t, size_t, void *); |
|
|
|
static struct ohash tag_data; |
static struct ohash tag_data; |
static struct tag_files tag_files; |
static struct tag_files tag_files; |
Line 58 static struct tag_files tag_files; |
|
Line 52 static struct tag_files tag_files; |
|
struct tag_files * |
struct tag_files * |
tag_init(void) |
tag_init(void) |
{ |
{ |
struct ohash_info tag_info; |
struct sigaction sa; |
int ofd; |
int ofd; |
|
|
ofd = -1; |
ofd = -1; |
tag_files.tfd = -1; |
tag_files.tfd = -1; |
|
tag_files.tcpgid = -1; |
|
|
|
/* Clean up when dying from a signal. */ |
|
|
|
memset(&sa, 0, sizeof(sa)); |
|
sigfillset(&sa.sa_mask); |
|
sa.sa_handler = tag_signal; |
|
sigaction(SIGHUP, &sa, NULL); |
|
sigaction(SIGINT, &sa, NULL); |
|
sigaction(SIGTERM, &sa, NULL); |
|
|
|
/* |
|
* POSIX requires that a process calling tcsetpgrp(3) |
|
* from the background gets a SIGTTOU signal. |
|
* In that case, do not stop. |
|
*/ |
|
|
|
sa.sa_handler = SIG_IGN; |
|
sigaction(SIGTTOU, &sa, NULL); |
|
|
/* Save the original standard output for use by the pager. */ |
/* Save the original standard output for use by the pager. */ |
|
|
if ((tag_files.ofd = dup(STDOUT_FILENO)) == -1) |
if ((tag_files.ofd = dup(STDOUT_FILENO)) == -1) |
|
|
sizeof(tag_files.ofn)); |
sizeof(tag_files.ofn)); |
(void)strlcpy(tag_files.tfn, "/tmp/man.XXXXXXXXXX", |
(void)strlcpy(tag_files.tfn, "/tmp/man.XXXXXXXXXX", |
sizeof(tag_files.tfn)); |
sizeof(tag_files.tfn)); |
signal(SIGHUP, tag_signal); |
|
signal(SIGINT, tag_signal); |
|
signal(SIGTERM, tag_signal); |
|
if ((ofd = mkstemp(tag_files.ofn)) == -1) |
if ((ofd = mkstemp(tag_files.ofn)) == -1) |
goto fail; |
goto fail; |
if ((tag_files.tfd = mkstemp(tag_files.tfn)) == -1) |
if ((tag_files.tfd = mkstemp(tag_files.tfn)) == -1) |
|
|
* where various marked-up terms are documented. |
* where various marked-up terms are documented. |
*/ |
*/ |
|
|
tag_info.alloc = tag_alloc; |
mandoc_ohash_init(&tag_data, 4, offsetof(struct tag_entry, s)); |
tag_info.calloc = tag_calloc; |
|
tag_info.free = tag_free; |
|
tag_info.key_offset = offsetof(struct tag_entry, s); |
|
tag_info.data = NULL; |
|
ohash_init(&tag_data, 4, &tag_info); |
|
return &tag_files; |
return &tag_files; |
|
|
fail: |
fail: |
Line 125 tag_put(const char *s, int prio, size_t line) |
|
Line 130 tag_put(const char *s, int prio, size_t line) |
|
size_t len; |
size_t len; |
unsigned int slot; |
unsigned int slot; |
|
|
|
/* Sanity checks. */ |
|
|
if (tag_files.tfd <= 0) |
if (tag_files.tfd <= 0) |
return; |
return; |
|
if (s[0] == '\\' && (s[1] == '&' || s[1] == 'e')) |
|
s += 2; |
|
if (*s == '\0' || strchr(s, ' ') != NULL) |
|
return; |
|
|
slot = ohash_qlookup(&tag_data, s); |
slot = ohash_qlookup(&tag_data, s); |
entry = ohash_find(&tag_data, slot); |
entry = ohash_find(&tag_data, slot); |
|
|
if (entry == NULL) { |
if (entry == NULL) { |
|
|
|
/* Build a new entry. */ |
|
|
len = strlen(s) + 1; |
len = strlen(s) + 1; |
entry = mandoc_malloc(sizeof(*entry) + len); |
entry = mandoc_malloc(sizeof(*entry) + len); |
memcpy(entry->s, s, len); |
memcpy(entry->s, s, len); |
|
entry->lines = NULL; |
|
entry->maxlines = entry->nlines = 0; |
ohash_insert(&tag_data, slot, entry); |
ohash_insert(&tag_data, slot, entry); |
} else if (entry->prio <= prio) |
|
return; |
} else { |
entry->line = line; |
|
|
/* A better entry is already present, ignore the new one. */ |
|
|
|
if (entry->prio < prio) |
|
return; |
|
|
|
/* The existing entry is worse, clear it. */ |
|
|
|
if (entry->prio > prio) |
|
entry->nlines = 0; |
|
} |
|
|
|
/* Remember the new line. */ |
|
|
|
if (entry->maxlines == entry->nlines) { |
|
entry->maxlines += 4; |
|
entry->lines = mandoc_reallocarray(entry->lines, |
|
entry->maxlines, sizeof(*entry->lines)); |
|
} |
|
entry->lines[entry->nlines++] = line; |
entry->prio = prio; |
entry->prio = prio; |
} |
} |
|
|
|
|
{ |
{ |
FILE *stream; |
FILE *stream; |
struct tag_entry *entry; |
struct tag_entry *entry; |
|
size_t i; |
unsigned int slot; |
unsigned int slot; |
|
|
if (tag_files.tfd <= 0) |
if (tag_files.tfd <= 0) |
|
|
entry = ohash_first(&tag_data, &slot); |
entry = ohash_first(&tag_data, &slot); |
while (entry != NULL) { |
while (entry != NULL) { |
if (stream != NULL) |
if (stream != NULL) |
fprintf(stream, "%s %s %zu\n", |
for (i = 0; i < entry->nlines; i++) |
entry->s, tag_files.ofn, entry->line); |
fprintf(stream, "%s %s %zu\n", |
|
entry->s, tag_files.ofn, entry->lines[i]); |
|
free(entry->lines); |
free(entry); |
free(entry); |
entry = ohash_next(&tag_data, &slot); |
entry = ohash_next(&tag_data, &slot); |
} |
} |
|
|
void |
void |
tag_unlink(void) |
tag_unlink(void) |
{ |
{ |
|
pid_t tc_pgid; |
|
|
|
if (tag_files.tcpgid != -1) { |
|
tc_pgid = tcgetpgrp(STDIN_FILENO); |
|
if (tc_pgid == tag_files.pager_pid || |
|
tc_pgid == getpgid(0) || |
|
getpgid(tc_pgid) == -1) |
|
(void)tcsetpgrp(STDIN_FILENO, tag_files.tcpgid); |
|
} |
if (*tag_files.ofn != '\0') |
if (*tag_files.ofn != '\0') |
unlink(tag_files.ofn); |
unlink(tag_files.ofn); |
if (*tag_files.tfn != '\0') |
if (*tag_files.tfn != '\0') |
Line 180 tag_unlink(void) |
|
Line 228 tag_unlink(void) |
|
static void |
static void |
tag_signal(int signum) |
tag_signal(int signum) |
{ |
{ |
|
struct sigaction sa; |
|
|
tag_unlink(); |
tag_unlink(); |
signal(signum, SIG_DFL); |
memset(&sa, 0, sizeof(sa)); |
|
sigemptyset(&sa.sa_mask); |
|
sa.sa_handler = SIG_DFL; |
|
sigaction(signum, &sa, NULL); |
kill(getpid(), signum); |
kill(getpid(), signum); |
/* NOTREACHED */ |
/* NOTREACHED */ |
_exit(1); |
_exit(1); |
} |
|
|
|
/* |
|
* Memory management callback functions for ohash. |
|
*/ |
|
static void * |
|
tag_alloc(size_t sz, void *arg) |
|
{ |
|
|
|
return mandoc_malloc(sz); |
|
} |
|
|
|
static void * |
|
tag_calloc(size_t nmemb, size_t sz, void *arg) |
|
{ |
|
|
|
return mandoc_calloc(nmemb, sz); |
|
} |
|
|
|
static void |
|
tag_free(void *p, void *arg) |
|
{ |
|
|
|
free(p); |
|
} |
} |