mirror of
https://github.com/balena-os/balena-supervisor.git
synced 2024-12-20 14:13:08 +00:00
42 lines
2.2 KiB
CoffeeScript
42 lines
2.2 KiB
CoffeeScript
Promise = require 'bluebird'
|
|
iptables = require '../src/lib/iptables'
|
|
|
|
m = require 'mochainon'
|
|
{ stub } = m.sinon
|
|
{ expect } = m.chai
|
|
|
|
describe 'iptables', ->
|
|
it 'calls iptables to delete and recreate rules to block a port', ->
|
|
stub(iptables, 'execAsync').returns(Promise.resolve())
|
|
iptables.rejectOnAllInterfacesExcept(['foo', 'bar'], 42)
|
|
.then ->
|
|
expect(iptables.execAsync.callCount).to.equal(6)
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -j REJECT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -A INPUT -p tcp --dport 42 -j REJECT')
|
|
.then ->
|
|
iptables.execAsync.restore()
|
|
|
|
it "falls back to blocking the port with DROP if there's no REJECT support", ->
|
|
stub(iptables, 'execAsync').callsFake (cmd) ->
|
|
if /REJECT$/.test(cmd)
|
|
Promise.reject()
|
|
else
|
|
Promise.resolve()
|
|
iptables.rejectOnAllInterfacesExcept(['foo', 'bar'], 42)
|
|
.then ->
|
|
expect(iptables.execAsync.callCount).to.equal(8)
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i foo -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -I INPUT -p tcp --dport 42 -i bar -j ACCEPT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -j REJECT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -A INPUT -p tcp --dport 42 -j REJECT')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -D INPUT -p tcp --dport 42 -j DROP')
|
|
expect(iptables.execAsync).to.be.calledWith('iptables -A INPUT -p tcp --dport 42 -j DROP')
|
|
.then ->
|
|
iptables.execAsync.restore()
|