From 14f8a27d2bca3e8bee8a6aaa8e6ebb36fcf09b90 Mon Sep 17 00:00:00 2001 From: Brian Candler Date: Sun, 1 May 2011 18:17:03 +0100 Subject: [PATCH] Tidy receive_stream --- lib/netlink/nlsocket.rb | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lib/netlink/nlsocket.rb b/lib/netlink/nlsocket.rb index 1d45afb..5e861b4 100644 --- a/lib/netlink/nlsocket.rb +++ b/lib/netlink/nlsocket.rb @@ -153,10 +153,11 @@ module Netlink end # Loop infinitely receiving messages of given type(s), ignoring pid and seq. - def receive_stream(*expected_type) + # Raises an exception on NLMSG_ERROR. + def receive_stream(*expected_types) loop do receive_response(nil) do |type, flags, seq, pid, msg| - if expected_type.include?(type) + if expected_types.include?(type) yield msg elsif type == NLMSG_ERROR raise ERRNO_MAP[-msg.error] || "Netlink Error: #{msg.inspect}"