x86_64-xlate.pl 43.8 KB
Newer Older
R
Rich Salz 已提交
1
#! /usr/bin/env perl
M
Matt Caswell 已提交
2
# Copyright 2005-2020 The OpenSSL Project Authors. All Rights Reserved.
R
Rich Salz 已提交
3 4 5 6 7 8
#
# Licensed under the OpenSSL license (the "License").  You may not use
# this file except in compliance with the License.  You can obtain a copy
# in the file LICENSE in the source distribution or at
# https://www.openssl.org/source/license.html

9

10
# Ascetic x86_64 AT&T to MASM/NASM assembler translator by <appro>.
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
#
# Why AT&T to MASM and not vice versa? Several reasons. Because AT&T
# format is way easier to parse. Because it's simpler to "gear" from
# Unix ABI to Windows one [see cross-reference "card" at the end of
# file]. Because Linux targets were available first...
#
# In addition the script also "distills" code suitable for GNU
# assembler, so that it can be compiled with more rigid assemblers,
# such as Solaris /usr/ccs/bin/as.
#
# This translator is not designed to convert *arbitrary* assembler
# code from AT&T format to MASM one. It's designed to convert just
# enough to provide for dual-ABI OpenSSL modules development...
# There *are* limitations and you might have to modify your assembler
# code or this script to achieve the desired result...
#
# Currently recognized limitations:
#
# - can't use multiple ops per line;
#
# Dual-ABI styling rules.
#
33 34
# 1. Adhere to Unix register and stack layout [see cross-reference
#    ABI "card" at the end for explanation].
35 36 37 38
# 2. Forget about "red zone," stick to more traditional blended
#    stack frame allocation. If volatile storage is actually required
#    that is. If not, just leave the stack as is.
# 3. Functions tagged with ".type name,@function" get crafted with
39
#    unified Win64 prologue and epilogue automatically. If you want
40
#    to take care of ABI differences yourself, tag functions as
41 42 43 44 45 46 47 48 49 50
#    ".type name,@abi-omnipotent" instead.
# 4. To optimize the Win64 prologue you can specify number of input
#    arguments as ".type name,@function,N." Keep in mind that if N is
#    larger than 6, then you *have to* write "abi-omnipotent" code,
#    because >6 cases can't be addressed with unified prologue.
# 5. Name local labels as .L*, do *not* use dynamic labels such as 1:
#    (sorry about latter).
# 6. Don't use [or hand-code with .byte] "rep ret." "ret" mnemonic is
#    required to identify the spots, where to inject Win64 epilogue!
#    But on the pros, it's then prefixed with rep automatically:-)
51 52 53
# 7. Stick to explicit ip-relative addressing. If you have to use
#    GOTPCREL addressing, stick to mov symbol@GOTPCREL(%rip),%r??.
#    Both are recognized and translated to proper Win64 addressing
54
#    modes.
55 56 57 58
#
# 8. In order to provide for structured exception handling unified
#    Win64 prologue copies %rsp value to %rax. For further details
#    see SEH paragraph at the end.
59
# 9. .init segment is allowed to contain calls to functions only.
60 61
# a. If function accepts more than 4 arguments *and* >4th argument
#    is declared as non 64-bit value, do clear its upper part.
62

63 64 65

use strict;

66 67 68
my $flavour = shift;
my $output  = shift;
if ($flavour =~ /\./) { $output = $flavour; undef $flavour; }
A
Andy Polyakov 已提交
69

70 71
open STDOUT,">$output" || die "can't open $output: $!"
	if (defined($output));
72

73 74 75 76
my $gas=1;	$gas=0 if ($output =~ /\.asm$/);
my $elf=1;	$elf=0 if (!$gas);
my $win64=0;
my $prefix="";
77
my $decor=".L";
78

79
my $masmref=8 + 50727*2**-32;	# 8.00.50727 shipped with VS2005
80
my $masm=0;
81 82
my $PTR=" PTR";

83
my $nasmref=2.03;
84 85
my $nasm=0;

86 87
if    ($flavour eq "mingw64")	{ $gas=1; $elf=0; $win64=1;
				  $prefix=`echo __USER_LABEL_PREFIX__ | $ENV{CC} -E -P -`;
88
				  $prefix =~ s|\R$||; # Better chomp
89
				}
90 91 92
elsif ($flavour eq "macosx")	{ $gas=1; $elf=0; $prefix="_"; $decor="L\$"; }
elsif ($flavour eq "masm")	{ $gas=0; $elf=0; $masm=$masmref; $win64=1; $decor="\$L\$"; }
elsif ($flavour eq "nasm")	{ $gas=0; $elf=0; $nasm=$nasmref; $win64=1; $decor="\$L\$"; $PTR=""; }
93
elsif (!$gas)
94 95
{   if ($ENV{ASM} =~ m/nasm/ && `nasm -v` =~ m/version ([0-9]+)\.([0-9]+)/i)
    {	$nasm = $1 + $2*0.01; $PTR="";  }
96 97
    elsif (`ml64 2>&1` =~ m/Version ([0-9]+)\.([0-9]+)(\.([0-9]+))?/)
    {	$masm = $1 + $2*2**-16 + $4*2**-32;   }
98
    die "no assembler found on %PATH%" if (!($nasm || $masm));
99 100
    $win64=1;
    $elf=0;
101
    $decor="\$L\$";
102
}
103 104 105

my $current_segment;
my $current_function;
106
my %globals;
107 108 109

