upd14

socket.c.v

@@ -39,7 +39,8 @@ pub fn Socket.new(domain AddrFamily, st SocketType, proto Protocol) !Socket {
 
 // type reports the actual socket type.
 pub fn (s Socket) type() !SocketType {
-	return s.get_option_int(C.SOL_SOCKET, C.SO_TYPE)!
+	// return s.get_option_int(sol_socket, so_type)!
+	return s.get_option[SocketType](sol_socket, so_type)!
 }
 
 // Socket shutdown modes. See [shutdown(3p)](https://man7.org/linux/man-pages/man3/shutdown.3p.html) for details.
@@ -88,28 +89,37 @@ pub fn (s Socket) accept() !(Socket, SocketAddr) {
 	return sock, sock_addr
 }
 
+fn (s Socket) set_option_raw(level SocketLevel, option SocketOption, value voidptr) ! {
+	if C.setsockopt(s.fd, i32(level), i32(option), value, sizeof(value)) == -1 {
+		return os.last_error()
+	}
+}
+
 // set_option sets the socket option. See [socket(7)](https://man7.org/linux/man-pages/man7/socket.7.html)
 // and [setsockopt(3p)](https://man7.org/linux/man-pages/man3/setsockopt.3p.html) for details.
-pub fn (s Socket) set_option(level SocketLevel, option SocketOption, value voidptr) ! {
-	if C.setsockopt(s.fd, i32(level), i32(option), value, sizeof(value)) == -1 {
+pub fn (s Socket) set_option[T](level SocketLevel, option SocketOption, value T) ! {
+	s.set_option_raw(level, option, &value)!
+}
+
+fn (s Socket) get_option_raw(level SocketLevel, option SocketOption, mut value voidptr, mut size voidptr) ! {
+	if C.getsockopt(s.fd, i32(level), i32(option), value, size) == -1 {
 		return os.last_error()
 	}
 }
 
 // get_option gets the socket option. See [socket(7)](https://man7.org/linux/man-pages/man7/socket.7.html)
 // and [getsockopt(3p)](https://man7.org/linux/man-pages/man3/getsockopt.3p.html) for details.
-pub fn (s Socket) get_option(level SocketLevel, option SocketOption, mut value voidptr, mut size voidptr) ! {
-	if C.getsockopt(s.fd, i32(level), i32(option), value, size) == -1 {
-		return os.last_error()
-	}
-}
-
-// get_option gets the socket option as int. The same as `get_option`, but always returns int.
-pub fn (s Socket) get_option_int(level SocketLevel, option SocketOption) !int {
-	mut result := -1
-	if C.getsockopt(s.fd, i32(level), i32(option), &result, sizeof(result)) == -1 {
-		return os.last_error()
-	}
+// Example:
+// ```v
+// import netio
+// mut socket := netio.Socket.new(netio.af_inet, netio.sock_stream, 0)!
+// socket.set_option(netio.sol_socket, netio.so_reuseaddr, true)!
+// assert socket.get_option[bool](netio.sol_socket, netio.so_reuseaddr)! == true
+// ```
+pub fn (s Socket) get_option[T](level SocketLevel, option SocketOption) !T {
+	mut result := T{}
+	mut size := sizeof(result)
+	s.get_option_raw(level, option, mut &result, mut &size)!
 	return result
 }