1570af302Sopenharmony_ci/*
2570af302Sopenharmony_ci * Copyright (c) 2022 Huawei Device Co., Ltd.
3570af302Sopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License");
4570af302Sopenharmony_ci * you may not use this file except in compliance with the License.
5570af302Sopenharmony_ci * You may obtain a copy of the License at
6570af302Sopenharmony_ci *
7570af302Sopenharmony_ci *     http://www.apache.org/licenses/LICENSE-2.0
8570af302Sopenharmony_ci *
9570af302Sopenharmony_ci * Unless required by applicable law or agreed to in writing, software
10570af302Sopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS,
11570af302Sopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12570af302Sopenharmony_ci * See the License for the specific language governing permissions and
13570af302Sopenharmony_ci * limitations under the License.
14570af302Sopenharmony_ci */
15570af302Sopenharmony_ci
16570af302Sopenharmony_ci#include <fcntl.h>
17570af302Sopenharmony_ci#include <stdlib.h>
18570af302Sopenharmony_ci#include <sys/stat.h>
19570af302Sopenharmony_ci#include <stdio.h>
20570af302Sopenharmony_ci#include <string.h>
21570af302Sopenharmony_ci#include <termios.h>
22570af302Sopenharmony_ci#include <time.h>
23570af302Sopenharmony_ci#include "test.h"
24570af302Sopenharmony_ci
25570af302Sopenharmony_ci/**
26570af302Sopenharmony_ci * @tc.name      : tcdrain_0100
27570af302Sopenharmony_ci * @tc.desc      : Wait until output has been transmitted
28570af302Sopenharmony_ci * @tc.level     : Level 0
29570af302Sopenharmony_ci */
30570af302Sopenharmony_civoid tcdrain_0100(void)
31570af302Sopenharmony_ci{
32570af302Sopenharmony_ci    char Master[] = "/dev/ptmx";
33570af302Sopenharmony_ci    char text[] = "text to be written to tty";
34570af302Sopenharmony_ci    char data[80];
35570af302Sopenharmony_ci    int fd_master, fd_slave, result;
36570af302Sopenharmony_ci    time_t T;
37570af302Sopenharmony_ci
38570af302Sopenharmony_ci    fd_master = open(Master, O_RDWR | O_NONBLOCK);
39570af302Sopenharmony_ci    if (fd_master < 0) {
40570af302Sopenharmony_ci        t_error("%s open master failed", __func__);
41570af302Sopenharmony_ci    }
42570af302Sopenharmony_ci
43570af302Sopenharmony_ci    result = grantpt(fd_master);
44570af302Sopenharmony_ci    if (result < 0) {
45570af302Sopenharmony_ci        t_error("%s grantpt failed", __func__);
46570af302Sopenharmony_ci        return;
47570af302Sopenharmony_ci    }
48570af302Sopenharmony_ci
49570af302Sopenharmony_ci    result = unlockpt(fd_master);
50570af302Sopenharmony_ci    if (result < 0) {
51570af302Sopenharmony_ci        t_error("%s unlockpt failed", __func__);
52570af302Sopenharmony_ci        return;
53570af302Sopenharmony_ci    }
54570af302Sopenharmony_ci
55570af302Sopenharmony_ci    char *Slave = ptsname(fd_master);
56570af302Sopenharmony_ci    fd_slave = open(Slave, O_RDWR | O_NONBLOCK);
57570af302Sopenharmony_ci    if (fd_slave < 0) {
58570af302Sopenharmony_ci        t_error("%s open slave failed", __func__);
59570af302Sopenharmony_ci    }
60570af302Sopenharmony_ci
61570af302Sopenharmony_ci    if (fork() == 0) {
62570af302Sopenharmony_ci        if (write(fd_slave, text, strlen(text) + 1) == -1) {
63570af302Sopenharmony_ci            t_error("%s write failed", __func__);
64570af302Sopenharmony_ci        }
65570af302Sopenharmony_ci        time(&T);
66570af302Sopenharmony_ci        printf("child has written to tty, tcdrain() started at %s", ctime(&T));
67570af302Sopenharmony_ci
68570af302Sopenharmony_ci        if (tcdrain(fd_slave) != 0) {
69570af302Sopenharmony_ci            t_error("%s tcdrain failed", __func__);
70570af302Sopenharmony_ci        }
71570af302Sopenharmony_ci
72570af302Sopenharmony_ci        time(&T);
73570af302Sopenharmony_ci        printf("tcdrain() returned at %s", ctime(&T));
74570af302Sopenharmony_ci        exit(0);
75570af302Sopenharmony_ci    }
76570af302Sopenharmony_ci
77570af302Sopenharmony_ci    time(&T);
78570af302Sopenharmony_ci    printf("parent is starting nap at %s", ctime(&T));
79570af302Sopenharmony_ci    sleep(2);
80570af302Sopenharmony_ci    time(&T);
81570af302Sopenharmony_ci    printf("parent is done with nap at %s", ctime(&T));
82570af302Sopenharmony_ci    if (read(fd_master, data, sizeof(data)) == -1) {
83570af302Sopenharmony_ci        t_error("%s read failed", __func__);
84570af302Sopenharmony_ci    } else {
85570af302Sopenharmony_ci        printf("read '%s' from the tty\n", data);
86570af302Sopenharmony_ci    }
87570af302Sopenharmony_ci    sleep(2);
88570af302Sopenharmony_ci    close(fd_slave);
89570af302Sopenharmony_ci    close(fd_master);
90570af302Sopenharmony_ci}
91570af302Sopenharmony_ci
92570af302Sopenharmony_ciint main(int argc, char *argv[])
93570af302Sopenharmony_ci{
94570af302Sopenharmony_ci    tcdrain_0100();
95570af302Sopenharmony_ci
96570af302Sopenharmony_ci    return t_status;
97570af302Sopenharmony_ci}