{ package opcode;	# pick up opcodes
    sub re {
110 111 112
	my	($class, $line) = @_;
	my	$self = {};
	my	$ret;
113

114 115
	if ($$line =~ /^([a-z][a-z0-9]*)/i) {
	    bless $self,$class;
116 117
	    $self->{op} = $1;
	    $ret = $self;
118
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
119 120

	    undef $self->{sz};
A
Andy Polyakov 已提交
121
	    if ($self->{op} =~ /^(movz)x?([bw]).*/) {	# movz is pain...
122
		$self->{op} = $1;
A
Andy Polyakov 已提交
123
		$self->{sz} = $2;
124
	    } elsif ($self->{op} =~ /call|jmp/) {
125
		$self->{sz} = "";
126
	    } elsif ($self->{op} =~ /^p/ && $' !~ /^(ush|op|insrw)/) { # SSEn
127
		$self->{sz} = "";
128
	    } elsif ($self->{op} =~ /^[vk]/) { # VEX or k* such as kmov
A
Andy Polyakov 已提交
129
		$self->{sz} = "";
130
	    } elsif ($self->{op} =~ /mov[dq]/ && $$line =~ /%xmm/) {
131
		$self->{sz} = "";
A
Andy Polyakov 已提交
132
	    } elsif ($self->{op} =~ /([a-z]{3,})([qlwb])$/) {
133 134 135 136 137 138 139
		$self->{op} = $1;
		$self->{sz} = $2;
	    }
	}
	$ret;
    }
    sub size {
140
	my ($self, $sz) = @_;
141 142 143 144 145
	$self->{sz} = $sz if (defined($sz) && !defined($self->{sz}));
	$self->{sz};
    }
    sub out {
	my $self = shift;
146
	if ($gas) {
A
Andy Polyakov 已提交
147
	    if ($self->{op} eq "movz") {	# movz is pain...
148
		sprintf "%s%s%s",$self->{op},$self->{sz},shift;
149
	    } elsif ($self->{op} =~ /^set/) {
A
Andy Polyakov 已提交
150
		"$self->{op}";
151
	    } elsif ($self->{op} eq "ret") {
152 153 154 155 156 157 158 159
		my $epilogue = "";
		if ($win64 && $current_function->{abi} eq "svr4") {
		    $epilogue = "movq	8(%rsp),%rdi\n\t" .
				"movq	16(%rsp),%rsi\n\t";
		}
	    	$epilogue . ".byte	0xf3,0xc3";
	    } elsif ($self->{op} eq "call" && !$elf && $current_segment eq ".init") {
		".p2align\t3\n\t.quad";
160 161 162 163
	    } else {
		"$self->{op}$self->{sz}";
	    }
	} else {
A
Andy Polyakov 已提交
164
	    $self->{op} =~ s/^movz/movzx/;
165 166
	    if ($self->{op} eq "ret") {
		$self->{op} = "";
167
		if ($win64 && $current_function->{abi} eq "svr4") {
168 169
		    $self->{op} = "mov	rdi,QWORD$PTR\[8+rsp\]\t;WIN64 epilogue\n\t".
				  "mov	rsi,QWORD$PTR\[16+rsp\]\n\t";
170 171
	    	}
		$self->{op} .= "DB\t0F3h,0C3h\t\t;repret";
172
	    } elsif ($self->{op} =~ /^(pop|push)f/) {
173
		$self->{op} .= $self->{sz};
174
	    } elsif ($self->{op} eq "call" && $current_segment eq ".CRT\$XCU") {
175
		$self->{op} = "\tDQ";
176
	    }
177 178 179
	    $self->{op};
	}
    }
180
    sub mnemonic {
181
	my ($self, $op) = @_;
182 183 184
	$self->{op}=$op if (defined($op));
	$self->{op};
    }
185 186 187
}
{ package const;	# pick up constants, which start with $
    sub re {
188 189 190
	my	($class, $line) = @_;
	my	$self = {};
	my	$ret;
191

192 193
	if ($$line =~ /^\$([^,]+)/) {
	    bless $self, $class;
194 195
	    $self->{value} = $1;
	    $ret = $self;
196
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
197 198 199 200 201
	}
	$ret;
    }
    sub out {
    	my $self = shift;
202

203
	$self->{value} =~ s/\b(0b[0-1]+)/oct($1)/eig;
204
	if ($gas) {
205 206
	    # Solaris /usr/ccs/bin/as can't handle multiplications
	    # in $self->{value}
207
	    my $value = $self->{value};
208
	    no warnings;    # oct might complain about overflow, ignore here...
209 210 211 212
	    $value =~ s/(?<![\w\$\.])(0x?[0-9a-f]+)/oct($1)/egi;
	    if ($value =~ s/([0-9]+\s*[\*\/\%]\s*[0-9]+)/eval($1)/eg) {
		$self->{value} = $value;
	    }
213 214
	    sprintf "\$%s",$self->{value};
	} else {
215 216 217
	    my $value = $self->{value};
	    $value =~ s/0x([0-9a-f]+)/0$1h/ig if ($masm);
	    sprintf "%s",$value;
218
	}
219 220 221
    }
}
{ package ea;		# pick up effective addresses: expr(%reg,%reg,scale)
222 223 224 225 226 227 228

    my %szmap = (	b=>"BYTE$PTR",    w=>"WORD$PTR",
			l=>"DWORD$PTR",   d=>"DWORD$PTR",
			q=>"QWORD$PTR",   o=>"OWORD$PTR",
			x=>"XMMWORD$PTR", y=>"YMMWORD$PTR",
			z=>"ZMMWORD$PTR" ) if (!$gas);

229
    sub re {
230 231 232
	my	($class, $line, $opcode) = @_;
	my	$self = {};
	my	$ret;
233

234
	# optional * ----vvv--- appears in indirect jmp/call
235
	if ($$line =~ /^(\*?)([^\(,]*)\(([%\w,]+)\)((?:{[^}]+})*)/) {
236
	    bless $self, $class;
237 238 239
	    $self->{asterisk} = $1;
	    $self->{label} = $2;
	    ($self->{base},$self->{index},$self->{scale})=split(/,/,$3);
240
	    $self->{scale} = 1 if (!defined($self->{scale}));
241
	    $self->{opmask} = $4;
242
	    $ret = $self;
243
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
244

245
	    if ($win64 && $self->{label} =~ s/\@GOTPCREL//) {
246 247
		die if ($opcode->mnemonic() ne "mov");
		$opcode->mnemonic("lea");
248
	    }
249 250
	    $self->{base}  =~ s/^%//;
	    $self->{index} =~ s/^%// if (defined($self->{index}));
251
	    $self->{opcode} = $opcode;
252 253 254 255 256
	}
	$ret;
    }
    sub size {}
    sub out {
257
	my ($self, $sz) = @_;
258

259
	$self->{label} =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
260
	$self->{label} =~ s/\.L/$decor/g;
261

262 263 264 265 266 267
	# Silently convert all EAs to 64-bit. This is required for
	# elder GNU assembler and results in more compact code,
	# *but* most importantly AES module depends on this feature!
	$self->{index} =~ s/^[er](.?[0-9xpi])[d]?$/r\1/;
	$self->{base}  =~ s/^[er](.?[0-9xpi])[d]?$/r\1/;

268
	# Solaris /usr/ccs/bin/as can't handle multiplications
269
	# in $self->{label}...
270 271
	use integer;
	$self->{label} =~ s/(?<![\w\$\.])(0x?[0-9a-f]+)/oct($1)/egi;
272
	$self->{label} =~ s/\b([0-9]+\s*[\*\/\%]\s*[0-9]+)\b/eval($1)/eg;
273 274 275 276 277 278 279 280

	# Some assemblers insist on signed presentation of 32-bit
	# offsets, but sign extension is a tricky business in perl...
	if ((1<<31)<<1) {
	    $self->{label} =~ s/\b([0-9]+)\b/$1<<32>>32/eg;
	} else {
	    $self->{label} =~ s/\b([0-9]+)\b/$1>>0/eg;
	}
281

282
	# if base register is %rbp or %r13, see if it's possible to
283
	# flip base and index registers [for better performance]
284 285 286 287 288
	if (!$self->{label} && $self->{index} && $self->{scale}==1 &&
	    $self->{base} =~ /(rbp|r13)/) {
		$self->{base} = $self->{index}; $self->{index} = $1;
	}

289
	if ($gas) {
290
	    $self->{label} =~ s/^___imp_/__imp__/   if ($flavour eq "mingw64");
291

292
	    if (defined($self->{index})) {
293 294
		sprintf "%s%s(%s,%%%s,%d)%s",
					$self->{asterisk},$self->{label},
295
					$self->{base}?"%$self->{base}":"",
296
					$self->{index},$self->{scale},
297
					$self->{opmask};
298
	    } else {
299
		sprintf "%s%s(%%%s)%s",	$self->{asterisk},$self->{label},
300
					$self->{base},$self->{opmask};
301 302
	    }
	} else {
303
	    $self->{label} =~ s/\./\$/g;
304
	    $self->{label} =~ s/(?<![\w\$\.])0x([0-9a-f]+)/0$1h/ig;
305
	    $self->{label} = "($self->{label})" if ($self->{label} =~ /[\*\+\-\/]/);
306

307 308 309 310 311 312
	    my $mnemonic = $self->{opcode}->mnemonic();
	    ($self->{asterisk})				&& ($sz="q") ||
	    ($mnemonic =~ /^v?mov([qd])$/)		&& ($sz=$1)  ||
	    ($mnemonic =~ /^v?pinsr([qdwb])$/)		&& ($sz=$1)  ||
	    ($mnemonic =~ /^vpbroadcast([qdwb])$/)	&& ($sz=$1)  ||
	    ($mnemonic =~ /^v(?!perm)[a-z]+[fi]128$/)	&& ($sz="x");
313

314
	    $self->{opmask}  =~ s/%(k[0-7])/$1/;
315

316
	    if (defined($self->{index})) {
317
		sprintf "%s[%s%s*%d%s]%s",$szmap{$sz},
318
					$self->{label}?"$self->{label}+":"",
319
					$self->{index},$self->{scale},
320
					$self->{base}?"+$self->{base}":"",
321
					$self->{opmask};
A
Andy Polyakov 已提交
322
	    } elsif ($self->{base} eq "rip") {
323
		sprintf "%s[%s]",$szmap{$sz},$self->{label};
324
	    } else {
325
		sprintf "%s[%s%s]%s",	$szmap{$sz},
326
					$self->{label}?"$self->{label}+":"",
327
					$self->{base},$self->{opmask};
328 329 330 331 332 333
	    }
	}
    }
}
{ package register;	# pick up registers, which start with %.
    sub re {
334
	my	($class, $line, $opcode) = @_;
335
	my	$self = {};
336
	my	$ret;
337

338
	# optional * ----vvv--- appears in indirect jmp/call
339
	if ($$line =~ /^(\*?)%(\w+)((?:{[^}]+})*)/) {
340
	    bless $self,$class;
341 342
	    $self->{asterisk} = $1;
	    $self->{value} = $2;
343
	    $self->{opmask} = $3;
344
	    $opcode->size($self->size());
345
	    $ret = $self;
346
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
347 348 349 350 351
	}
	$ret;
    }
    sub size {
	my	$self = shift;
352
	my	$ret;
353 354 355 356 357 358 359 360 361 362 363 364 365 366

	if    ($self->{value} =~ /^r[\d]+b$/i)	{ $ret="b"; }
	elsif ($self->{value} =~ /^r[\d]+w$/i)	{ $ret="w"; }
	elsif ($self->{value} =~ /^r[\d]+d$/i)	{ $ret="l"; }
	elsif ($self->{value} =~ /^r[\w]+$/i)	{ $ret="q"; }
	elsif ($self->{value} =~ /^[a-d][hl]$/i){ $ret="b"; }
	elsif ($self->{value} =~ /^[\w]{2}l$/i)	{ $ret="b"; }
	elsif ($self->{value} =~ /^[\w]{2}$/i)	{ $ret="w"; }
	elsif ($self->{value} =~ /^e[a-z]{2}$/i){ $ret="l"; }

	$ret;
    }
    sub out {
    	my $self = shift;
367 368
	if ($gas)	{ sprintf "%s%%%s%s",	$self->{asterisk},
						$self->{value},
369 370 371
						$self->{opmask}; }
	else		{ $self->{opmask} =~ s/%(k[0-7])/$1/;
			  $self->{value}.$self->{opmask}; }
372 373 374 375
    }
}
{ package label;	# pick up labels, which end with :
    sub re {
376 377 378
	my	($class, $line) = @_;
	my	$self = {};
	my	$ret;
379

380 381
	if ($$line =~ /(^[\.\w]+)\:/) {
	    bless $self,$class;
382 383
	    $self->{value} = $1;
	    $ret = $self;
384
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
385

386
	    $self->{value} =~ s/^\.L/$decor/;
387 388 389 390 391 392
	}
	$ret;
    }
    sub out {
	my $self = shift;

393 394
	if ($gas) {
	    my $func = ($globals{$self->{value}} or $self->{value}) . ":";
395 396
	    if ($win64	&& $current_function->{name} eq $self->{value}
			&& $current_function->{abi} eq "svr4") {
397 398 399 400
		$func .= "\n";
		$func .= "	movq	%rdi,8(%rsp)\n";
		$func .= "	movq	%rsi,16(%rsp)\n";
		$func .= "	movq	%rsp,%rax\n";
401
		$func .= "${decor}SEH_begin_$current_function->{name}:\n";
402 403 404 405 406 407 408 409 410 411 412
		my $narg = $current_function->{narg};
		$narg=6 if (!defined($narg));
		$func .= "	movq	%rcx,%rdi\n" if ($narg>0);
		$func .= "	movq	%rdx,%rsi\n" if ($narg>1);
		$func .= "	movq	%r8,%rdx\n"  if ($narg>2);
		$func .= "	movq	%r9,%rcx\n"  if ($narg>3);
		$func .= "	movq	40(%rsp),%r8\n" if ($narg>4);
		$func .= "	movq	48(%rsp),%r9\n" if ($narg>5);
	    }
	    $func;
	} elsif ($self->{value} ne "$current_function->{name}") {
413 414
	    # Make all labels in masm global.
	    $self->{value} .= ":" if ($masm);
415 416
	    $self->{value} . ":";
	} elsif ($win64 && $current_function->{abi} eq "svr4") {
417 418 419
	    my $func =	"$current_function->{name}" .
			($nasm ? ":" : "\tPROC $current_function->{scope}") .
			"\n";
420 421
	    $func .= "	mov	QWORD$PTR\[8+rsp\],rdi\t;WIN64 prologue\n";
	    $func .= "	mov	QWORD$PTR\[16+rsp\],rsi\n";
422
	    $func .= "	mov	rax,rsp\n";
423
	    $func .= "${decor}SEH_begin_$current_function->{name}:";
424 425
	    $func .= ":" if ($masm);
	    $func .= "\n";
426 427 428 429 430 431
	    my $narg = $current_function->{narg};
	    $narg=6 if (!defined($narg));
	    $func .= "	mov	rdi,rcx\n" if ($narg>0);
	    $func .= "	mov	rsi,rdx\n" if ($narg>1);
	    $func .= "	mov	rdx,r8\n"  if ($narg>2);
	    $func .= "	mov	rcx,r9\n"  if ($narg>3);
432 433
	    $func .= "	mov	r8,QWORD$PTR\[40+rsp\]\n" if ($narg>4);
	    $func .= "	mov	r9,QWORD$PTR\[48+rsp\]\n" if ($narg>5);
434 435
	    $func .= "\n";
	} else {
436 437
	   "$current_function->{name}".
			($nasm ? ":" : "\tPROC $current_function->{scope}");
438 439 440
	}
    }
}
F
FdaSilvaYY 已提交
441
{ package expr;		# pick up expressions
442
    sub re {
443 444 445
	my	($class, $line, $opcode) = @_;
	my	$self = {};
	my	$ret;
446

447 448
	if ($$line =~ /(^[^,]+)/) {
	    bless $self,$class;
449 450
	    $self->{value} = $1;
	    $ret = $self;
451
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
452

453 454
	    $self->{value} =~ s/\@PLT// if (!$elf);
	    $self->{value} =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
455
	    $self->{value} =~ s/\.L/$decor/g;
456
	    $self->{opcode} = $opcode;
457 458 459 460 461
	}
	$ret;
    }
    sub out {
	my $self = shift;
462
	if ($nasm && $self->{opcode}->mnemonic()=~m/^j(?![re]cxz)/) {
463 464 465 466
	    "NEAR ".$self->{value};
	} else {
	    $self->{value};
	}
467 468
    }
}
469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533
{ package cfi_directive;
    # CFI directives annotate instructions that are significant for
    # stack unwinding procedure compliant with DWARF specification,
    # see http://dwarfstd.org/. Besides naturally expected for this
    # script platform-specific filtering function, this module adds
    # three auxiliary synthetic directives not recognized by [GNU]
    # assembler:
    #
    # - .cfi_push to annotate push instructions in prologue, which
    #   translates to .cfi_adjust_cfa_offset (if needed) and
    #   .cfi_offset;
    # - .cfi_pop to annotate pop instructions in epilogue, which
    #   translates to .cfi_adjust_cfa_offset (if needed) and
    #   .cfi_restore;
    # - [and most notably] .cfi_cfa_expression which encodes
    #   DW_CFA_def_cfa_expression and passes it to .cfi_escape as
    #   byte vector;
    #
    # CFA expressions were introduced in DWARF specification version
    # 3 and describe how to deduce CFA, Canonical Frame Address. This
    # becomes handy if your stack frame is variable and you can't
    # spare register for [previous] frame pointer. Suggested directive
    # syntax is made-up mix of DWARF operator suffixes [subset of]
    # and references to registers with optional bias. Following example
    # describes offloaded *original* stack pointer at specific offset
    # from *current* stack pointer:
    #
    #   .cfi_cfa_expression     %rsp+40,deref,+8
    #
    # Final +8 has everything to do with the fact that CFA is defined
    # as reference to top of caller's stack, and on x86_64 call to
    # subroutine pushes 8-byte return address. In other words original
    # stack pointer upon entry to a subroutine is 8 bytes off from CFA.

    # Below constants are taken from "DWARF Expressions" section of the
    # DWARF specification, section is numbered 7.7 in versions 3 and 4.
    my %DW_OP_simple = (	# no-arg operators, mapped directly
	deref	=> 0x06,	dup	=> 0x12,
	drop	=> 0x13,	over	=> 0x14,
	pick	=> 0x15,	swap	=> 0x16,
	rot	=> 0x17,	xderef	=> 0x18,

	abs	=> 0x19,	and	=> 0x1a,
	div	=> 0x1b,	minus	=> 0x1c,
	mod	=> 0x1d,	mul	=> 0x1e,
	neg	=> 0x1f,	not	=> 0x20,
	or	=> 0x21,	plus	=> 0x22,
	shl	=> 0x24,	shr	=> 0x25,
	shra	=> 0x26,	xor	=> 0x27,
	);

    my %DW_OP_complex = (	# used in specific subroutines
	constu		=> 0x10,	# uleb128
	consts		=> 0x11,	# sleb128
	plus_uconst	=> 0x23,	# uleb128
	lit0 		=> 0x30,	# add 0-31 to opcode
	reg0		=> 0x50,	# add 0-31 to opcode
	breg0		=> 0x70,	# add 0-31 to opcole, sleb128
	regx		=> 0x90,	# uleb28
	fbreg		=> 0x91,	# sleb128
	bregx		=> 0x92,	# uleb128, sleb128
	piece		=> 0x93,	# uleb128
	);

    # Following constants are defined in x86_64 ABI supplement, for
J
Josh Soref 已提交
534
    # example available at https://www.uclibc.org/docs/psABI-x86_64.pdf,
535 536 537 538 539 540 541 542 543
    # see section 3.7 "Stack Unwind Algorithm".
    my %DW_reg_idx = (
	"%rax"=>0,  "%rdx"=>1,  "%rcx"=>2,  "%rbx"=>3,
	"%rsi"=>4,  "%rdi"=>5,  "%rbp"=>6,  "%rsp"=>7,
	"%r8" =>8,  "%r9" =>9,  "%r10"=>10, "%r11"=>11,
	"%r12"=>12, "%r13"=>13, "%r14"=>14, "%r15"=>15
	);

    my ($cfa_reg, $cfa_rsp);
544
    my @cfa_stack;
545 546 547

    # [us]leb128 format is variable-length integer representation base
    # 2^128, with most significant bit of each byte being 0 denoting
J
Josh Soref 已提交
548
    # *last* most significant digit. See "Variable Length Data" in the
549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614
    # DWARF specification, numbered 7.6 at least in versions 3 and 4.
    sub sleb128 {
	use integer;	# get right shift extend sign

	my $val = shift;
	my $sign = ($val < 0) ? -1 : 0;
	my @ret = ();

	while(1) {
	    push @ret, $val&0x7f;

	    # see if remaining bits are same and equal to most
	    # significant bit of the current digit, if so, it's
	    # last digit...
	    last if (($val>>6) == $sign);

	    @ret[-1] |= 0x80;
	    $val >>= 7;
	}

	return @ret;
    }
    sub uleb128 {
	my $val = shift;
	my @ret = ();

	while(1) {
	    push @ret, $val&0x7f;

	    # see if it's last significant digit...
	    last if (($val >>= 7) == 0);

	    @ret[-1] |= 0x80;
	}

	return @ret;
    }
    sub const {
	my $val = shift;

	if ($val >= 0 && $val < 32) {
            return ($DW_OP_complex{lit0}+$val);
	}
	return ($DW_OP_complex{consts}, sleb128($val));
    }
    sub reg {
	my $val = shift;

	return if ($val !~ m/^(%r\w+)(?:([\+\-])((?:0x)?[0-9a-f]+))?/);

	my $reg = $DW_reg_idx{$1};
	my $off = eval ("0 $2 $3");

	return (($DW_OP_complex{breg0} + $reg), sleb128($off));
	# Yes, we use DW_OP_bregX+0 to push register value and not
	# DW_OP_regX, because latter would require even DW_OP_piece,
	# which would be a waste under the circumstances. If you have
	# to use DWP_OP_reg, use "regx:N"...
    }
    sub cfa_expression {
	my $line = shift;
	my @ret;

	foreach my $token (split(/,\s*/,$line)) {
	    if ($token =~ /^%r/) {
		push @ret,reg($token);
615 616
	    } elsif ($token =~ /((?:0x)?[0-9a-f]+)\((%r\w+)\)/) {
		push @ret,reg("$2+$1");
617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
	    } elsif ($token =~ /(\w+):(\-?(?:0x)?[0-9a-f]+)(U?)/i) {
		my $i = 1*eval($2);
		push @ret,$DW_OP_complex{$1}, ($3 ? uleb128($i) : sleb128($i));
	    } elsif (my $i = 1*eval($token) or $token eq "0") {
		if ($token =~ /^\+/) {
		    push @ret,$DW_OP_complex{plus_uconst},uleb128($i);
		} else {
		    push @ret,const($i);
		}
	    } else {
		push @ret,$DW_OP_simple{$token};
	    }
	}

	# Finally we return DW_CFA_def_cfa_expression, 15, followed by
	# length of the expression and of course the expression itself.
	return (15,scalar(@ret),@ret);
    }
    sub re {
	my	($class, $line) = @_;
	my	$self = {};
	my	$ret;

640
	if ($$line =~ s/^\s*\.cfi_(\w+)\s*//) {
641 642 643 644 645 646 647 648 649 650 651
	    bless $self,$class;
	    $ret = $self;
	    undef $self->{value};
	    my $dir = $1;

	    SWITCH: for ($dir) {
	    # What is $cfa_rsp? Effectively it's difference between %rsp
	    # value and current CFA, Canonical Frame Address, which is
	    # why it starts with -8. Recall that CFA is top of caller's
	    # stack...
	    /startproc/	&& do {	($cfa_reg, $cfa_rsp) = ("%rsp", -8); last; };
652 653 654 655 656 657 658
	    /endproc/	&& do {	($cfa_reg, $cfa_rsp) = ("%rsp",  0);
				# .cfi_remember_state directives that are not
				# matched with .cfi_restore_state are
				# unnecessary.
				die "unpaired .cfi_remember_state" if (@cfa_stack);
				last;
			      };
659 660 661 662 663 664 665 666 667 668
	    /def_cfa_register/
			&& do {	$cfa_reg = $$line; last; };
	    /def_cfa_offset/
			&& do {	$cfa_rsp = -1*eval($$line) if ($cfa_reg eq "%rsp");
				last;
			      };
	    /adjust_cfa_offset/
			&& do {	$cfa_rsp -= 1*eval($$line) if ($cfa_reg eq "%rsp");
				last;
			      };
669
	    /def_cfa/	&& do {	if ($$line =~ /(%r\w+)\s*,\s*(.+)/) {
670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697
				    $cfa_reg = $1;
				    $cfa_rsp = -1*eval($2) if ($cfa_reg eq "%rsp");
				}
				last;
			      };
	    /push/	&& do {	$dir = undef;
				$cfa_rsp -= 8;
				if ($cfa_reg eq "%rsp") {
				    $self->{value} = ".cfi_adjust_cfa_offset\t8\n";
				}
				$self->{value} .= ".cfi_offset\t$$line,$cfa_rsp";
				last;
			      };
	    /pop/	&& do {	$dir = undef;
				$cfa_rsp += 8;
				if ($cfa_reg eq "%rsp") {
				    $self->{value} = ".cfi_adjust_cfa_offset\t-8\n";
				}
				$self->{value} .= ".cfi_restore\t$$line";
				last;
			      };
	    /cfa_expression/
			&& do {	$dir = undef;
				$self->{value} = ".cfi_escape\t" .
					join(",", map(sprintf("0x%02x", $_),
						      cfa_expression($$line)));
				last;
			      };
698 699 700 701 702 703 704 705
	    /remember_state/
			&& do {	push @cfa_stack, [$cfa_reg, $cfa_rsp];
				last;
			      };
	    /restore_state/
			&& do {	($cfa_reg, $cfa_rsp) = @{pop @cfa_stack};
				last;
			      };
706 707 708 709 710 711 712 713 714 715 716 717 718 719
	    }

	    $self->{value} = ".cfi_$dir\t$$line" if ($dir);

	    $$line = "";
	}

	return $ret;
    }
    sub out {
	my $self = shift;
	return ($elf ? $self->{value} : undef);
    }
}
720 721
{ package directive;	# pick up directives, which start with .
    sub re {
722 723 724
	my	($class, $line) = @_;
	my	$self = {};
	my	$ret;
725 726
	my	$dir;

727 728 729
	# chain-call to cfi_directive
	$ret = cfi_directive->re($line) and return $ret;

730 731
	if ($$line =~ /^\s*(\.\w+)/) {
	    bless $self,$class;
732 733 734
	    $dir = $1;
	    $ret = $self;
	    undef $self->{value};
735
	    $$line = substr($$line,@+[0]); $$line =~ s/^\s+//;
736 737 738

	    SWITCH: for ($dir) {
		/\.global|\.globl|\.extern/
739 740
			    && do { $globals{$$line} = $prefix . $$line;
				    $$line = $globals{$$line} if ($prefix);
741 742
				    last;
				  };
743
		/\.type/    && do { my ($sym,$type,$narg) = split(',',$$line);
744 745 746 747 748 749 750 751 752 753 754
				    if ($type eq "\@function") {
					undef $current_function;
					$current_function->{name} = $sym;
					$current_function->{abi}  = "svr4";
					$current_function->{narg} = $narg;
					$current_function->{scope} = defined($globals{$sym})?"PUBLIC":"PRIVATE";
				    } elsif ($type eq "\@abi-omnipotent") {
					undef $current_function;
					$current_function->{name} = $sym;
					$current_function->{scope} = defined($globals{$sym})?"PUBLIC":"PRIVATE";
				    }
755 756
				    $$line =~ s/\@abi\-omnipotent/\@function/;
				    $$line =~ s/\@function.*/\@function/;
757 758
				    last;
				  };
759
		/\.asciz/   && do { if ($$line =~ /^"(.*)"$/) {
760
					$dir  = ".byte";
761
					$$line = join(",",unpack("C*",$1),0);
762 763 764
				    }
				    last;
				  };
765
		/\.rva|\.long|\.quad/
766 767
			    && do { $$line =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
				    $$line =~ s/\.L/$decor/g;
768 769
				    last;
				  };
770 771 772
	    }

	    if ($gas) {
773
		$self->{value} = $dir . "\t" . $$line;
774 775

		if ($dir =~ /\.extern/) {
A
Andy Polyakov 已提交
776
		    $self->{value} = ""; # swallow extern
777 778 779 780 781
		} elsif (!$elf && $dir =~ /\.type/) {
		    $self->{value} = "";
		    $self->{value} = ".def\t" . ($globals{$1} or $1) . ";\t" .
				(defined($globals{$1})?".scl 2;":".scl 3;") .
				"\t.type 32;\t.endef"
782
				if ($win64 && $$line =~ /([^,]+),\@function/);
783 784 785
		} elsif (!$elf && $dir =~ /\.size/) {
		    $self->{value} = "";
		    if (defined($current_function)) {
786
			$self->{value} .= "${decor}SEH_end_$current_function->{name}:"
787 788 789 790
				if ($win64 && $current_function->{abi} eq "svr4");
			undef $current_function;
		    }
		} elsif (!$elf && $dir =~ /\.align/) {
791
		    $self->{value} = ".p2align\t" . (log($$line)/log(2));
792
		} elsif ($dir eq ".section") {
793
		    $current_segment=$$line;
794 795 796 797 798 799
		    if (!$elf && $current_segment eq ".init") {
			if	($flavour eq "macosx")	{ $self->{value} = ".mod_init_func"; }
			elsif	($flavour eq "mingw64")	{ $self->{value} = ".section\t.ctors"; }
		    }
		} elsif ($dir =~ /\.(text|data)/) {
		    $current_segment=".$1";
800
		} elsif ($dir =~ /\.hidden/) {
801
		    if    ($flavour eq "macosx")  { $self->{value} = ".private_extern\t$prefix$$line"; }
802 803
		    elsif ($flavour eq "mingw64") { $self->{value} = ""; }
		} elsif ($dir =~ /\.comm/) {
804
		    $self->{value} = "$dir\t$prefix$$line";
805
		    $self->{value} =~ s|,([0-9]+),([0-9]+)$|",$1,".log($2)/log(2)|e if ($flavour eq "macosx");
806
		}
807
		$$line = "";
808 809 810
		return $self;
	    }

811
	    # non-gas case or nasm/masm
812
	    SWITCH: for ($dir) {
813
		/\.text/    && do { my $v=undef;
814
				    if ($nasm) {
815
					$v="section	.text code align=64\n";
816 817
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
818
					$current_segment = ".text\$";
819
					$v.="$current_segment\tSEGMENT ";
820
					$v.=$masm>=$masmref ? "ALIGN(256)" : "PAGE";
821 822
					$v.=" 'CODE'";
				    }
823 824 825
				    $self->{value} = $v;
				    last;
				  };
826 827 828 829 830 831 832 833 834 835 836 837
		/\.data/    && do { my $v=undef;
				    if ($nasm) {
					$v="section	.data data align=8\n";
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
					$current_segment = "_DATA";
					$v.="$current_segment\tSEGMENT";
				    }
				    $self->{value} = $v;
				    last;
				  };
		/\.section/ && do { my $v=undef;
838 839
				    $$line =~ s/([^,]*).*/$1/;
				    $$line = ".CRT\$XCU" if ($$line eq ".init");
840
				    if ($nasm) {
841 842
					$v="section	$$line";
					if ($$line=~/\.([px])data/) {
843 844
					    $v.=" rdata align=";
					    $v.=$1 eq "p"? 4 : 8;
845
					} elsif ($$line=~/\.CRT\$/i) {
846
					    $v.=" rdata align=8";
847 848 849
					}
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
850 851
					$v.="$$line\tSEGMENT";
					if ($$line=~/\.([px])data/) {
852 853
					    $v.=" READONLY";
					    $v.=" ALIGN(".($1 eq "p" ? 4 : 8).")" if ($masm>=$masmref);
854
					} elsif ($$line=~/\.CRT\$/i) {
855 856
					    $v.=" READONLY ";
					    $v.=$masm>=$masmref ? "ALIGN(8)" : "DWORD";
857 858
					}
				    }
859
				    $current_segment = $$line;
860 861 862
				    $self->{value} = $v;
				    last;
				  };
863
		/\.extern/  && do { $self->{value}  = "EXTERN\t".$$line;
864 865 866
				    $self->{value} .= ":NEAR" if ($masm);
				    last;
				  };
867 868
		/\.globl|.global/
			    && do { $self->{value}  = $masm?"PUBLIC":"global";
869
				    $self->{value} .= "\t".$$line;
870 871
				    last;
				  };
872
		/\.size/    && do { if (defined($current_function)) {
A
Andy Polyakov 已提交
873 874
					undef $self->{value};
					if ($current_function->{abi} eq "svr4") {
875
					    $self->{value}="${decor}SEH_end_$current_function->{name}:";
A
Andy Polyakov 已提交
876 877
					    $self->{value}.=":\n" if($masm);
					}
878
					$self->{value}.="$current_function->{name}\tENDP" if($masm && $current_function->{name});
879 880 881 882
					undef $current_function;
				    }
				    last;
				  };
883
		/\.align/   && do { my $max = ($masm && $masm>=$masmref) ? 256 : 4096;
884
				    $self->{value} = "ALIGN\t".($$line>$max?$max:$$line);
885 886
				    last;
				  };
887
		/\.(value|long|rva|quad)/
888
			    && do { my $sz  = substr($1,0,1);
889
				    my @arr = split(/,\s*/,$$line);
890
				    my $last = pop(@arr);
891
				    my $conv = sub  {	my $var=shift;
892
							$var=~s/^(0b[0-1]+)/oct($1)/eig;
893
							$var=~s/^0x([0-9a-f]+)/0$1h/ig if ($masm);
894
							if ($sz eq "D" && ($current_segment=~/.[px]data/ || $dir eq ".rva"))
895
							{ $var=~s/^([_a-z\$\@][_a-z0-9\$\@]*)/$nasm?"$1 wrt ..imagebase":"imagerel $1"/egi; }
896
							$var;
897
						    };
898

899
				    $sz =~ tr/bvlrq/BWDDQ/;
900
				    $self->{value} = "\tD$sz\t";
901 902
				    for (@arr) { $self->{value} .= &$conv($_).","; }
				    $self->{value} .= &$conv($last);
903 904
				    last;
				  };
905
		/\.byte/    && do { my @str=split(/,\s*/,$$line);
906
				    map(s/(0b[0-1]+)/oct($1)/eig,@str);
907
				    map(s/0x([0-9a-f]+)/0$1h/ig,@str) if ($masm);
908
				    while ($#str>15) {
909
					$self->{value}.="DB\t"
910 911
						.join(",",@str[0..15])."\n";
					foreach (0..15) { shift @str; }
912
				    }
913 914
				    $self->{value}.="DB\t"
						.join(",",@str) if (@str);
915 916
				    last;
				  };
917
		/\.comm/    && do { my @str=split(/,\s*/,$$line);
918 919
				    my $v=undef;
				    if ($nasm) {
920
					$v.="common	$prefix@str[0] @str[1]";
921 922
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
923
					$current_segment = "_DATA";
924 925 926 927 928 929
					$v.="$current_segment\tSEGMENT\n";
					$v.="COMM	@str[0]:DWORD:".@str[1]/4;
				    }
				    $self->{value} = $v;
				    last;
				  };
930
	    }
931
	    $$line = "";
932 933 934 935 936 937 938 939 940 941
	}

	$ret;
    }
    sub out {
	my $self = shift;
	$self->{value};
    }
}

