summaryrefslogtreecommitdiff
path: root/extension/ordchr.c
blob: c5a1375f45cebd81c4af4ea013fe17327311ec94 (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
/*
 * ordchr.c - Builtin functions that provide ord() and chr() functions.
 *
 * Arnold Robbins
 * arnold@skeeve.com
 * 8/2001
 * Revised 6/2004
 */

/*
 * Copyright (C) 2001, 2004 the Free Software Foundation, Inc.
 * 
 * This file is part of GAWK, the GNU implementation of the
 * AWK Programming Language.
 * 
 * GAWK is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 * 
 * GAWK is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

#include "awk.h"

/*  do_ord --- return numeric value of first char of string */

static NODE *
do_ord(int nargs)
{
	NODE *str;
	int ret = -1;

	if  (do_lint && get_curfunc_arg_count() > 1)
		lintwarn("ord: called with too many arguments");

	str = get_scalar_argument(0, FALSE);
	if (str != NULL) {
		(void) force_string(str);
		ret = str->stptr[0];
	} else if (do_lint)
		lintwarn("ord: called with no arguments");


	/* Set the return value */
	return make_number((AWKNUM) ret);
}

/*  do_chr --- turn numeric value into a string */

static NODE *
do_chr(int nargs)
{
	NODE *num;
	unsigned int ret = 0;
	AWKNUM val = 0.0;
	char str[2];

	str[0] = str[1] = '\0';

	if  (do_lint && get_curfunc_arg_count() > 1)
		lintwarn("chr: called with too many arguments");

	num = get_scalar_argument(0, FALSE);
	if (num != NULL) {
		val = force_number(num);
		ret = val;	/* convert to int */
		ret &= 0xff;
		str[0] = ret;
		str[1] = '\0';
	} else if (do_lint)
		lintwarn("chr: called with no arguments");

	/* Set the return value */
	return make_string(str, 1);
}

/* dlload --- load new builtins in this library */

NODE *
dlload(tree, dl)
NODE *tree;
void *dl;
{
	make_builtin("ord", do_ord, 1);
	make_builtin("chr", do_chr, 1);

	return make_number((AWKNUM) 0);
}