x86_64-xlate.pl 33.1 KB
Newer Older
1 2
#!/usr/bin/env perl

3
# Ascetic x86_64 AT&T to MASM/NASM assembler translator by <appro>.
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
#
# 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.
#
26 27
# 1. Adhere to Unix register and stack layout [see cross-reference
#    ABI "card" at the end for explanation].
28 29 30 31
# 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
32
#    unified Win64 prologue and epilogue automatically. If you want
33
#    to take care of ABI differences yourself, tag functions as
34 35 36 37 38 39 40 41 42 43
#    ".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:-)
44 45 46 47 48 49
# 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
#    modes. To support legacy code a synthetic directive, .picmeup,
#    is implemented. It puts address of the *next* instruction into
#    target register, e.g.:
50 51 52
#
#		.picmeup	%rax
#		lea		.Label-.(%rax),%rax
53 54 55 56
#
# 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.
57
# 9. .init segment is allowed to contain calls to functions only.
58 59
# a. If function accepts more than 4 arguments *and* >4th argument
#    is declared as non 64-bit value, do clear its upper part.
60

61 62 63
my $flavour = shift;
my $output  = shift;
if ($flavour =~ /\./) { $output = $flavour; undef $flavour; }
A
Andy Polyakov 已提交
64 65 66 67 68 69 70

{ my ($stddev,$stdino,@junk)=stat(STDOUT);
  my ($outdev,$outino,@junk)=stat($output);

    open STDOUT,">$output" || die "can't open $output: $!"
	if ($stddev!=$outdev || $stdino!=$outino);
}
71

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

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

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

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

my $current_segment;
my $current_function;
105
my %globals;
106 107 108 109 110 111 112