942 943 944 945 946 947 948
# Upon initial x86_64 introduction SSE>2 extensions were not introduced
# yet. In order not to be bothered by tracing exact assembler versions,
# but at the same time to provide a bare security minimum of AES-NI, we
# hard-code some instructions. Extensions past AES-NI on the other hand
# are traced by examining assembler version in individual perlasm
# modules...

949 950 951
my %regrm = (	"%eax"=>0, "%ecx"=>1, "%edx"=>2, "%ebx"=>3,
		"%esp"=>4, "%ebp"=>5, "%esi"=>6, "%edi"=>7	);

952 953 954 955 956 957 958 959 960
sub rex {
 my $opcode=shift;
 my ($dst,$src,$rex)=@_;

   $rex|=0x04 if($dst>=8);
   $rex|=0x01 if($src>=8);
   push @$opcode,($rex|0x40) if ($rex);
}

961 962 963
my $movq = sub {	# elderly gas can't handle inter-register movq
  my $arg = shift;
  my @opcode=(0x66);
964
    if ($arg =~ /%xmm([0-9]+),\s*%r(\w+)/) {
965 966 967 968 969 970
	my ($src,$dst)=($1,$2);
	if ($dst !~ /[0-9]+/)	{ $dst = $regrm{"%e$dst"}; }
	rex(\@opcode,$src,$dst,0x8);
	push @opcode,0x0f,0x7e;
	push @opcode,0xc0|(($src&7)<<3)|($dst&7);	# ModR/M
	@opcode;
971
    } elsif ($arg =~ /%r(\w+),\s*%xmm([0-9]+)/) {
972 973 974 975 976 977 978 979 980 981 982
	my ($src,$dst)=($2,$1);
	if ($dst !~ /[0-9]+/)	{ $dst = $regrm{"%e$dst"}; }
	rex(\@opcode,$src,$dst,0x8);
	push @opcode,0x0f,0x6e;
	push @opcode,0xc0|(($src&7)<<3)|($dst&7);	# ModR/M
	@opcode;
    } else {
	();
    }
};

