fixed some build file flaws
This commit is contained in:
33
scripts/generrcode.hawk
Normal file
33
scripts/generrcode.hawk
Normal file
@ -0,0 +1,33 @@
|
||||
#
|
||||
# generrcode.hawk
|
||||
#
|
||||
# hawk -f generrcode.hawk hawk-cmn.h
|
||||
#
|
||||
|
||||
BEGIN {
|
||||
collect=0;
|
||||
tab3="\t\t";
|
||||
tab4="\t\t\t";
|
||||
}
|
||||
|
||||
/^[[:space:]]*enum[[:space:]]+hawk_errnum_t[[:space:]]*$/ {
|
||||
collect=1;
|
||||
print tab3 "// generated by generrcode.awk";
|
||||
print tab3 "enum ErrorNumber";
|
||||
print tab3 "{";
|
||||
}
|
||||
|
||||
collect && /^[[:space:]]*};[[:space:]]*$/ {
|
||||
print tab3 "};";
|
||||
print tab3 "// end of enum ErrorNumber";
|
||||
print "";
|
||||
collect=0;
|
||||
}
|
||||
|
||||
collect && /^[[:space:]]*HAWK_E[[:alnum:]]+/ {
|
||||
split ($1, flds, ",");
|
||||
name=flds[1];
|
||||
|
||||
print tab4 "ERR_" substr (name,7,length(name)-6) " = " name ",";
|
||||
}
|
||||
|
||||
33
scripts/genoptcode.hawk
Normal file
33
scripts/genoptcode.hawk
Normal file
@ -0,0 +1,33 @@
|
||||
#
|
||||
# genoptcode.hawk
|
||||
#
|
||||
# hawk -f genoptcode.hawk hawk.h
|
||||
#
|
||||
|
||||
BEGIN {
|
||||
collect=0;
|
||||
tab3="\t\t";
|
||||
tab4="\t\t\t";
|
||||
}
|
||||
|
||||
/^[[:space:]]*enum[[:space:]]+hawk_trait_t[[:space:]]*$/ {
|
||||
collect=1;
|
||||
print tab3 "// generated by genoptcode.awk";
|
||||
print tab3 "enum Option";
|
||||
print tab3 "{";
|
||||
}
|
||||
|
||||
collect && /^[[:space:]]*};[[:space:]]*$/ {
|
||||
print tab3 "};";
|
||||
print tab3 "// end of enum Option";
|
||||
print "";
|
||||
collect=0;
|
||||
}
|
||||
|
||||
collect && /^[[:space:]]*HAWK_[[:alnum:]]+/ {
|
||||
split ($1, flds, ",");
|
||||
name=flds[1];
|
||||
|
||||
print tab4 "TRAIT_" substr (name,6,length(name)-5) " = " name ",";
|
||||
}
|
||||
|
||||
323
scripts/tcp-proxy.hawk
Normal file
323
scripts/tcp-proxy.hawk
Normal file
@ -0,0 +1,323 @@
|
||||
@pragma entry main
|
||||
@pragma implicit off
|
||||
|
||||
function destroy_bridge (&brtab, fd, mx)
|
||||
{
|
||||
@local pfd;
|
||||
|
||||
pfd = brtab[fd];
|
||||
|
||||
print "Info: Destroying", fd, pfd;
|
||||
|
||||
sys::delfrommux (mx, fd);
|
||||
sys::delfrommux (mx, pfd);
|
||||
sys::close (fd);
|
||||
sys::close (pfd);
|
||||
|
||||
delete brtab[fd];
|
||||
delete brtab[fd,"connecting"];
|
||||
delete brtab[fd,"pending"];
|
||||
delete brtab[fd,"evmask"];
|
||||
delete brtab[fd,"eof"];
|
||||
|
||||
delete brtab[pfd];
|
||||
delete brtab[pfd,"connecting"];
|
||||
delete brtab[pfd,"pending"];
|
||||
delete brtab[pfd,"evmask"];
|
||||
delete brtab[pfd,"eof"];
|
||||
}
|
||||
|
||||
function prepare_bridge (&brtab, fd, pfd, mx)
|
||||
{
|
||||
if (sys::addtomux(mx, pfd, sys::MUX_EVT_OUT) <= -1) return -1;
|
||||
|
||||
print "Info: Preparing", fd, pfd;
|
||||
|
||||
brtab[fd] = pfd;
|
||||
brtab[fd,"evmask"] = 0;
|
||||
|
||||
brtab[pfd] = fd;
|
||||
brtab[pfd,"connecting"] = 1;
|
||||
brtab[pfd,"evmask"] = sys::MUX_EVT_OUT;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
function establish_bridge (&brtab, fd, mx)
|
||||
{
|
||||
@local pfd;
|
||||
|
||||
pfd = brtab[fd];
|
||||
if (sys::addtomux(mx, pfd, sys::MUX_EVT_IN) <= -1 ||
|
||||
sys::modinmux(mx, fd, sys::MUX_EVT_IN) <= -1) return -1;
|
||||
|
||||
brtab[fd,"evmask"] = sys::MUX_EVT_IN;
|
||||
brtab[pfd,"evmask"] = sys::MUX_EVT_IN;
|
||||
|
||||
delete brtab[fd,"connecting"];
|
||||
return 0;
|
||||
}
|
||||
|
||||
function handle_bridge_eof (&brtab, fd, mx)
|
||||
{
|
||||
@local pfd, evmask;
|
||||
|
||||
pfd = brtab[fd];
|
||||
|
||||
evmask = brtab[fd,"evmask"] & ~sys::MUX_EVT_IN;
|
||||
if (sys::modinmux(mx, fd, evmask) <= -1) return -1;
|
||||
brtab[fd,"evmask"] = evmask;
|
||||
brtab[fd,"eof"] = 1;
|
||||
|
||||
## EOF on fd, disable transmission on pfd if no pending
|
||||
if (length(brtab[pfd,"pending"]) == 0) sys::shutdown (pfd, sys::SHUT_WR);
|
||||
|
||||
# if peer fd reached eof and there are no pending data, arrange to abort the bridge
|
||||
if (brtab[pfd,"eof"] != 0 && length(brtab[pfd,"pending"]) == 0 && length(brtab[fd,"pending"]) == 0) return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
function bridge_traffic (&brtab, fd, mx)
|
||||
{
|
||||
@local len, buf, pfd, pos, x, evmask;
|
||||
|
||||
pfd = brtab[fd];
|
||||
|
||||
len = sys::read(fd, buf, 8096);
|
||||
if (len <= -1)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
else if (len == 0)
|
||||
{
|
||||
return handle_bridge_eof (brtab, fd, mx);
|
||||
}
|
||||
|
||||
pos = 1;
|
||||
while (pos <= len)
|
||||
{
|
||||
x = sys::write(pfd, buf, pos);
|
||||
if (x == sys::RC_EAGAIN)
|
||||
{
|
||||
evmask = brtab[fd,"evmask"] & ~sys::MUX_EVT_IN;
|
||||
if (sys::modinmux(mx, fd, evmask) <= -1) return -1;
|
||||
brtab[fd,"evmask"] = evmask;
|
||||
|
||||
evmask = brtab[pfd,"evmask"] | sys::MUX_EVT_OUT;
|
||||
if (sys::modinmux(mx, pfd, evmask) <= -1) return -1;
|
||||
brtab[pfd,"evmask"] = evmask;
|
||||
|
||||
brtab[pfd,"pending"] %%= substr(buf, pos);
|
||||
break;
|
||||
}
|
||||
else if (x <= -1)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
pos += x;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
function bridge_pendind_data (&brtab, fd, mx)
|
||||
{
|
||||
@local pending_data, pos, len, x, pfd, evmask;
|
||||
|
||||
pfd = brtab[fd];
|
||||
pending_data = brtab[fd,"pending"];
|
||||
|
||||
len = length(pending_data);
|
||||
pos = 1;
|
||||
while (pos <= len)
|
||||
{
|
||||
x = sys::write(fd, pending_data, pos);
|
||||
if (x == sys::RC_EAGAIN)
|
||||
{
|
||||
brtab[fd,"pending"] = substr(pending_data, pos);
|
||||
return 0;
|
||||
}
|
||||
pos += x;
|
||||
}
|
||||
|
||||
## sent all pending data.
|
||||
delete brtab[fd,"pending"];
|
||||
|
||||
evmask = brtab[fd,"evmask"] & ~sys::MUX_EVT_OUT;
|
||||
if (sys::modinmux(mx, fd, evmask) <= -1) return -1;
|
||||
brtab[fd,"evmask"] = evmask;
|
||||
|
||||
if (brtab[pfd,"eof"] == 0)
|
||||
{
|
||||
evmask = brtab[pfd,"evmask"] | sys::MUX_EVT_IN;
|
||||
if (sys::modinmux(mx, pfd, evmask) <= -1) return -1;
|
||||
brtab[pfd,"evmask"] = evmask;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
function handle_bridge_event (&brtab, fd, mx, evmask)
|
||||
{
|
||||
if (evmask & sys::MUX_EVT_ERR)
|
||||
{
|
||||
destroy_bridge (brtab, fd, mx);
|
||||
}
|
||||
else if (evmask & sys::MUX_EVT_HUP)
|
||||
{
|
||||
print "Info: HUP detected on", fd;
|
||||
if (handle_bridge_eof(brtab, fd, mx) <= -1)
|
||||
{
|
||||
destroy_bridge (brtab, fd, mx);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (evmask & sys::MUX_EVT_OUT)
|
||||
{
|
||||
if ((fd,"connecting") in brtab)
|
||||
{
|
||||
## remote peer connection
|
||||
if (establish_bridge(brtab, fd, mx) <= -1)
|
||||
{
|
||||
destroy_bridge (brtab, fd, mx);
|
||||
return;
|
||||
}
|
||||
}
|
||||
else if ((fd,"pending") in brtab)
|
||||
{
|
||||
if (bridge_pending_data(brtab, fd, mx) <= -1)
|
||||
{
|
||||
destroy_bridge (brtab, fd, mx);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (evmask & sys::MUX_EVT_IN)
|
||||
{
|
||||
if (bridge_traffic(brtab, fd, mx) <= -1)
|
||||
{
|
||||
destroy_bridge (brtab, fd, mx);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function serve_connections (mx, ss, remoteaddr)
|
||||
{
|
||||
@local brtab, remotetab
|
||||
|
||||
while (1)
|
||||
{
|
||||
@local x, i;
|
||||
|
||||
if ((x = sys::waitonmux(mx, 3.10)) <= -1)
|
||||
{
|
||||
if (x == sys::RC_EINTR) continue;
|
||||
print "Error: problem while waiting on multiplexer -", sys::errmsg();
|
||||
break;
|
||||
}
|
||||
|
||||
if (x == 0) continue; ## timed out
|
||||
|
||||
|
||||
for (i = 0; i < x; i++)
|
||||
{
|
||||
@local fd, evmask;
|
||||
|
||||
if (sys::getmuxevt(mx, i, fd, evmask) <= -1) continue;
|
||||
|
||||
if (fd == ss)
|
||||
{
|
||||
## server socket event
|
||||
@local l, r;
|
||||
|
||||
l = sys::accept(ss, sys::SOCK_CLOEXEC | sys::SOCK_NONBLOCK);
|
||||
if (l <= -1)
|
||||
{
|
||||
print "Error: failed to accept connection -", sys::errmsg();
|
||||
}
|
||||
else
|
||||
{
|
||||
r = sys::socket(sys::sockaddrdom(remoteaddr), sys::SOCK_STREAM | sys::SOCK_CLOEXEC | sys::SOCK_NONBLOCK, 0);
|
||||
if (r <= -1)
|
||||
{
|
||||
print "Error: unable to create remote socket for local socket", l, "-", sys::errmsg();
|
||||
sys::close(l);
|
||||
}
|
||||
else
|
||||
{
|
||||
@local rc;
|
||||
rc = sys::connect(r, remoteaddr);
|
||||
if ((rc <= -1 && rc != sys::RC_EINPROG) || prepare_bridge(brtab, l, r, mx) <= -1)
|
||||
{
|
||||
print "Error: unable to conneect to", remoteaddr, "-", sys::errmsg();
|
||||
sys::close (r);
|
||||
sys::close (l);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (fd in brtab) handle_bridge_event (brtab, fd, mx, evmask);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
function main (localaddr, remoteaddr, c)
|
||||
{
|
||||
@local ss, mx;
|
||||
|
||||
if (ARGC != 3)
|
||||
{
|
||||
###printf ("Usage: %s -f %s local-address remote-address\n", ARGV[0], SCRIPTNAME);
|
||||
printf ("Usage: %s local-address remote-address\n", SCRIPTNAME);
|
||||
return -1;
|
||||
}
|
||||
|
||||
mx = sys::openmux();
|
||||
if (mx <= -1)
|
||||
{
|
||||
print "Error: unable to create multiplexer -", sys::errmsg();
|
||||
return -1;
|
||||
}
|
||||
|
||||
ss = sys::socket(sys::sockaddrdom(localaddr), sys::SOCK_STREAM | sys::SOCK_CLOEXEC | sys::SOCK_NONBLOCK, 0);
|
||||
if (ss <= -1)
|
||||
{
|
||||
print "Error: unable to create socket -", sys::errmsg();
|
||||
return -1;
|
||||
}
|
||||
|
||||
sys::setsockopt(ss, sys::SOL_SOCKET, sys::SO_REUSEADDR, 1);
|
||||
sys::setsockopt(ss, sys::SOL_SOCKET, sys::SO_REUSEPORT, 1);
|
||||
|
||||
if (sys::bind(ss, localaddr) <= -1 || sys::listen(ss, 0) <= -1)
|
||||
{
|
||||
print "Error: socket error -", sys::errmsg();
|
||||
sys::close (ss);
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (sys::addtomux(mx, ss, sys::MUX_EVT_IN) <= -1)
|
||||
{
|
||||
print "Error: unable to add server socket to multiplexer -", sys::errmsg();
|
||||
sys::close (ss);
|
||||
return -1;
|
||||
}
|
||||
|
||||
serve_connections (mx, ss, remoteaddr);
|
||||
|
||||
sys::delfrommux (mx, ss);
|
||||
sys::close (ss);
|
||||
sys::close (mx);
|
||||
}
|
||||
Reference in New Issue
Block a user