{ package opcode;	# pick up opcodes
    sub re {
	my	$self = shift;	# single instance in enough...
	local	*line = shift;
	undef	$ret;

A
Andy Polyakov 已提交
113
	if ($line =~ /^([a-z][a-z0-9]*)/i) {
114 115 116 117 118
	    $self->{op} = $1;
	    $ret = $self;
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;

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

	if ($line =~ /^\$([^,]+)/) {
	    $self->{value} = $1;
	    $ret = $self;
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;
	}
	$ret;
    }
    sub out {
    	my $self = shift;
201

202
	if ($gas) {
203 204
	    # Solaris /usr/ccs/bin/as can't handle multiplications
	    # in $self->{value}
205
	    $self->{value} =~ s/(?<![\w\$\.])(0x?[0-9a-f]+)/oct($1)/egi;
206
	    $self->{value} =~ s/([0-9]+\s*[\*\/\%]\s*[0-9]+)/eval($1)/eg;
207 208
	    sprintf "\$%s",$self->{value};
	} else {
209 210
	    $self->{value} =~ s/(0b[0-1]+)/oct($1)/eig;
	    $self->{value} =~ s/0x([0-9a-f]+)/0$1h/ig if ($masm);
211 212
	    sprintf "%s",$self->{value};
	}
213 214 215 216 217 218 219 220
    }
}
{ package ea;		# pick up effective addresses: expr(%reg,%reg,scale)
    sub re {
	my	$self = shift;	# single instance in enough...
	local	*line = shift;
	undef	$ret;

221 222 223 224 225
	# optional * ---vvv--- appears in indirect jmp/call
	if ($line =~ /^(\*?)([^\(,]*)\(([%\w,]+)\)/) {
	    $self->{asterisk} = $1;
	    $self->{label} = $2;
	    ($self->{base},$self->{index},$self->{scale})=split(/,/,$3);
226 227 228 229
	    $self->{scale} = 1 if (!defined($self->{scale}));
	    $ret = $self;
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;

230 231 232 233
	    if ($win64 && $self->{label} =~ s/\@GOTPCREL//) {
		die if (opcode->mnemonic() ne "mov");
		opcode->mnemonic("lea");
	    }
234 235 236 237 238 239 240 241 242 243
	    $self->{base}  =~ s/^%//;
	    $self->{index} =~ s/^%// if (defined($self->{index}));
	}
	$ret;
    }
    sub size {}
    sub out {
    	my $self = shift;
	my $sz = shift;

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

247 248 249 250 251 252
	# 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/;

253 254 255 256 257 258 259
	# Solaris /usr/ccs/bin/as can't handle multiplications
	# in $self->{label}, new gas requires sign extension...
	use integer;
	$self->{label} =~ s/(?<![\w\$\.])(0x?[0-9a-f]+)/oct($1)/egi;
	$self->{label} =~ s/([0-9]+\s*[\*\/\%]\s*[0-9]+)/eval($1)/eg;
	$self->{label} =~ s/([0-9]+)/$1<<32>>32/eg;

260
	if ($gas) {
261
	    $self->{label} =~ s/^___imp_/__imp__/   if ($flavour eq "mingw64");
262

263
	    if (defined($self->{index})) {
264 265 266
		sprintf "%s%s(%s,%%%s,%d)",$self->{asterisk},
					$self->{label},
					$self->{base}?"%$self->{base}":"",
267
					$self->{index},$self->{scale};
268
	    } else {
269
		sprintf "%s%s(%%%s)",	$self->{asterisk},$self->{label},$self->{base};
270 271
	    }
	} else {
272
	    %szmap = (	b=>"BYTE$PTR", w=>"WORD$PTR", l=>"DWORD$PTR",
273
	    		q=>"QWORD$PTR",o=>"OWORD$PTR",x=>"XMMWORD$PTR" );
274

275
	    $self->{label} =~ s/\./\$/g;
276
	    $self->{label} =~ s/(?<![\w\$\.])0x([0-9a-f]+)/0$1h/ig;
277
	    $self->{label} = "($self->{label})" if ($self->{label} =~ /[\*\+\-\/]/);
278
	    $sz="q" if ($self->{asterisk} || opcode->mnemonic() eq "movq");
279

280
	    if (defined($self->{index})) {
281
		sprintf "%s[%s%s*%d%s]",$szmap{$sz},
282
					$self->{label}?"$self->{label}+":"",
283
					$self->{index},$self->{scale},
284
					$self->{base}?"+$self->{base}":"";
A
Andy Polyakov 已提交
285
	    } elsif ($self->{base} eq "rip") {
286
		sprintf "%s[%s]",$szmap{$sz},$self->{label};
287
	    } else {
288 289 290
		sprintf "%s[%s%s]",$szmap{$sz},
					$self->{label}?"$self->{label}+":"",
					$self->{base};
291 292 293 294 295 296 297 298 299 300 301
	    }
	}
    }
}
{ package register;	# pick up registers, which start with %.
    sub re {
	my	$class = shift;	# muliple instances...
	my	$self = {};
	local	*line = shift;
	undef	$ret;

302 303
	# optional * ---vvv--- appears in indirect jmp/call
	if ($line =~ /^(\*?)%(\w+)/) {
304
	    bless $self,$class;
305 306
	    $self->{asterisk} = $1;
	    $self->{value} = $2;
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328
	    $ret = $self;
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;
	}
	$ret;
    }
    sub size {
	my	$self = shift;
	undef	$ret;

	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;
329
	if ($gas)	{ sprintf "%s%%%s",$self->{asterisk},$self->{value}; }
330
	else		{ $self->{value}; }
331 332 333 334 335 336 337 338
    }
}
{ package label;	# pick up labels, which end with :
    sub re {
	my	$self = shift;	# single instance is enough...
	local	*line = shift;
	undef	$ret;

339
	if ($line =~ /(^[\.\w]+)\:/) {
340 341 342 343
	    $self->{value} = $1;
	    $ret = $self;
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;

344
	    $self->{value} =~ s/^\.L/$decor/;
345 346 347 348 349 350
	}
	$ret;
    }
    sub out {
	my $self = shift;

351 352 353 354 355 356 357 358 359
	if ($gas) {
	    my $func = ($globals{$self->{value}} or $self->{value}) . ":";
	    if ($win64	&&
			$current_function->{name} eq $self->{value} &&
			$current_function->{abi} eq "svr4") {
		$func .= "\n";
		$func .= "	movq	%rdi,8(%rsp)\n";
		$func .= "	movq	%rsi,16(%rsp)\n";
		$func .= "	movq	%rsp,%rax\n";
360
		$func .= "${decor}SEH_begin_$current_function->{name}:\n";
361 362 363 364 365 366 367 368 369 370 371
		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}") {
372
	    $self->{value} .= ":" if ($masm && $ret!~m/^\$/);
373 374
	    $self->{value} . ":";
	} elsif ($win64 && $current_function->{abi} eq "svr4") {
375 376 377 378 379
	    my $func =	"$current_function->{name}" .
			($nasm ? ":" : "\tPROC $current_function->{scope}") .
			"\n";
	    $func .= "	mov	QWORD${PTR}[8+rsp],rdi\t;WIN64 prologue\n";
	    $func .= "	mov	QWORD${PTR}[16+rsp],rsi\n";
380
	    $func .= "	mov	rax,rsp\n";
381
	    $func .= "${decor}SEH_begin_$current_function->{name}:";
382 383
	    $func .= ":" if ($masm);
	    $func .= "\n";
384 385 386 387 388 389
	    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);
390 391
	    $func .= "	mov	r8,QWORD${PTR}[40+rsp]\n" if ($narg>4);
	    $func .= "	mov	r9,QWORD${PTR}[48+rsp]\n" if ($narg>5);
392 393
	    $func .= "\n";
	} else {
394 395
	   "$current_function->{name}".
			($nasm ? ":" : "\tPROC $current_function->{scope}");
396 397 398 399 400 401 402 403 404 405 406 407 408 409
	}
    }
}
{ package expr;		# pick up expressioins
    sub re {
	my	$self = shift;	# single instance is enough...
	local	*line = shift;
	undef	$ret;

	if ($line =~ /(^[^,]+)/) {
	    $self->{value} = $1;
	    $ret = $self;
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;

410 411
	    $self->{value} =~ s/\@PLT// if (!$elf);
	    $self->{value} =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
412
	    $self->{value} =~ s/\.L/$decor/g;
413 414 415 416 417
	}
	$ret;
    }
    sub out {
	my $self = shift;
418 419 420 421 422
	if ($nasm && opcode->mnemonic()=~m/^j/) {
	    "NEAR ".$self->{value};
	} else {
	    $self->{value};
	}
423 424 425 426 427 428 429 430
    }
}
{ package directive;	# pick up directives, which start with .
    sub re {
	my	$self = shift;	# single instance is enough...
	local	*line = shift;
	undef	$ret;
	my	$dir;
431 432 433 434 435 436 437 438 439
	my	%opcode =	# lea 2f-1f(%rip),%dst; 1: nop; 2:
		(	"%rax"=>0x01058d48,	"%rcx"=>0x010d8d48,
			"%rdx"=>0x01158d48,	"%rbx"=>0x011d8d48,
			"%rsp"=>0x01258d48,	"%rbp"=>0x012d8d48,
			"%rsi"=>0x01358d48,	"%rdi"=>0x013d8d48,
			"%r8" =>0x01058d4c,	"%r9" =>0x010d8d4c,
			"%r10"=>0x01158d4c,	"%r11"=>0x011d8d4c,
			"%r12"=>0x01258d4c,	"%r13"=>0x012d8d4c,
			"%r14"=>0x01358d4c,	"%r15"=>0x013d8d4c	);
440 441

	if ($line =~ /^\s*(\.\w+)/) {
442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480
	    $dir = $1;
	    $ret = $self;
	    undef $self->{value};
	    $line = substr($line,@+[0]); $line =~ s/^\s+//;

	    SWITCH: for ($dir) {
		/\.picmeup/ && do { if ($line =~ /(%r[\w]+)/i) {
			    		$dir="\t.long";
					$line=sprintf "0x%x,0x90000000",$opcode{$1};
				    }
				    last;
				  };
		/\.global|\.globl|\.extern/
			    && do { $globals{$line} = $prefix . $line;
				    $line = $globals{$line} if ($prefix);
				    last;
				  };
		/\.type/    && do { ($sym,$type,$narg) = split(',',$line);
				    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";
				    }
				    $line =~ s/\@abi\-omnipotent/\@function/;
				    $line =~ s/\@function.*/\@function/;
				    last;
				  };
		/\.asciz/   && do { if ($line =~ /^"(.*)"$/) {
					$dir  = ".byte";
					$line = join(",",unpack("C*",$1),0);
				    }
				    last;
				  };
481 482 483 484 485
		/\.rva|\.long|\.quad/
			    && do { $line =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
				    $line =~ s/\.L/$decor/g;
				    last;
				  };
486 487 488 489 490 491
	    }

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

		if ($dir =~ /\.extern/) {
A
Andy Polyakov 已提交
492
		    $self->{value} = ""; # swallow extern
493 494 495 496 497 498 499 500 501
		} 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"
				if ($win64 && $line =~ /([^,]+),\@function/);
		} elsif (!$elf && $dir =~ /\.size/) {
		    $self->{value} = "";
		    if (defined($current_function)) {
502
			$self->{value} .= "${decor}SEH_end_$current_function->{name}:"
503 504 505 506 507 508 509 510 511 512 513 514 515
				if ($win64 && $current_function->{abi} eq "svr4");
			undef $current_function;
		    }
		} elsif (!$elf && $dir =~ /\.align/) {
		    $self->{value} = ".p2align\t" . (log($line)/log(2));
		} elsif ($dir eq ".section") {
		    $current_segment=$line;
		    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";
516 517 518 519 520
		} elsif ($dir =~ /\.hidden/) {
		    if    ($flavour eq "macosx")  { $self->{value} = ".private_extern\t$prefix$line"; }
		    elsif ($flavour eq "mingw64") { $self->{value} = ""; }
		} elsif ($dir =~ /\.comm/) {
		    $self->{value} = "$dir\t$prefix$line";
521
		}
522 523 524 525
		$line = "";
		return $self;
	    }

526
	    # non-gas case or nasm/masm
527
	    SWITCH: for ($dir) {
528
		/\.text/    && do { my $v=undef;
529
				    if ($nasm) {
530
					$v="section	.text code align=64\n";
531 532
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
533
					$current_segment = ".text\$";
534 535 536 537
					$v.="$current_segment\tSEGMENT ";
					$v.=$masm>=$masmref ? "ALIGN(64)" : "PAGE";
					$v.=" 'CODE'";
				    }
538 539 540
				    $self->{value} = $v;
				    last;
				  };
541 542 543 544 545 546 547 548 549 550 551 552
		/\.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;
553 554
				    $line =~ s/([^,]*).*/$1/;
				    $line = ".CRT\$XCU" if ($line eq ".init");
555 556 557 558 559
				    if ($nasm) {
					$v="section	$line";
					if ($line=~/\.([px])data/) {
					    $v.=" rdata align=";
					    $v.=$1 eq "p"? 4 : 8;
560 561
					} elsif ($line=~/\.CRT\$/i) {
					    $v.=" rdata align=8";
562 563 564 565 566 567 568
					}
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
					$v.="$line\tSEGMENT";
					if ($line=~/\.([px])data/) {
					    $v.=" READONLY";
					    $v.=" ALIGN(".($1 eq "p" ? 4 : 8).")" if ($masm>=$masmref);
569 570
					} elsif ($line=~/\.CRT\$/i) {
					    $v.=" READONLY DWORD";
571 572 573 574 575 576
					}
				    }
				    $current_segment = $line;
				    $self->{value} = $v;
				    last;
				  };
577
		/\.extern/  && do { $self->{value}  = "EXTERN\t".$line;
578 579 580
				    $self->{value} .= ":NEAR" if ($masm);
				    last;
				  };
581 582 583
		/\.globl|.global/
			    && do { $self->{value}  = $masm?"PUBLIC":"global";
				    $self->{value} .= "\t".$line;
584 585
				    last;
				  };
586
		/\.size/    && do { if (defined($current_function)) {
A
Andy Polyakov 已提交
587 588
					undef $self->{value};
					if ($current_function->{abi} eq "svr4") {
589
					    $self->{value}="${decor}SEH_end_$current_function->{name}:";
A
Andy Polyakov 已提交
590 591 592
					    $self->{value}.=":\n" if($masm);
					}
					$self->{value}.="$current_function->{name}\tENDP" if($masm);
593 594 595 596 597
					undef $current_function;
				    }
				    last;
				  };
		/\.align/   && do { $self->{value} = "ALIGN\t".$line; last; };
598
		/\.(value|long|rva|quad)/
599
			    && do { my $sz  = substr($1,0,1);
600
				    my @arr = split(/,\s*/,$line);
601
				    my $last = pop(@arr);
602
				    my $conv = sub  {	my $var=shift;
603
							$var=~s/^(0b[0-1]+)/oct($1)/eig;
604
							$var=~s/^0x([0-9a-f]+)/0$1h/ig if ($masm);
605
							if ($sz eq "D" && ($current_segment=~/.[px]data/ || $dir eq ".rva"))
606
							{ $var=~s/([_a-z\$\@][_a-z0-9\$\@]*)/$nasm?"$1 wrt ..imagebase":"imagerel $1"/egi; }
607
							$var;
608
						    };  
609

610
				    $sz =~ tr/bvlrq/BWDDQ/;
611
				    $self->{value} = "\tD$sz\t";
612 613
				    for (@arr) { $self->{value} .= &$conv($_).","; }
				    $self->{value} .= &$conv($last);
614 615
				    last;
				  };
616
		/\.byte/    && do { my @str=split(/,\s*/,$line);
617
				    map(s/(0b[0-1]+)/oct($1)/eig,@str);
618
				    map(s/0x([0-9a-f]+)/0$1h/ig,@str) if ($masm);	
619
				    while ($#str>15) {
620
					$self->{value}.="DB\t"
621 622
						.join(",",@str[0..15])."\n";
					foreach (0..15) { shift @str; }
623
				    }
624 625
				    $self->{value}.="DB\t"
						.join(",",@str) if (@str);
626 627
				    last;
				  };
628 629 630
		/\.comm/    && do { my @str=split(/,\s*/,$line);
				    my $v=undef;
				    if ($nasm) {
631
					$v.="common	$prefix@str[0] @str[1]";
632 633
				    } else {
					$v="$current_segment\tENDS\n" if ($current_segment);
634
					$current_segment = "_DATA";
635 636 637 638 639 640
					$v.="$current_segment\tSEGMENT\n";
					$v.="COMM	@str[0]:DWORD:".@str[1]/4;
				    }
				    $self->{value} = $v;
				    last;
				  };
641 642 643 644 645 646 647 648 649 650 651 652
	    }
	    $line = "";
	}

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