983
my $pextrd = sub {
984
    if (shift =~ /\$([0-9]+),\s*%xmm([0-9]+),\s*(%\w+)/) {
985
      my @opcode=(0x66);
986 987 988
	my $imm=$1;
	my $src=$2;
	my $dst=$3;
989 990 991 992 993 994
	if ($dst =~ /%r([0-9]+)d/)	{ $dst = $1; }
	elsif ($dst =~ /%e/)		{ $dst = $regrm{$dst}; }
	rex(\@opcode,$src,$dst);
	push @opcode,0x0f,0x3a,0x16;
	push @opcode,0xc0|(($src&7)<<3)|($dst&7);	# ModR/M
	push @opcode,$imm;
995
	@opcode;
996
    } else {
997
	();
998
    }
999
};
1000 1001

my $pinsrd = sub {
1002
    if (shift =~ /\$([0-9]+),\s*(%\w+),\s*%xmm([0-9]+)/) {
1003
      my @opcode=(0x66);
1004 1005 1006
	my $imm=$1;
	my $src=$2;
	my $dst=$3;
1007
	if ($src =~ /%r([0-9]+)/)	{ $src = $1; }
1008 1009 1010 1011 1012
	elsif ($src =~ /%e/)		{ $src = $regrm{$src}; }
	rex(\@opcode,$dst,$src);
	push @opcode,0x0f,0x3a,0x22;
	push @opcode,0xc0|(($dst&7)<<3)|($src&7);	# ModR/M
	push @opcode,$imm;
1013
	@opcode;
1014
    } else {
1015
	();
1016
    }
1017
};
1018 1019

