aboutsummaryrefslogtreecommitdiff
path: root/src/bin/pg_encoding/pg_encoding.c
blob: e94c8d3d877bfa4b8f9da772bcfe5b81dce3d23c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
/*-------------------------------------------------------------------------
 *
 * pg_encoding.c
 *
 *
 * Copyright (c) 1998, PostgreSQL development group
 *
 *
 * IDENTIFICATION
 *	  $Header: /cvsroot/pgsql/src/bin/pg_encoding/Attic/pg_encoding.c,v 1.12 2003/08/04 00:43:28 momjian Exp $
 *
 *-------------------------------------------------------------------------
 */
#include "postgres.h"
#include "miscadmin.h"
#include "mb/pg_wchar.h"

#include <ctype.h>

static void usage(void);

int
main(int argc, char **argv)
{
	char	   *p;
	int			enc;
	bool		be_only = FALSE;

	if (argc < 2)
	{
		usage();
		exit(1);
	}

	if (strcmp(argv[1], "-b") == 0)
	{
		if (argc < 3)
		{
			usage();
			exit(1);
		}
		be_only = TRUE;
		p = argv[2];
	}
	else
		p = argv[1];

	if (p && *p && isdigit((unsigned char) *p))
	{
		/*
		 * Encoding number to name
		 */
		char	   *name;

		enc = atoi(p);

		if ((name = (char *) pg_encoding_to_char(enc)))
		{
			if (be_only && pg_valid_server_encoding(name) < 0)
				exit(1);

			/*
			 * pg_encoding_to_char() returns "" if invalid encoding number
			 * is given
			 */
			else if (strcmp("", name))
				printf("%s\n", name);
			else
				exit(1);
		}
		exit(0);
	}
	else if (p && *p)
	{
		/*
		 * Encoding name to encoding number
		 */
		if ((enc = pg_char_to_encoding(p)) >= 0)
		{
			if (be_only && pg_valid_server_encoding(p) < 0)
				exit(1);
			printf("%d\n", enc);
		}
		else
			exit(1);

		exit(0);
	}
	exit(1);
}

static void
usage()
{
	fprintf(stderr,
	 "\nUsage: pg_encoding [options] encoding_name | encoding_number\n\n"
			"options:"
		  "         -b        check if encoding is valid for backend\n\n"
		);
}