1//===- AutoConvert.cpp - Auto conversion between ASCII/EBCDIC -------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file contains functions used for auto conversion between
10// ASCII/EBCDIC codepages specific to z/OS.
11//
12//===----------------------------------------------------------------------===//
13
14#ifdef __MVS__
15
16#include "llvm/Support/AutoConvert.h"
17#include <cassert>
18#include <fcntl.h>
19#include <sys/stat.h>
20#include <unistd.h>
21
22static int savedStdHandleAutoConversionMode[3] = {-1, -1, -1};
23
24int disableAutoConversion(int FD) {
25 static const struct f_cnvrt Convert = {
26 SETCVTOFF, // cvtcmd
27 0, // pccsid
28 0, // fccsid
29 };
30
31 return fcntl(FD, F_CONTROL_CVT, &Convert);
32}
33
34int restoreStdHandleAutoConversion(int FD) {
35 assert(FD == STDIN_FILENO || FD == STDOUT_FILENO || FD == STDERR_FILENO);
36 if (savedStdHandleAutoConversionMode[FD] == -1)
37 return 0;
38 struct f_cnvrt Cvt = {
39 savedStdHandleAutoConversionMode[FD], // cvtcmd
40 0, // pccsid
41 0, // fccsid
42 };
43 return (fcntl(FD, F_CONTROL_CVT, &Cvt));
44}
45
46int enableAutoConversion(int FD) {
47 struct f_cnvrt Query = {
48 QUERYCVT, // cvtcmd
49 0, // pccsid
50 0, // fccsid
51 };
52
53 if (fcntl(FD, F_CONTROL_CVT, &Query) == -1)
54 return -1;
55
56 // We don't need conversion for UTF-8 tagged files.
57 // TODO: Remove the assumption of ISO8859-1 = UTF-8 here when we fully resolve
58 // problems related to UTF-8 tagged source files.
59 // When the pccsid is not ISO8859-1, autoconversion is still needed.
60 if (Query.pccsid == CCSID_ISO8859_1 &&
61 (Query.fccsid == CCSID_UTF_8 || Query.fccsid == CCSID_ISO8859_1))
62 return 0;
63
64 // Save the state of std handles before we make changes to it.
65 if ((FD == STDIN_FILENO || FD == STDOUT_FILENO || FD == STDERR_FILENO) &&
66 savedStdHandleAutoConversionMode[FD] == -1)
67 savedStdHandleAutoConversionMode[FD] = Query.cvtcmd;
68
69 if (FD == STDOUT_FILENO || FD == STDERR_FILENO)
70 Query.cvtcmd = SETCVTON;
71 else
72 Query.cvtcmd = SETCVTALL;
73
74 Query.pccsid =
75 (FD == STDIN_FILENO || FD == STDOUT_FILENO || FD == STDERR_FILENO)
76 ? 0
77 : CCSID_UTF_8;
78 // Assume untagged files to be IBM-1047 encoded.
79 Query.fccsid = (Query.fccsid == FT_UNTAGGED) ? CCSID_IBM_1047 : Query.fccsid;
80 return fcntl(FD, F_CONTROL_CVT, &Query);
81}
82
83std::error_code llvm::disableAutoConversion(int FD) {
84 if (::disableAutoConversion(FD) == -1)
85 return std::error_code(errno, std::generic_category());
86
87 return std::error_code();
88}
89
90std::error_code llvm::enableAutoConversion(int FD) {
91 if (::enableAutoConversion(FD) == -1)
92 return std::error_code(errno, std::generic_category());
93
94 return std::error_code();
95}
96
97std::error_code llvm::restoreStdHandleAutoConversion(int FD) {
98 if (::restoreStdHandleAutoConversion(FD) == -1)
99 return std::error_code(errno, std::generic_category());
100
101 return std::error_code();
102}
103
104std::error_code llvm::setFileTag(int FD, int CCSID, bool Text) {
105 assert((!Text || (CCSID != FT_UNTAGGED && CCSID != FT_BINARY)) &&
106 "FT_UNTAGGED and FT_BINARY are not allowed for text files");
107 struct file_tag Tag;
108 Tag.ft_ccsid = CCSID;
109 Tag.ft_txtflag = Text;
110 Tag.ft_deferred = 0;
111 Tag.ft_rsvflags = 0;
112
113 if (fcntl(FD, F_SETTAG, &Tag) == -1)
114 return std::error_code(errno, std::generic_category());
115 return std::error_code();
116}
117
118#endif // __MVS__
119

source code of llvm/lib/Support/AutoConvert.cpp