my $pshufb = sub {
1020
    if (shift =~ /%xmm([0-9]+),\s*%xmm([0-9]+)/) {
1021
      my @opcode=(0x66);
1022
	rex(\@opcode,$2,$1);
1023
	push @opcode,0x0f,0x38,0x00;
1024 1025
	push @opcode,0xc0|($1&7)|(($2&7)<<3);		# ModR/M
	@opcode;
1026
    } else {
1027
	();
1028
    }
1029
};
1030

1031
my $palignr = sub {
1032
    if (shift =~ /\$([0-9]+),\s*%xmm([0-9]+),\s*%xmm([0-9]+)/) {
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
      my @opcode=(0x66);
	rex(\@opcode,$3,$2);
	push @opcode,0x0f,0x3a,0x0f;
	push @opcode,0xc0|($2&7)|(($3&7)<<3);		# ModR/M
	push @opcode,$1;
	@opcode;
    } else {
	();
    }
};

my $pclmulqdq = sub {
    if (shift =~ /\$([x0-9a-f]+),\s*%xmm([0-9]+),\s*%xmm([0-9]+)/) {
      my @opcode=(0x66);
	rex(\@opcode,$3,$2);
	push @opcode,0x0f,0x3a,0x44;
	push @opcode,0xc0|($2&7)|(($3&7)<<3);		# ModR/M
	my $c=$1;
	push @opcode,$c=~/^0/?oct($c):$c;
	@opcode;
    } else {
	();
    }
};

