x86_64-xlate.pl 33.3 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
	    $sz="l" if (opcode->mnemonic() eq "movd");
280

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

303 304
	# optional * ---vvv--- appears in indirect jmp/call
	if ($line =~ /^(\*?)%(\w+)/) {
305
	    bless $self,$class;
306 307
	    $self->{asterisk} = $1;
	    $self->{value} = $2;
308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329
	    $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;
330
	if ($gas)	{ sprintf "%s%%%s",$self->{asterisk},$self->{value}; }
331
	else		{ $self->{value}; }
332 333 334 335 336 337 338 339
    }
}
{ package label;	# pick up labels, which end with :
    sub re {
	my	$self = shift;	# single instance is enough...
	local	*line = shift;
	undef	$ret;

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

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

352 353 354 355 356 357 358 359 360
	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";
361
		$func .= "${decor}SEH_begin_$current_function->{name}:\n";
362 363 364 365 366 367 368 369 370 371 372
		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}") {
373
	    $self->{value} .= ":" if ($masm && $ret!~m/^\$/);
374 375
	    $self->{value} . ":";
	} elsif ($win64 && $current_function->{abi} eq "svr4") {
376 377 378 379 380
	    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";
381
	    $func .= "	mov	rax,rsp\n";
382
	    $func .= "${decor}SEH_begin_$current_function->{name}:";
383 384
	    $func .= ":" if ($masm);
	    $func .= "\n";
385 386 387 388 389 390
	    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);
391 392
	    $func .= "	mov	r8,QWORD${PTR}[40+rsp]\n" if ($narg>4);
	    $func .= "	mov	r9,QWORD${PTR}[48+rsp]\n" if ($narg>5);
393 394
	    $func .= "\n";
	} else {
395 396
	   "$current_function->{name}".
			($nasm ? ":" : "\tPROC $current_function->{scope}");
397 398 399 400 401 402 403 404 405 406 407 408 409 410
	}
    }
}
{ 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+//;

411 412
	    $self->{value} =~ s/\@PLT// if (!$elf);
	    $self->{value} =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
413
	    $self->{value} =~ s/\.L/$decor/g;
414 415 416 417 418
	}
	$ret;
    }
    sub out {
	my $self = shift;
419 420 421 422 423
	if ($nasm && opcode->mnemonic()=~m/^j/) {
	    "NEAR ".$self->{value};
	} else {
	    $self->{value};
	}
424 425 426 427 428 429 430 431
    }
}
{ package directive;	# pick up directives, which start with .
    sub re {
	my	$self = shift;	# single instance is enough...
	local	*line = shift;
	undef	$ret;
	my	$dir;
432 433 434 435 436 437 438 439 440
	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	);
441 442

	if ($line =~ /^\s*(\.\w+)/) {
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 481
	    $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;
				  };
482 483 484 485 486
		/\.rva|\.long|\.quad/
			    && do { $line =~ s/([_a-z][_a-z0-9]*)/$globals{$1} or $1/gei;
				    $line =~ s/\.L/$decor/g;
				    last;
				  };
487 488 489 490 491 492
	    }

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

		if ($dir =~ /\.extern/) {
A
Andy Polyakov 已提交
493
		    $self->{value} = ""; # swallow extern
494 495 496 497 498 499 500 501 502
		} 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)) {
503
			$self->{value} .= "${decor}SEH_end_$current_function->{name}:"
504 505 506 507 508 509 510 511 512 513 514 515 516
				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";
517 518 519 520 521
		} 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";
522
		    $self->{value} =~ s|,([0-9]+),([0-9]+)$|",$1,".log($2)/log(2)|e if ($flavour eq "macosx");
523
		}
524 525 526 527
		$line = "";
		return $self;
	    }

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

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

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

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

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

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

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

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

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

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

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

765 766 767 768 769 770 771 772 773 774 775 776 777
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 {
	();
    }
};

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

    chomp($line);

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

    undef $label;
    undef $opcode;
798
    undef @args;
799 800 801 802 803

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

    if (directive->re(\$line)) {
	printf "%s",directive->out();
804 805 806 807 808 809 810 811 812 813
    } 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) {
814
	my $arg;
815

816 817 818 819 820
	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; }
821

822
	push @args,$arg;
823

824
	last ARGUMENT if ($line !~ /^,/);
825

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

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

833
	    if ($gas) {
834
		$insn = $opcode->out($#args>=1?$args[$#args]->size():$sz);
835
		@args = map($_->out($sz),@args);
836
		printf "\t%s\t%s",$insn,join(",",@args);
837
	    } else {
838
		$insn = $opcode->out();
839 840
		foreach (@args) {
		    my $arg = $_->out();
841 842 843
		    # $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; }
844
		}
845
		@args = reverse(@args);
846
		undef $sz if ($nasm && $opcode->mnemonic() eq "lea");
847
		printf "\t%s\t%s",$insn,join(",",map($_->out($sz),@args));
848 849 850 851 852 853 854 855 856
	    }
	} else {
	    printf "\t%s",$opcode->out();
	}
    }

    print $line,"\n";
}

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

close STDOUT;

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 897 898
# 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
899 900
# 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 已提交
901 902
# Point is that at apparently no occasion Windows kernel would alter
# the area above user stack pointer in true asynchronous manner...
903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
#
# 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
926
#
927 928 929
#################################################
# Win64 SEH, Structured Exception Handling.
#
930 931 932 933 934 935 936 937 938 939 940 941 942 943
# 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:
#
944
# .type	function,@function
945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963
# 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
964
# .size function,.-function
965 966 967 968 969 970 971 972 973
#
# 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)
974 975
# {	ULONG64 *rsp = (ULONG64 *)context->Rax;
#	if (context->Rip >= magic_point)
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 1012 1013
#	{   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
1014
#	CONTEXT.Xmm6				512
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026
#	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
#
1027 1028 1029
# In order to tie the handler to the function one has to compose
# couple of structures: one for .xdata segment and one for .pdata.
#
1030
# UNWIND_INFO structure for .xdata segment would be
1031 1032 1033
#
# function_unwind_info:
#	.byte	9,0,0,0
1034
#	.rva	handler
1035 1036 1037 1038 1039 1040
#
# 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
1041
# .LSEH_begin_{function_name} label. This essentially defines the SEH
1042 1043 1044
# 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.
1045
# End of function is marked with respective .LSEH_end_{function_name}
1046 1047
# label. To summarize, .pdata segment would contain
#
1048 1049 1050
#	.rva	.LSEH_begin_function
#	.rva	.LSEH_end_function
#	.rva	function_unwind_info
1051 1052
#
# Reference to functon_unwind_info from .xdata segment is the anchor.
1053
# In case you wonder why references are 32-bit .rvas and not 64-bit
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077
# .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...
1078 1079 1080 1081 1082 1083
#
# (*)	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).