forked from unpbook/unpv13e
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sourcetcp.c
60 lines (50 loc) · 1.64 KB
/
sourcetcp.c
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
/*
* Copyright (c) 1993 W. Richard Stevens. All rights reserved.
* Permission to use or modify this software and its documentation only for
* educational purposes and without fee is hereby granted, provided that
* the above copyright notice appear in all copies. The author makes no
* representations about the suitability of this software for any purpose.
* It is provided "as is" without express or implied warranty.
*/
#include "sock.h"
void
source_tcp(int sockfd)
{
int i, n, option;
socklen_t optlen;
char oob;
pattern(wbuf, writelen); /* fill send buffer with a pattern */
if (pauseinit)
sleep_us(pauseinit*1000);
for (i = 1; i <= nbuf; i++) {
if (urgwrite == i) {
oob = urgwrite;
if ( (n = send(sockfd, &oob, 1, MSG_OOB)) != 1)
err_sys("send of MSG_OOB returned %d, expected %d",
n, writelen);
if (verbose)
fprintf(stderr, "wrote %d byte of urgent data\n", n);
}
if ( (n = write(sockfd, wbuf, writelen)) != writelen) {
if (ignorewerr) {
err_ret("write returned %d, expected %d", n, writelen);
/* also call getsockopt() to clear so_error */
optlen = sizeof(option);
if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR,
&option, &optlen) < 0)
err_sys("SO_ERROR getsockopt error");
} else
err_sys("write returned %d, expected %d", n, writelen);
} else if (verbose)
fprintf(stderr, "wrote %d bytes\n", n);
if (pauserw)
sleep_us(pauserw*1000);
}
if (pauseclose) {
if (verbose)
fprintf(stderr, "pausing before close\n");
sleep_us(pauseclose*1000);
}
if (close(sockfd) < 0)
err_sys("close error"); /* since SO_LINGER may be set */
}