1058 1059 1060 1061 1062
my $rdrand = sub {
    if (shift =~ /%[er](\w+)/) {
      my @opcode=();
      my $dst=$1;
	if ($dst !~ /[0-9]+/) { $dst = $regrm{"%e$dst"}; }
1063
	rex(\@opcode,0,$dst,8);
1064 1065 1066 1067 1068 1069 1070
	push @opcode,0x0f,0xc7,0xf0|($dst&7);
	@opcode;
    } else {
	();
    }
};

1071 1072 1073 1074 1075
my $rdseed = sub {
    if (shift =~ /%[er](\w+)/) {
      my @opcode=();
      my $dst=$1;
	if ($dst !~ /[0-9]+/) { $dst = $regrm{"%e$dst"}; }
1076
	rex(\@opcode,0,$dst,8);
1077 1078 1079 1080 1081 1082 1083
	push @opcode,0x0f,0xc7,0xf8|($dst&7);
	@opcode;
    } else {
	();
    }
};

1084 1085 1086 1087
# Not all AVX-capable assemblers recognize AMD XOP extension. Since we
# are using only two instructions hand-code them in order to be excused
# from chasing assembler versions...

1088
sub rxb {
1089
 my $opcode=shift;
1090 1091 1092 1093 1094 1095
 my ($dst,$src1,$src2,$rxb)=@_;

   $rxb|=0x7<<5;
   $rxb&=~(0x04<<5) if($dst>=8);
   $rxb&=~(0x01<<5) if($src1>=8);
   $rxb&=~(0x02<<5) if($src2>=8);
1096
   push @$opcode,$rxb;
1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
}

