From da1d60f6ed18ce94b138f5c2b2eb9f8be3dec197 Mon Sep 17 00:00:00 2001
From: Alexander Mikhalitsyn <aleksandr.mikhalitsyn@canonical.com>
Date: Tue, 17 Jan 2023 19:18:51 +0100
Subject: [PATCH] container: add SetTimeout

Extend container API with a new method SetTimeout
which allows to set SO_RCVTIMEO for LXC client socket.

Issue #4257

Signed-off-by: Alexander Mikhalitsyn <aleksandr.mikhalitsyn@canonical.com>
---
 container.go  | 17 +++++++++++++++++
 lxc-binding.c | 11 +++++++++++
 lxc-binding.h |  1 +
 3 files changed, 29 insertions(+)

diff --git a/container.go b/container.go
index 18c03e6..d22345d 100644
--- a/container.go
+++ b/container.go
@@ -408,6 +408,23 @@ func (c *Container) SeccompNotifyFdActive() (*os.File, error) {
 	return os.NewFile(uintptr(notifyFd), "seccomp notify"), nil
 }
 
+// SetTimeout sets the response receive timeout for commands
+func (c *Container) SetTimeout(timeout time.Duration) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+
+	if c.container == nil {
+		return ErrNotDefined
+	}
+
+	ret := int(C.go_lxc_set_timeout(c.container, C.int(timeout.Seconds())))
+	if ret < 0 {
+		return unix.Errno(-ret)
+	}
+
+	return nil
+}
+
 // Daemonize returns true if the container wished to be daemonized.
 func (c *Container) Daemonize() bool {
 	c.mu.RLock()
diff --git a/lxc-binding.c b/lxc-binding.c
index 7a0a022..f27683c 100644
--- a/lxc-binding.c
+++ b/lxc-binding.c
@@ -75,6 +75,17 @@ int go_lxc_seccomp_notify_fd_active(struct lxc_container *c) {
 #endif
 }
 
+int go_lxc_set_timeout(struct lxc_container *c, int timeout) {
+#if VERSION_AT_LEAST(5, 0, 4)
+	if (!c->set_timeout(c, timeout))
+		return ret_errno(EINVAL);
+
+	return 0;
+#else
+	return ret_errno(ENOSYS);
+#endif
+}
+
 int go_lxc_devpts_fd(struct lxc_container *c) {
 #if VERSION_AT_LEAST(4, 0, 5)
 	return c->devpts_fd(c);
diff --git a/lxc-binding.h b/lxc-binding.h
index 2a3e8ba..2aed765 100644
--- a/lxc-binding.h
+++ b/lxc-binding.h
@@ -97,6 +97,7 @@ extern int go_lxc_init_pidfd(struct lxc_container *c);
 extern int go_lxc_devpts_fd(struct lxc_container *c);
 extern int go_lxc_seccomp_notify_fd(struct lxc_container *c);
 extern int go_lxc_seccomp_notify_fd_active(struct lxc_container *c);
+extern int go_lxc_set_timeout(struct lxc_container *c, int timeout);
 extern bool go_lxc_checkpoint(struct lxc_container *c, char *directory, bool stop, bool verbose);
 extern bool go_lxc_restore(struct lxc_container *c, char *directory, bool verbose);
 extern bool go_lxc_config_item_is_supported(const char *key);