summaryrefslogtreecommitdiffhomepage
path: root/ryu/lib/sockaddr.py
diff options
context:
space:
mode:
Diffstat (limited to 'ryu/lib/sockaddr.py')
-rw-r--r--ryu/lib/sockaddr.py74
1 files changed, 74 insertions, 0 deletions
diff --git a/ryu/lib/sockaddr.py b/ryu/lib/sockaddr.py
new file mode 100644
index 00000000..460f2159
--- /dev/null
+++ b/ryu/lib/sockaddr.py
@@ -0,0 +1,74 @@
+# Copyright (C) 2014 Nippon Telegraph and Telephone Corporation.
+# Copyright (C) 2014 YAMAMOTO Takashi <yamamoto at valinux co jp>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import platform
+import socket
+import struct
+
+from ryu.lib import addrconv
+
+
+system = platform.system()
+if system == 'Linux':
+ # on linux,
+ # no ss_len
+ # u16 ss_family
+ _HDR_FMT = "H"
+ _HAVE_SS_LEN = False
+else:
+ # assume
+ # u8 ss_len
+ # u8 ss_family
+ _HDR_FMT = "BB"
+ _HAVE_SS_LEN = True
+
+
+# RFC 2553
+_SS_MAXSIZE = 128
+_SS_ALIGNSIZE = 8
+
+_SIN_SIZE = 16 # sizeof(struct sockaddr_in)
+
+_HDR_LEN = struct.calcsize(_HDR_FMT)
+
+
+def _hdr(ss_len, af):
+ if _HAVE_SS_LEN:
+ return struct.pack(_HDR_FMT, ss_len, af)
+ else:
+ return struct.pack(_HDR_FMT, af)
+
+
+def _pad_to(data, total_len):
+ pad_len = total_len - len(data)
+ return data + pad_len * '\0'
+
+
+def sa_in4(addr, port=0):
+ data = struct.pack("!H4s", port, addrconv.ipv4.text_to_bin(addr))
+ hdr = _hdr(_SIN_SIZE, socket.AF_INET)
+ return _pad_to(hdr + data, _SIN_SIZE)
+
+
+def sa_in6(addr, port=0, flowinfo=0, scope_id=0):
+ data = struct.pack("!HI16sI", port, flowinfo,
+ addrconv.ipv6.text_to_bin(addr), scope_id)
+ hdr = _hdr(_HDR_LEN + len(data), socket.AF_INET6)
+ return hdr + data
+
+
+def sa_to_ss(sa):
+ return _pad_to(sa, _SS_MAXSIZE)