my $vprotd = sub {
    if (shift =~ /\$([x0-9a-f]+),\s*%xmm([0-9]+),\s*%xmm([0-9]+)/) {
      my @opcode=(0x8f);
	rxb(\@opcode,$3,$2,-1,0x08);
	push @opcode,0x78,0xc2;
	push @opcode,0xc0|($2&7)|(($3&7)<<3);		# ModR/M
	my $c=$1;
	push @opcode,$c=~/^0/?oct($c):$c;
	@opcode;
    } else {
	();
    }
};

my $vprotq = sub {
    if (shift =~ /\$([x0-9a-f]+),\s*%xmm([0-9]+),\s*%xmm([0-9]+)/) {
      my @opcode=(0x8f);
	rxb(\@opcode,$3,$2,-1,0x08);
	push @opcode,0x78,0xc3;
	push @opcode,0xc0|($2&7)|(($3&7)<<3);		# ModR/M
	my $c=$1;
	push @opcode,$c=~/^0/?oct($c):$c;
	@opcode;
    } else {
	();
    }
};

1127 1128 1129
# Intel Control-flow Enforcement Technology extension. All functions and
# indirect branch targets will have to start with this instruction...

1130 1131 1132 1133
my $endbranch = sub {
    (0xf3,0x0f,0x1e,0xfa);
};

1134 1135
########################################################################

