Mirroring/endlessh
1
0
Fork 0
mirror of https://github.com/skeeto/endlessh.git synced 2025-12-08 22:23:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Example of making the first line special

Browse Source
This commit is contained in:
Christopher Wellons
2019-03-26 10:48:21 -04:00
parent e7c93c6664
commit 30e3ff1f32
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
endlessh.c
View File

@@ -41,6 +41,9 @@
#define DEFAULT_MAX_CLIENTS 4096 #define DEFAULT_MAX_CLIENTS 4096
#define DEFAULT_CONFIG_FILE "/etc/endlessh/config" #define DEFAULT_CONFIG_FILE "/etc/endlessh/config"
#define FLAG_SPECIAL_SENT (1 << 0)
#define SPECIAL_MESSAGE "Stay awhile and listen\r\n"
#define XSTR(s) STR(s) #define XSTR(s) STR(s)
#define STR(s) #s #define STR(s) #s
@@ -91,6 +94,7 @@ struct client {
struct client *next; struct client *next;
int port; int port;
int fd; int fd;
int flags;
}; };
static struct client * static struct client *
@@ -104,6 +108,7 @@ client_new(int fd, long long send_next)
c->bytes_sent = 0; c->bytes_sent = 0;
c->next = 0; c->next = 0;
c->fd = fd; c->fd = fd;
c->flags = 0;
/* Set the smallest possible recieve buffer. This reduces local /* Set the smallest possible recieve buffer. This reduces local
* resource usage and slows down the remote end. * resource usage and slows down the remote end.
@@ -730,7 +735,15 @@ main(int argc, char **argv)
} else if (revents & POLLOUT) { } else if (revents & POLLOUT) {
char line[256]; char line[256];
int len = randline(line, config.max_line_length, &rng); int len;
if (!(client->flags & FLAG_SPECIAL_SENT)) {
static const char special[] = SPECIAL_MESSAGE;
len = sizeof(special) - 1;
memcpy(line, special, len);
client->flags |= FLAG_SPECIAL_SENT;
} else {
len = randline(line, config.max_line_length, &rng);
}
for (;;) { for (;;) {
/* Don't really care if send is short */ /* Don't really care if send is short */
ssize_t out = send(fd, line, len, MSG_DONTWAIT); ssize_t out = send(fd, line, len, MSG_DONTWAIT);