653 654
sub rex {
 local *opcode=shift;
655 656 657 658 659
 my ($dst,$src,$rex)=@_;

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

662
# older gas and ml64 don't handle SSE>2 instructions
663 664 665
my %regrm = (	"%eax"=>0, "%ecx"=>1, "%edx"=>2, "%ebx"=>3,
		"%esp"=>4, "%ebp"=>5, "%esi"=>6, "%edi"=>7	);

666 667 668
my $movq = sub {	# elderly gas can't handle inter-register movq
  my $arg = shift;
  my @opcode=(0x66);
669
    if ($arg =~ /%xmm([0-9]+),\s*%r(\w+)/) {
670 671 672 673 674 675
	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;
676
    } elsif ($arg =~ /%r(\w+),\s*%xmm([0-9]+)/) {
677 678 679 680 681 682 683 684 685 686 687
	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 {
	();
    }
};

688
my $pextrd = sub {
689
    if (shift =~ /\$([0-9]+),\s*%xmm([0-9]+),\s*(%\w+)/) {
690 691 692
      my @opcode=(0x66);
	$imm=$1;
	$src=$2;
693
	$dst=$3;
694 695 696 697 698 699
	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;
700
	@opcode;
701
    } else {
702
	();
703
    }
704
};
705 706

my $pinsrd = sub {
707
    if (shift =~ /\$([0-9]+),\s*(%\w+),\s*%xmm([0-9]+)/) {
708 709
      my @opcode=(0x66);
	$imm=$1;
710 711
	$src=$2;
	$dst=$3;
712
	if ($src =~ /%r([0-9]+)/)	{ $src = $1; }
713 714 715 716 717
	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;
718
	@opcode;
719
    } else {
720
	();
721
    }
722
};
723 724

my $pshufb = sub {
725
    if (shift =~ /%xmm([0-9]+),\s*%xmm([0-9]+)/) {
726
      my @opcode=(0x66);
727
	rex(\@opcode,$2,$1);
728
	push @opcode,0x0f,0x38,0x00;
729 730
	push @opcode,0xc0|($1&7)|(($2&7)<<3);		# ModR/M
	@opcode;
731
    } else {
732
	();
733
    }
734
};
735

736
my $palignr = sub {
737
    if (shift =~ /\$([0-9]+),\s*%xmm([0-9]+),\s*%xmm([0-9]+)/) {
738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762
      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 {
	();
    }
};

763 764 765 766 767 768 769 770 771 772 773 774 775
my $rdrand = sub {
    if (shift =~ /%[er](\w+)/) {
      my @opcode=();
      my $dst=$1;
	if ($dst !~ /[0-9]+/) { $dst = $regrm{"%e$dst"}; }
	rex(\@opcode,0,$1,8);
	push @opcode,0x0f,0xc7,0xf0|($dst&7);
	@opcode;
    } else {
	();
    }
};

776 777 778
if ($nasm) {
    print <<___;
default	rel
779
%define XMMWORD
780 781 782 783 784 785
___
} elsif ($masm) {
    print <<___;
OPTION	DOTNAME
___
}
786 787 788 789
while($line=<>) {

    chomp($line);

790 791 792
    $line =~ s|[#!].*$||;	# get rid of asm-style comments...
    $line =~ s|/\*.*\*/||;	# ... and C-style comments...
    $line =~ s|^\s+||;		# ... and skip white spaces in beginning
793 794 795

    undef $label;
    undef $opcode;
796
    undef @args;
797 798 799 800 801

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

    if (directive->re(\$line)) {
	printf "%s",directive->out();
802 803 804 805 806 807 808 809 810 811
    } elsif ($opcode=opcode->re(\$line)) {
	my $asm = eval("\$".$opcode->mnemonic());
	undef @bytes;
	
	if ((ref($asm) eq 'CODE') && scalar(@bytes=&$asm($line))) {
	    print $gas?".byte\t":"DB\t",join(',',@bytes),"\n";
	    next;
	}

	ARGUMENT: while (1) {
812
	my $arg;
813

814 815 816 817 818
	if ($arg=register->re(\$line))	{ opcode->size($arg->size()); }
	elsif ($arg=const->re(\$line))	{ }
	elsif ($arg=ea->re(\$line))	{ }
	elsif ($arg=expr->re(\$line))	{ }
	else				{ last ARGUMENT; }
819

820
	push @args,$arg;
821

822
	last ARGUMENT if ($line !~ /^,/);
823

824
	$line =~ s/^,\s*//;
825 826
	} # ARGUMENT:

827 828
	if ($#args>=0) {
	    my $insn;
829 830
	    my $sz=opcode->size();

831
	    if ($gas) {
832
		$insn = $opcode->out($#args>=1?$args[$#args]->size():$sz);
833
		@args = map($_->out($sz),@args);
834
		printf "\t%s\t%s",$insn,join(",",@args);
835
	    } else {
836
		$insn = $opcode->out();
837 838
		foreach (@args) {
		    my $arg = $_->out();
839 840 841
		    # $insn.=$sz compensates for movq, pinsrw, ...
		    if ($arg =~ /^xmm[0-9]+$/) { $insn.=$sz; $sz="x" if(!$sz); last; }
		    if ($arg =~ /^mm[0-9]+$/)  { $insn.=$sz; $sz="q" if(!$sz); last; }
842
		}
843
		@args = reverse(@args);
844
		undef $sz if ($nasm && $opcode->mnemonic() eq "lea");
845
		printf "\t%s\t%s",$insn,join(",",map($_->out($sz),@args));
846 847 848 849 850 851 852 853 854
	    }
	} else {
	    printf "\t%s",$opcode->out();
	}
    }

    print $line,"\n";
}

855 856
print "\n$current_segment\tENDS\n"	if ($current_segment && $masm);
print "END\n"				if ($masm);
857 858 859

close STDOUT;

860
#################################################
861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896
# 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		-		-
# 
# (*)	volatile register
# (-)	preserved by callee
# (#)	Nth argument, volatile
#
# In Unix terms top of stack is argument transfer area for arguments
# which could not be accomodated in registers. Or in other words 7th
# [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
897 898
# 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 已提交
899 900
# Point is that at apparently no occasion Windows kernel would alter
# the area above user stack pointer in true asynchronous manner...
901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923
#
# All the above means that if assembler programmer adheres to Unix
# register and stack layout, but disregards the "red zone" existense,
# 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
924
#
925 926 927
#################################################
# Win64 SEH, Structured Exception Handling.
#
928 929 930 931 932 933 934 935 936 937 938 939 940 941
# 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:
#
942
# .type	function,@function
943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961
# 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
#	ret
962
# .size function,.-function
963 964 965 966 967 968 969 970 971
#
# 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)
972 973
# {	ULONG64 *rsp = (ULONG64 *)context->Rax;
#	if (context->Rip >= magic_point)
974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011
#	{   rsp = ((ULONG64 **)context->Rsp)[0];
#	    context->Rbp = rsp[-3];
#	    context->Rbx = rsp[-2];
#	    context->R15 = rsp[-1];
#	}
#	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
1012
#	CONTEXT.Xmm6				512
1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024
#	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
#
1025 1026 1027
# In order to tie the handler to the function one has to compose
# couple of structures: one for .xdata segment and one for .pdata.
#
1028
# UNWIND_INFO structure for .xdata segment would be
1029 1030 1031
#
# function_unwind_info:
#	.byte	9,0,0,0
1032
#	.rva	handler
1033 1034 1035 1036 1037 1038
#
# 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
1039
# .LSEH_begin_{function_name} label. This essentially defines the SEH
1040 1041 1042
# 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.
1043
# End of function is marked with respective .LSEH_end_{function_name}
1044 1045
# label. To summarize, .pdata segment would contain
#
1046 1047 1048
#	.rva	.LSEH_begin_function
#	.rva	.LSEH_end_function
#	.rva	function_unwind_info
1049 1050
#
# Reference to functon_unwind_info from .xdata segment is the anchor.
1051
# In case you wonder why references are 32-bit .rvas and not 64-bit
1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075
# .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...
#
# Attentive reader can notice that exceptions would be mishandled in
# auto-generated "gear" epilogue. Well, exception effectively can't
# occur there, because if memory area used by it was subject to
# segmentation violation, then it would be raised upon call to the
# function (and as already mentioned be accounted to caller, which is
# not a problem). If you're still not comfortable, then define tail
# "magic point" just prior ret instruction and have handler treat it...
1076 1077 1078 1079 1080 1081
#
# (*)	Note that we're talking about run-time, not debug-time. Lack of
#	unwind information makes debugging hard on both Windows and
#	Unix. "Unlike" referes to the fact that on Unix signal handler
#	will always be invoked, core dumped and appropriate exit code
#	returned to parent (for user notification).