1136 1137 1138
if ($nasm) {
    print <<___;
default	rel
1139
%define XMMWORD
1140 1141
%define YMMWORD
%define ZMMWORD
1142 1143 1144 1145 1146 1147
___
} elsif ($masm) {
    print <<___;
OPTION	DOTNAME
___
}
1148
while(defined(my $line=<>)) {
1149

1150
    $line =~ s|\R$||;           # Better chomp
1151

1152 1153 1154
    $line =~ s|[#!].*$||;	# get rid of asm-style comments...
    $line =~ s|/\*.*\*/||;	# ... and C-style comments...
    $line =~ s|^\s+||;		# ... and skip white spaces in beginning
A
Andy Polyakov 已提交
1155
    $line =~ s|\s+$||;		# ... and at the end
1156

1157
    if (my $label=label->re(\$line))	{ print $label->out(); }
1158

1159 1160 1161
    if (my $directive=directive->re(\$line)) {
	printf "%s",$directive->out();
    } elsif (my $opcode=opcode->re(\$line)) {
1162
	my $asm = eval("\$".$opcode->mnemonic());
1163

1164
	if ((ref($asm) eq 'CODE') && scalar(my @bytes=&$asm($line))) {
1165 1166 1167 1168
	    print $gas?".byte\t":"DB\t",join(',',@bytes),"\n";
	    next;
	}

1169
	my @args;
1170
	ARGUMENT: while (1) {
1171
	    my $arg;
1172

1173 1174 1175 1176 1177
	    ($arg=register->re(\$line, $opcode))||
	    ($arg=const->re(\$line))		||
	    ($arg=ea->re(\$line, $opcode))	||
	    ($arg=expr->re(\$line, $opcode))	||
	    last ARGUMENT;
1178

1179
	    push @args,$arg;
1180

1181
	    last ARGUMENT if ($line !~ /^,/);
1182

1183
	    $line =~ s/^,\s*//;
1184 1185
	} # ARGUMENT:

1186 1187
	if ($#args>=0) {
	    my $insn;
1188
	    my $sz=$opcode->size();
1189

1190
	    if ($gas) {
1191
		$insn = $opcode->out($#args>=1?$args[$#args]->size():$sz);
1192
		@args = map($_->out($sz),@args);
1193
		printf "\t%s\t%s",$insn,join(",",@args);
1194
	    } else {
1195
		$insn = $opcode->out();
1196 1197
		foreach (@args) {
		    my $arg = $_->out();
1198 1199
		    # $insn.=$sz compensates for movq, pinsrw, ...
		    if ($arg =~ /^xmm[0-9]+$/) { $insn.=$sz; $sz="x" if(!$sz); last; }
A
Andy Polyakov 已提交
1200
		    if ($arg =~ /^ymm[0-9]+$/) { $insn.=$sz; $sz="y" if(!$sz); last; }
1201
		    if ($arg =~ /^zmm[0-9]+$/) { $insn.=$sz; $sz="z" if(!$sz); last; }
1202
		    if ($arg =~ /^mm[0-9]+$/)  { $insn.=$sz; $sz="q" if(!$sz); last; }
1203
		}
1204
		@args = reverse(@args);
1205
		undef $sz if ($nasm && $opcode->mnemonic() eq "lea");
1206
		printf "\t%s\t%s",$insn,join(",",map($_->out($sz),@args));
1207 1208 1209 1210 1211 1212 1213 1214 1215
	    }
	} else {
	    printf "\t%s",$opcode->out();
	}
    }

    print $line,"\n";
}

1216 1217
print "\n$current_segment\tENDS\n"	if ($current_segment && $masm);
print "END\n"				if ($masm);
1218

1219
close STDOUT or die "error closing STDOUT: $!";
1220

1221
#################################################
1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240
# Cross-reference x86_64 ABI "card"
#
# 		Unix		Win64
# %rax		*		*
# %rbx		-		-
# %rcx		#4		#1
# %rdx		#3		#2
# %rsi		#2		-
# %rdi		#1		-
# %rbp		-		-
# %rsp		-		-
# %r8		#5		#3
# %r9		#6		#4
# %r10		*		*
# %r11		*		*
# %r12		-		-
# %r13		-		-
# %r14		-		-
# %r15		-		-
1241
#
1242 1243 1244 1245 1246
# (*)	volatile register
# (-)	preserved by callee
# (#)	Nth argument, volatile
#
# In Unix terms top of stack is argument transfer area for arguments
F
FdaSilvaYY 已提交
1247
# which could not be accommodated in registers. Or in other words 7th
1248 1249 1250 1251 1252 1253 1254 1255 1256 1257
# [integer] argument resides at 8(%rsp) upon function entry point.
# 128 bytes above %rsp constitute a "red zone" which is not touched
# by signal handlers and can be used as temporal storage without
# allocating a frame.
#
# In Win64 terms N*8 bytes on top of stack is argument transfer area,
# which belongs to/can be overwritten by callee. N is the number of
# arguments passed to callee, *but* not less than 4! This means that
# upon function entry point 5th argument resides at 40(%rsp), as well
# as that 32 bytes from 8(%rsp) can always be used as temporal
1258 1259
# storage [without allocating a frame]. One can actually argue that
# one can assume a "red zone" above stack pointer under Win64 as well.
A
Andy Polyakov 已提交
1260 1261
# Point is that at apparently no occasion Windows kernel would alter
# the area above user stack pointer in true asynchronous manner...
1262 1263
#
# All the above means that if assembler programmer adheres to Unix
F
FdaSilvaYY 已提交
1264
# register and stack layout, but disregards the "red zone" existence,
1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284
# it's possible to use following prologue and epilogue to "gear" from
# Unix to Win64 ABI in leaf functions with not more than 6 arguments.
#
# omnipotent_function:
# ifdef WIN64
#	movq	%rdi,8(%rsp)
#	movq	%rsi,16(%rsp)
#	movq	%rcx,%rdi	; if 1st argument is actually present
#	movq	%rdx,%rsi	; if 2nd argument is actually ...
#	movq	%r8,%rdx	; if 3rd argument is ...
#	movq	%r9,%rcx	; if 4th argument ...
#	movq	40(%rsp),%r8	; if 5th ...
#	movq	48(%rsp),%r9	; if 6th ...
# endif
#	...
# ifdef WIN64
#	movq	8(%rsp),%rdi
#	movq	16(%rsp),%rsi
# endif
#	ret
1285
#
1286 1287 1288
#################################################
# Win64 SEH, Structured Exception Handling.
#
1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302
# Unlike on Unix systems(*) lack of Win64 stack unwinding information
# has undesired side-effect at run-time: if an exception is raised in
# assembler subroutine such as those in question (basically we're
# referring to segmentation violations caused by malformed input
# parameters), the application is briskly terminated without invoking
# any exception handlers, most notably without generating memory dump
# or any user notification whatsoever. This poses a problem. It's
# possible to address it by registering custom language-specific
# handler that would restore processor context to the state at
# subroutine entry point and return "exception is not handled, keep
# unwinding" code. Writing such handler can be a challenge... But it's
# doable, though requires certain coding convention. Consider following
# snippet:
#
1303
# .type	function,@function
1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321
# function:
#	movq	%rsp,%rax	# copy rsp to volatile register
#	pushq	%r15		# save non-volatile registers
#	pushq	%rbx
#	pushq	%rbp
#	movq	%rsp,%r11
#	subq	%rdi,%r11	# prepare [variable] stack frame
#	andq	$-64,%r11
#	movq	%rax,0(%r11)	# check for exceptions
#	movq	%r11,%rsp	# allocate [variable] stack frame
#	movq	%rax,0(%rsp)	# save original rsp value
# magic_point:
#	...
#	movq	0(%rsp),%rcx	# pull original rsp value
#	movq	-24(%rcx),%rbp	# restore non-volatile registers
#	movq	-16(%rcx),%rbx
#	movq	-8(%rcx),%r15
#	movq	%rcx,%rsp	# restore original rsp
1322
# magic_epilogue:
1323
#	ret
1324
# .size function,.-function
1325 1326 1327 1328 1329 1330 1331 1332 1333
#
# The key is that up to magic_point copy of original rsp value remains
# in chosen volatile register and no non-volatile register, except for
# rsp, is modified. While past magic_point rsp remains constant till
# the very end of the function. In this case custom language-specific
# exception handler would look like this:
#
# EXCEPTION_DISPOSITION handler (EXCEPTION_RECORD *rec,ULONG64 frame,
#		CONTEXT *context,DISPATCHER_CONTEXT *disp)
1334
# {	ULONG64 *rsp = (ULONG64 *)context->Rax;
1335 1336 1337 1338 1339 1340 1341 1342 1343 1344
#	ULONG64  rip = context->Rip;
#
#	if (rip >= magic_point)
#	{   rsp = (ULONG64 *)context->Rsp;
#	    if (rip < magic_epilogue)
#	    {	rsp = (ULONG64 *)rsp[0];
#		context->Rbp = rsp[-3];
#		context->Rbx = rsp[-2];
#		context->R15 = rsp[-1];
#	    }
1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378
#	}
#	context->Rsp = (ULONG64)rsp;
#	context->Rdi = rsp[1];
#	context->Rsi = rsp[2];
#
#	memcpy (disp->ContextRecord,context,sizeof(CONTEXT));
#	RtlVirtualUnwind(UNW_FLAG_NHANDLER,disp->ImageBase,
#		dips->ControlPc,disp->FunctionEntry,disp->ContextRecord,
#		&disp->HandlerData,&disp->EstablisherFrame,NULL);
#	return ExceptionContinueSearch;
# }
#
# It's appropriate to implement this handler in assembler, directly in
# function's module. In order to do that one has to know members'
# offsets in CONTEXT and DISPATCHER_CONTEXT structures and some constant
# values. Here they are:
#
#	CONTEXT.Rax				120
#	CONTEXT.Rcx				128
#	CONTEXT.Rdx				136
#	CONTEXT.Rbx				144
#	CONTEXT.Rsp				152
#	CONTEXT.Rbp				160
#	CONTEXT.Rsi				168
#	CONTEXT.Rdi				176
#	CONTEXT.R8				184
#	CONTEXT.R9				192
#	CONTEXT.R10				200
#	CONTEXT.R11				208
#	CONTEXT.R12				216
#	CONTEXT.R13				224
#	CONTEXT.R14				232
#	CONTEXT.R15				240
#	CONTEXT.Rip				248
1379
#	CONTEXT.Xmm6				512
1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391
#	sizeof(CONTEXT)				1232
#	DISPATCHER_CONTEXT.ControlPc		0
#	DISPATCHER_CONTEXT.ImageBase		8
#	DISPATCHER_CONTEXT.FunctionEntry	16
#	DISPATCHER_CONTEXT.EstablisherFrame	24
#	DISPATCHER_CONTEXT.TargetIp		32
#	DISPATCHER_CONTEXT.ContextRecord	40
#	DISPATCHER_CONTEXT.LanguageHandler	48
#	DISPATCHER_CONTEXT.HandlerData		56
#	UNW_FLAG_NHANDLER			0
#	ExceptionContinueSearch			1
#
1392 1393 1394
# In order to tie the handler to the function one has to compose
# couple of structures: one for .xdata segment and one for .pdata.
#
1395
# UNWIND_INFO structure for .xdata segment would be
1396 1397 1398
#
# function_unwind_info:
#	.byte	9,0,0,0
1399
#	.rva	handler
1400 1401 1402 1403 1404 1405
#
# This structure designates exception handler for a function with
# zero-length prologue, no stack frame or frame register.
#
# To facilitate composing of .pdata structures, auto-generated "gear"
# prologue copies rsp value to rax and denotes next instruction with
1406
# .LSEH_begin_{function_name} label. This essentially defines the SEH
1407 1408 1409
# styling rule mentioned in the beginning. Position of this label is
# chosen in such manner that possible exceptions raised in the "gear"
# prologue would be accounted to caller and unwound from latter's frame.
1410
# End of function is marked with respective .LSEH_end_{function_name}
1411 1412
# label. To summarize, .pdata segment would contain
#
1413 1414 1415
#	.rva	.LSEH_begin_function
#	.rva	.LSEH_end_function
#	.rva	function_unwind_info
1416
#
1417
# Reference to function_unwind_info from .xdata segment is the anchor.
1418
# In case you wonder why references are 32-bit .rvas and not 64-bit
1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435
# .quads. References put into these two segments are required to be
# *relative* to the base address of the current binary module, a.k.a.
# image base. No Win64 module, be it .exe or .dll, can be larger than
# 2GB and thus such relative references can be and are accommodated in
# 32 bits.
#
# Having reviewed the example function code, one can argue that "movq
# %rsp,%rax" above is redundant. It is not! Keep in mind that on Unix
# rax would contain an undefined value. If this "offends" you, use
# another register and refrain from modifying rax till magic_point is
# reached, i.e. as if it was a non-volatile register. If more registers
# are required prior [variable] frame setup is completed, note that
# nobody says that you can have only one "magic point." You can
# "liberate" non-volatile registers by denoting last stack off-load
# instruction and reflecting it in finer grade unwind logic in handler.
# After all, isn't it why it's called *language-specific* handler...
#
1436 1437 1438 1439 1440 1441
# SE handlers are also involved in unwinding stack when executable is
# profiled or debugged. Profiling implies additional limitations that
# are too subtle to discuss here. For now it's sufficient to say that
# in order to simplify handlers one should either a) offload original
# %rsp to stack (like discussed above); or b) if you have a register to
# spare for frame pointer, choose volatile one.
1442 1443 1444
#
# (*)	Note that we're talking about run-time, not debug-time. Lack of
#	unwind information makes debugging hard on both Windows and
J
Josh Soref 已提交
1445
#	Unix. "Unlike" refers to the fact that on Unix signal handler
1446 1447
#	will always be invoked, core dumped and appropriate exit code
#	returned to parent (for user notification).