1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/*
* safetitle sets the title of the terminal window to the given string,
* unless the terminal type is blacklisted.
*/
#include <err.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/select.h>
#include <termios.h>
#include <unistd.h>
#define WSCONS_DA (char[]){27,91,62,50,52,59,50,48,59,48,99,0}
#define SCREEN_DA (char[]){27,91,62,56,51,59,52,48,56,48,48,59,48,99,0}
int screen = 0;
int
blacklisted(int fd)
{
char *buf, c;
fd_set fds;
int i, r;
struct timeval timeout;
if((buf = malloc(200*sizeof(char)))==NULL)
err(1, "malloc");
/* check if GNU screen */
write(fd, "\033[>c", 4);
i = 0;
while(r = read(fd, &c, 1)){
if(r==-1)
return 1;
buf[i++] = c;
if(c=='c') break;
}
buf[i] = '\0';
screen = strcmp(buf, SCREEN_DA)==0;
/* ignore strange trailing nul before reading response */
FD_ZERO(&fds);
FD_SET(fd, &fds);
timeout.tv_sec = 0;
timeout.tv_usec = 0;
while(select(fd+1, &fds, NULL, NULL, &timeout)>0)
read(fd, &c, 1);
/* check (real) terminal type(s) */
if(screen) write(fd, "\033P\033[>c\033\\", 8);
else write(fd, "\033[>c", 4);
i = 0;
while(r = read(fd, &c, 1)){
loop:
if(r==-1)
return 1;
buf[i++] = c;
if(c==99) break;
}
buf[i] = '\0';
if(strcmp(buf, WSCONS_DA)==0)
return 1;
/* go back and check remaining responses (if any) */
check:
FD_ZERO(&fds);
FD_SET(fd, &fds);
timeout.tv_sec = 0;
timeout.tv_usec = 1; /* it takes a millisecond sometimes */
if(screen && select(fd+1, &fds, NULL, NULL, &timeout)>0){
/* ignore strange trailing nul before reading response */
if(read(fd, &c, 1)==-1)
return 1;
i = 0;
if(c==0) goto check;
else goto loop;
}
return 0;
}
void
safetitle(char *title)
{
char *t, *username;
int ttyfd;
if((ttyfd = open("/dev/tty", O_RDWR))==-1)
err(1, "open");
/* get device attributes for real terminal */
if(blacklisted(ttyfd))
return;
username = getenv("USER");
t = malloc((strlen(title)+1) * sizeof(char));
if (t == NULL)
err(1, "malloc");
strcpy(t, title);
if (strncmp(t, "/home/", 6) == 0) {
t += 5;
if (strncmp(t + 1, username, strlen(username)) == 0)
t += strlen(username);
t[0] = '~';
}
if(screen) dprintf(ttyfd, "\033P\033]2;%s\007\033\\", t);
else dprintf(ttyfd, "\033]2;%s\007", t);
}
|