Under Wayland, we should check WAYLAND_DISPLAY instead.

Signed-off-by: Tiwei Bie <tiwei....@antgroup.com>
---
 arch/um/drivers/xterm.c | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/arch/um/drivers/xterm.c b/arch/um/drivers/xterm.c
index e4316c7981e8..f607af738eac 100644
--- a/arch/um/drivers/xterm.c
+++ b/arch/um/drivers/xterm.c
@@ -97,12 +97,9 @@ static int xterm_open(int input, int output, int primary, 
void *d,
        if (access(argv[4], X_OK) < 0)
                argv[4] = "port-helper";
 
-       /*
-        * Check that DISPLAY is set, this doesn't guarantee the xterm
-        * will work but w/o it we can be pretty sure it won't.
-        */
-       if (getenv("DISPLAY") == NULL) {
-               printk(UM_KERN_ERR "xterm_open: $DISPLAY not set.\n");
+       /* Ensure we are running on Xorg or Wayland. */
+       if (!getenv("DISPLAY") && !getenv("WAYLAND_DISPLAY")) {
+               printk(UM_KERN_ERR "xterm_open : neither $DISPLAY nor 
$WAYLAND_DISPLAY is set.\n");
                return -ENODEV;
        }
 
-- 
2.34.1


Reply via email to