i8254.c 18.9 KB
Newer Older
S
Sheng Yang 已提交
1 2 3 4 5 6 7
/*
 * 8253/8254 interval timer emulation
 *
 * Copyright (c) 2003-2004 Fabrice Bellard
 * Copyright (c) 2006 Intel Corporation
 * Copyright (c) 2007 Keir Fraser, XenSource Inc
 * Copyright (c) 2008 Intel Corporation
N
Nicolas Kaiser 已提交
8
 * Copyright 2009 Red Hat, Inc. and/or its affiliates.
S
Sheng Yang 已提交
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 * Authors:
 *   Sheng Yang <sheng.yang@intel.com>
 *   Based on QEMU and Xen.
 */

J
Joe Perches 已提交
33 34
#define pr_fmt(fmt) "pit: " fmt

S
Sheng Yang 已提交
35
#include <linux/kvm_host.h>
36
#include <linux/slab.h>
S
Sheng Yang 已提交
37 38 39 40 41

#include "irq.h"
#include "i8254.h"

#ifndef CONFIG_X86_64
R
Roman Zippel 已提交
42
#define mod_64(x, y) ((x) - (y) * div64_u64(x, y))
S
Sheng Yang 已提交
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66
#else
#define mod_64(x, y) ((x) % (y))
#endif

#define RW_STATE_LSB 1
#define RW_STATE_MSB 2
#define RW_STATE_WORD0 3
#define RW_STATE_WORD1 4

/* Compute with 96 bit intermediate result: (a*b)/c */
static u64 muldiv64(u64 a, u32 b, u32 c)
{
	union {
		u64 ll;
		struct {
			u32 low, high;
		} l;
	} u, res;
	u64 rl, rh;

	u.ll = a;
	rl = (u64)u.l.low * (u64)b;
	rh = (u64)u.l.high * (u64)b;
	rh += (rl >> 32);
R
Roman Zippel 已提交
67 68
	res.l.high = div64_u64(rh, c);
	res.l.low = div64_u64(((mod_64(rh, c) << 32) + (rl & 0xffffffff)), c);
S
Sheng Yang 已提交
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
	return res.ll;
}

static void pit_set_gate(struct kvm *kvm, int channel, u32 val)
{
	struct kvm_kpit_channel_state *c =
		&kvm->arch.vpit->pit_state.channels[channel];

	WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));

	switch (c->mode) {
	default:
	case 0:
	case 4:
		/* XXX: just disable/enable counting */
		break;
	case 1:
	case 2:
	case 3:
	case 5:
		/* Restart counting on rising edge. */
		if (c->gate < val)
			c->count_load_time = ktime_get();
		break;
	}

	c->gate = val;
}

98
static int pit_get_gate(struct kvm *kvm, int channel)
S
Sheng Yang 已提交
99 100 101 102 103 104
{
	WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));

	return kvm->arch.vpit->pit_state.channels[channel].gate;
}

105 106 107 108 109 110
static s64 __kpit_elapsed(struct kvm *kvm)
{
	s64 elapsed;
	ktime_t remaining;
	struct kvm_kpit_state *ps = &kvm->arch.vpit->pit_state;

111 112 113
	if (!ps->pit_timer.period)
		return 0;

114 115 116 117 118 119 120 121 122
	/*
	 * The Counter does not stop when it reaches zero. In
	 * Modes 0, 1, 4, and 5 the Counter ``wraps around'' to
	 * the highest count, either FFFF hex for binary counting
	 * or 9999 for BCD counting, and continues counting.
	 * Modes 2 and 3 are periodic; the Counter reloads
	 * itself with the initial count and continues counting
	 * from there.
	 */
123
	remaining = hrtimer_get_remaining(&ps->pit_timer.timer);
124 125
	elapsed = ps->pit_timer.period - ktime_to_ns(remaining);
	elapsed = mod_64(elapsed, ps->pit_timer.period);
126 127 128 129 130 131 132 133 134 135 136 137 138

	return elapsed;
}

static s64 kpit_elapsed(struct kvm *kvm, struct kvm_kpit_channel_state *c,
			int channel)
{
	if (channel == 0)
		return __kpit_elapsed(kvm);

	return ktime_to_ns(ktime_sub(ktime_get(), c->count_load_time));
}

S
Sheng Yang 已提交
139 140 141 142 143 144 145 146 147
static int pit_get_count(struct kvm *kvm, int channel)
{
	struct kvm_kpit_channel_state *c =
		&kvm->arch.vpit->pit_state.channels[channel];
	s64 d, t;
	int counter;

	WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));

148
	t = kpit_elapsed(kvm, c, channel);
S
Sheng Yang 已提交
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177
	d = muldiv64(t, KVM_PIT_FREQ, NSEC_PER_SEC);

	switch (c->mode) {
	case 0:
	case 1:
	case 4:
	case 5:
		counter = (c->count - d) & 0xffff;
		break;
	case 3:
		/* XXX: may be incorrect for odd counts */
		counter = c->count - (mod_64((2 * d), c->count));
		break;
	default:
		counter = c->count - mod_64(d, c->count);
		break;
	}
	return counter;
}

static int pit_get_out(struct kvm *kvm, int channel)
{
	struct kvm_kpit_channel_state *c =
		&kvm->arch.vpit->pit_state.channels[channel];
	s64 d, t;
	int out;

	WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));

178
	t = kpit_elapsed(kvm, c, channel);
S
Sheng Yang 已提交
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
	d = muldiv64(t, KVM_PIT_FREQ, NSEC_PER_SEC);

	switch (c->mode) {
	default:
	case 0:
		out = (d >= c->count);
		break;
	case 1:
		out = (d < c->count);
		break;
	case 2:
		out = ((mod_64(d, c->count) == 0) && (d != 0));
		break;
	case 3:
		out = (mod_64(d, c->count) < ((c->count + 1) >> 1));
		break;
	case 4:
	case 5:
		out = (d == c->count);
		break;
	}

	return out;
}

static void pit_latch_count(struct kvm *kvm, int channel)
{
	struct kvm_kpit_channel_state *c =
		&kvm->arch.vpit->pit_state.channels[channel];

	WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));

	if (!c->count_latched) {
		c->latched_count = pit_get_count(kvm, channel);
		c->count_latched = c->rw_mode;
	}
}

static void pit_latch_status(struct kvm *kvm, int channel)
{
	struct kvm_kpit_channel_state *c =
		&kvm->arch.vpit->pit_state.channels[channel];

	WARN_ON(!mutex_is_locked(&kvm->arch.vpit->pit_state.lock));

	if (!c->status_latched) {
		/* TODO: Return NULL COUNT (bit 6). */
		c->status = ((pit_get_out(kvm, channel) << 7) |
				(c->rw_mode << 4) |
				(c->mode << 1) |
				c->bcd);
		c->status_latched = 1;
	}
}

234
static void kvm_pit_ack_irq(struct kvm_irq_ack_notifier *kian)
235 236 237
{
	struct kvm_kpit_state *ps = container_of(kian, struct kvm_kpit_state,
						 irq_ack_notifier);
238 239 240 241 242 243 244 245
	int value;

	spin_lock(&ps->inject_lock);
	value = atomic_dec_return(&ps->pit_timer.pending);
	if (value < 0)
		/* spurious acks can be generated if, for example, the
		 * PIC is being reset.  Handle it gracefully here
		 */
246
		atomic_inc(&ps->pit_timer.pending);
247 248 249 250
	else if (value > 0)
		/* in this case, we had multiple outstanding pit interrupts
		 * that we needed to inject.  Reinject
		 */
251
		queue_kthread_work(&ps->pit->worker, &ps->pit->expired);
252
	ps->irq_ack = 1;
253
	spin_unlock(&ps->inject_lock);
254 255
}

M
Marcelo Tosatti 已提交
256 257 258 259 260
void __kvm_migrate_pit_timer(struct kvm_vcpu *vcpu)
{
	struct kvm_pit *pit = vcpu->kvm->arch.vpit;
	struct hrtimer *timer;

261
	if (!kvm_vcpu_is_bsp(vcpu) || !pit)
M
Marcelo Tosatti 已提交
262 263 264 265
		return;

	timer = &pit->pit_state.pit_timer.timer;
	if (hrtimer_cancel(timer))
266
		hrtimer_start_expires(timer, HRTIMER_MODE_ABS);
M
Marcelo Tosatti 已提交
267 268
}

269
static void destroy_pit_timer(struct kvm_pit *pit)
S
Sheng Yang 已提交
270
{
271
	hrtimer_cancel(&pit->pit_state.pit_timer.timer);
272
	flush_kthread_work(&pit->expired);
S
Sheng Yang 已提交
273 274
}

275
static void pit_do_work(struct kthread_work *work)
276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313
{
	struct kvm_pit *pit = container_of(work, struct kvm_pit, expired);
	struct kvm *kvm = pit->kvm;
	struct kvm_vcpu *vcpu;
	int i;
	struct kvm_kpit_state *ps = &pit->pit_state;
	int inject = 0;

	/* Try to inject pending interrupts when
	 * last one has been acked.
	 */
	spin_lock(&ps->inject_lock);
	if (ps->irq_ack) {
		ps->irq_ack = 0;
		inject = 1;
	}
	spin_unlock(&ps->inject_lock);
	if (inject) {
		kvm_set_irq(kvm, kvm->arch.vpit->irq_source_id, 0, 1);
		kvm_set_irq(kvm, kvm->arch.vpit->irq_source_id, 0, 0);

		/*
		 * Provides NMI watchdog support via Virtual Wire mode.
		 * The route is: PIT -> PIC -> LVT0 in NMI mode.
		 *
		 * Note: Our Virtual Wire implementation is simplified, only
		 * propagating PIT interrupts to all VCPUs when they have set
		 * LVT0 to NMI delivery. Other PIC interrupts are just sent to
		 * VCPU0, and only if its LVT0 is in EXTINT mode.
		 */
		if (kvm->arch.vapics_in_nmi_mode > 0)
			kvm_for_each_vcpu(i, vcpu, kvm)
				kvm_apic_nmi_wd_deliver(vcpu);
	}
}

static enum hrtimer_restart pit_timer_fn(struct hrtimer *data)
{
314
	struct kvm_pit_timer *ktimer = container_of(data, struct kvm_pit_timer, timer);
315 316 317 318
	struct kvm_pit *pt = ktimer->kvm->arch.vpit;

	if (ktimer->reinject || !atomic_read(&ktimer->pending)) {
		atomic_inc(&ktimer->pending);
319
		queue_kthread_work(&pt->worker, &pt->expired);
320 321
	}

A
Avi Kivity 已提交
322
	if (pt->pit_state.is_periodic) {
323 324 325 326 327 328
		hrtimer_add_expires_ns(&ktimer->timer, ktimer->period);
		return HRTIMER_RESTART;
	} else
		return HRTIMER_NORESTART;
}

329
static void create_pit_timer(struct kvm *kvm, u32 val, int is_period)
S
Sheng Yang 已提交
330
{
331
	struct kvm_kpit_state *ps = &kvm->arch.vpit->pit_state;
332
	struct kvm_pit_timer *pt = &ps->pit_timer;
S
Sheng Yang 已提交
333 334
	s64 interval;

J
Jan Kiszka 已提交
335
	if (!irqchip_in_kernel(kvm) || ps->flags & KVM_PIT_FLAGS_HPET_LEGACY)
336 337
		return;

S
Sheng Yang 已提交
338 339
	interval = muldiv64(val, NSEC_PER_SEC, KVM_PIT_FREQ);

J
Joe Perches 已提交
340
	pr_debug("create pit timer, interval is %llu nsec\n", interval);
S
Sheng Yang 已提交
341 342 343

	/* TODO The new value only affected after the retriggered */
	hrtimer_cancel(&pt->timer);
344
	flush_kthread_work(&ps->pit->expired);
345
	pt->period = interval;
346 347
	ps->is_periodic = is_period;

348
	pt->timer.function = pit_timer_fn;
349 350
	pt->kvm = ps->pit->kvm;

S
Sheng Yang 已提交
351
	atomic_set(&pt->pending, 0);
352
	ps->irq_ack = 1;
S
Sheng Yang 已提交
353 354 355 356 357 358 359 360 361 362 363

	hrtimer_start(&pt->timer, ktime_add_ns(ktime_get(), interval),
		      HRTIMER_MODE_ABS);
}

static void pit_load_count(struct kvm *kvm, int channel, u32 val)
{
	struct kvm_kpit_state *ps = &kvm->arch.vpit->pit_state;

	WARN_ON(!mutex_is_locked(&ps->lock));

J
Joe Perches 已提交
364
	pr_debug("load_count val is %d, channel is %d\n", val, channel);
S
Sheng Yang 已提交
365 366

	/*
367 368
	 * The largest possible initial count is 0; this is equivalent
	 * to 216 for binary counting and 104 for BCD counting.
S
Sheng Yang 已提交
369 370 371 372 373 374
	 */
	if (val == 0)
		val = 0x10000;

	ps->channels[channel].count = val;

375 376
	if (channel != 0) {
		ps->channels[channel].count_load_time = ktime_get();
S
Sheng Yang 已提交
377
		return;
378
	}
S
Sheng Yang 已提交
379 380 381 382

	/* Two types of timer
	 * mode 1 is one shot, mode 2 is period, otherwise del timer */
	switch (ps->channels[0].mode) {
383
	case 0:
S
Sheng Yang 已提交
384
	case 1:
M
Marcelo Tosatti 已提交
385 386
        /* FIXME: enhance mode 4 precision */
	case 4:
J
Jan Kiszka 已提交
387
		create_pit_timer(kvm, val, 0);
S
Sheng Yang 已提交
388 389
		break;
	case 2:
A
Aurelien Jarno 已提交
390
	case 3:
J
Jan Kiszka 已提交
391
		create_pit_timer(kvm, val, 1);
S
Sheng Yang 已提交
392 393
		break;
	default:
394
		destroy_pit_timer(kvm->arch.vpit);
S
Sheng Yang 已提交
395 396 397
	}
}

B
Beth Kon 已提交
398
void kvm_pit_load_count(struct kvm *kvm, int channel, u32 val, int hpet_legacy_start)
399
{
B
Beth Kon 已提交
400 401 402 403 404 405 406 407 408 409
	u8 saved_mode;
	if (hpet_legacy_start) {
		/* save existing mode for later reenablement */
		saved_mode = kvm->arch.vpit->pit_state.channels[0].mode;
		kvm->arch.vpit->pit_state.channels[0].mode = 0xff; /* disable timer */
		pit_load_count(kvm, channel, val);
		kvm->arch.vpit->pit_state.channels[0].mode = saved_mode;
	} else {
		pit_load_count(kvm, channel, val);
	}
410 411
}

G
Gregory Haskins 已提交
412 413 414 415 416 417 418 419 420 421
static inline struct kvm_pit *dev_to_pit(struct kvm_io_device *dev)
{
	return container_of(dev, struct kvm_pit, dev);
}

static inline struct kvm_pit *speaker_to_pit(struct kvm_io_device *dev)
{
	return container_of(dev, struct kvm_pit, speaker_dev);
}

422 423 424 425 426 427 428 429
static inline int pit_in_range(gpa_t addr)
{
	return ((addr >= KVM_PIT_BASE_ADDRESS) &&
		(addr < KVM_PIT_BASE_ADDRESS + KVM_PIT_MEM_LENGTH));
}

static int pit_ioport_write(struct kvm_io_device *this,
			    gpa_t addr, int len, const void *data)
S
Sheng Yang 已提交
430
{
G
Gregory Haskins 已提交
431
	struct kvm_pit *pit = dev_to_pit(this);
S
Sheng Yang 已提交
432 433 434 435 436
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	int channel, access;
	struct kvm_kpit_channel_state *s;
	u32 val = *(u32 *) data;
437 438
	if (!pit_in_range(addr))
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
439 440 441 442 443 444 445

	val  &= 0xff;
	addr &= KVM_PIT_CHANNEL_MASK;

	mutex_lock(&pit_state->lock);

	if (val != 0)
J
Joe Perches 已提交
446 447
		pr_debug("write addr is 0x%x, len is %d, val is 0x%x\n",
			 (unsigned int)addr, len, val);
S
Sheng Yang 已提交
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 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500

	if (addr == 3) {
		channel = val >> 6;
		if (channel == 3) {
			/* Read-Back Command. */
			for (channel = 0; channel < 3; channel++) {
				s = &pit_state->channels[channel];
				if (val & (2 << channel)) {
					if (!(val & 0x20))
						pit_latch_count(kvm, channel);
					if (!(val & 0x10))
						pit_latch_status(kvm, channel);
				}
			}
		} else {
			/* Select Counter <channel>. */
			s = &pit_state->channels[channel];
			access = (val >> 4) & KVM_PIT_CHANNEL_MASK;
			if (access == 0) {
				pit_latch_count(kvm, channel);
			} else {
				s->rw_mode = access;
				s->read_state = access;
				s->write_state = access;
				s->mode = (val >> 1) & 7;
				if (s->mode > 5)
					s->mode -= 4;
				s->bcd = val & 1;
			}
		}
	} else {
		/* Write Count. */
		s = &pit_state->channels[addr];
		switch (s->write_state) {
		default:
		case RW_STATE_LSB:
			pit_load_count(kvm, addr, val);
			break;
		case RW_STATE_MSB:
			pit_load_count(kvm, addr, val << 8);
			break;
		case RW_STATE_WORD0:
			s->write_latch = val;
			s->write_state = RW_STATE_WORD1;
			break;
		case RW_STATE_WORD1:
			pit_load_count(kvm, addr, s->write_latch | (val << 8));
			s->write_state = RW_STATE_WORD0;
			break;
		}
	}

	mutex_unlock(&pit_state->lock);
501
	return 0;
S
Sheng Yang 已提交
502 503
}

504 505
static int pit_ioport_read(struct kvm_io_device *this,
			   gpa_t addr, int len, void *data)
S
Sheng Yang 已提交
506
{
G
Gregory Haskins 已提交
507
	struct kvm_pit *pit = dev_to_pit(this);
S
Sheng Yang 已提交
508 509 510 511
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	int ret, count;
	struct kvm_kpit_channel_state *s;
512 513
	if (!pit_in_range(addr))
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
514 515

	addr &= KVM_PIT_CHANNEL_MASK;
516 517 518
	if (addr == 3)
		return 0;

S
Sheng Yang 已提交
519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
	s = &pit_state->channels[addr];

	mutex_lock(&pit_state->lock);

	if (s->status_latched) {
		s->status_latched = 0;
		ret = s->status;
	} else if (s->count_latched) {
		switch (s->count_latched) {
		default:
		case RW_STATE_LSB:
			ret = s->latched_count & 0xff;
			s->count_latched = 0;
			break;
		case RW_STATE_MSB:
			ret = s->latched_count >> 8;
			s->count_latched = 0;
			break;
		case RW_STATE_WORD0:
			ret = s->latched_count & 0xff;
			s->count_latched = RW_STATE_MSB;
			break;
		}
	} else {
		switch (s->read_state) {
		default:
		case RW_STATE_LSB:
			count = pit_get_count(kvm, addr);
			ret = count & 0xff;
			break;
		case RW_STATE_MSB:
			count = pit_get_count(kvm, addr);
			ret = (count >> 8) & 0xff;
			break;
		case RW_STATE_WORD0:
			count = pit_get_count(kvm, addr);
			ret = count & 0xff;
			s->read_state = RW_STATE_WORD1;
			break;
		case RW_STATE_WORD1:
			count = pit_get_count(kvm, addr);
			ret = (count >> 8) & 0xff;
			s->read_state = RW_STATE_WORD0;
			break;
		}
	}

	if (len > sizeof(ret))
		len = sizeof(ret);
	memcpy(data, (char *)&ret, len);

	mutex_unlock(&pit_state->lock);
571
	return 0;
S
Sheng Yang 已提交
572 573
}

574 575
static int speaker_ioport_write(struct kvm_io_device *this,
				gpa_t addr, int len, const void *data)
S
Sheng Yang 已提交
576
{
G
Gregory Haskins 已提交
577
	struct kvm_pit *pit = speaker_to_pit(this);
S
Sheng Yang 已提交
578 579 580
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	u32 val = *(u32 *) data;
581 582
	if (addr != KVM_SPEAKER_BASE_ADDRESS)
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
583 584 585 586 587

	mutex_lock(&pit_state->lock);
	pit_state->speaker_data_on = (val >> 1) & 1;
	pit_set_gate(kvm, 2, val & 1);
	mutex_unlock(&pit_state->lock);
588
	return 0;
S
Sheng Yang 已提交
589 590
}

591 592
static int speaker_ioport_read(struct kvm_io_device *this,
			       gpa_t addr, int len, void *data)
S
Sheng Yang 已提交
593
{
G
Gregory Haskins 已提交
594
	struct kvm_pit *pit = speaker_to_pit(this);
S
Sheng Yang 已提交
595 596 597 598
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	unsigned int refresh_clock;
	int ret;
599 600
	if (addr != KVM_SPEAKER_BASE_ADDRESS)
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
601 602 603 604 605 606 607 608 609 610 611

	/* Refresh clock toggles at about 15us. We approximate as 2^14ns. */
	refresh_clock = ((unsigned int)ktime_to_ns(ktime_get()) >> 14) & 1;

	mutex_lock(&pit_state->lock);
	ret = ((pit_state->speaker_data_on << 1) | pit_get_gate(kvm, 2) |
		(pit_get_out(kvm, 2) << 5) | (refresh_clock << 4));
	if (len > sizeof(ret))
		len = sizeof(ret);
	memcpy(data, (char *)&ret, len);
	mutex_unlock(&pit_state->lock);
612
	return 0;
S
Sheng Yang 已提交
613 614
}

615
void kvm_pit_reset(struct kvm_pit *pit)
S
Sheng Yang 已提交
616 617
{
	int i;
618 619 620
	struct kvm_kpit_channel_state *c;

	mutex_lock(&pit->pit_state.lock);
B
Beth Kon 已提交
621
	pit->pit_state.flags = 0;
622 623 624 625 626 627 628 629 630
	for (i = 0; i < 3; i++) {
		c = &pit->pit_state.channels[i];
		c->mode = 0xff;
		c->gate = (i != 2);
		pit_load_count(pit->kvm, i, 0);
	}
	mutex_unlock(&pit->pit_state.lock);

	atomic_set(&pit->pit_state.pit_timer.pending, 0);
631
	pit->pit_state.irq_ack = 1;
632 633
}

634 635 636 637 638 639 640 641 642 643
static void pit_mask_notifer(struct kvm_irq_mask_notifier *kimn, bool mask)
{
	struct kvm_pit *pit = container_of(kimn, struct kvm_pit, mask_notifier);

	if (!mask) {
		atomic_set(&pit->pit_state.pit_timer.pending, 0);
		pit->pit_state.irq_ack = 1;
	}
}

G
Gregory Haskins 已提交
644 645 646 647 648 649 650 651 652 653
static const struct kvm_io_device_ops pit_dev_ops = {
	.read     = pit_ioport_read,
	.write    = pit_ioport_write,
};

static const struct kvm_io_device_ops speaker_dev_ops = {
	.read     = speaker_ioport_read,
	.write    = speaker_ioport_write,
};

654
/* Caller must hold slots_lock */
655
struct kvm_pit *kvm_create_pit(struct kvm *kvm, u32 flags)
656
{
S
Sheng Yang 已提交
657 658
	struct kvm_pit *pit;
	struct kvm_kpit_state *pit_state;
659 660
	struct pid *pid;
	pid_t pid_nr;
661
	int ret;
S
Sheng Yang 已提交
662 663 664 665 666

	pit = kzalloc(sizeof(struct kvm_pit), GFP_KERNEL);
	if (!pit)
		return NULL;

667
	pit->irq_source_id = kvm_request_irq_source_id(kvm);
668 669
	if (pit->irq_source_id < 0) {
		kfree(pit);
670
		return NULL;
671
	}
672

S
Sheng Yang 已提交
673 674
	mutex_init(&pit->pit_state.lock);
	mutex_lock(&pit->pit_state.lock);
675 676
	spin_lock_init(&pit->pit_state.inject_lock);

677 678 679 680 681 682 683 684
	pid = get_pid(task_tgid(current));
	pid_nr = pid_vnr(pid);
	put_pid(pid);

	init_kthread_worker(&pit->worker);
	pit->worker_task = kthread_run(kthread_worker_fn, &pit->worker,
				       "kvm-pit/%d", pid_nr);
	if (IS_ERR(pit->worker_task)) {
685
		mutex_unlock(&pit->pit_state.lock);
686
		kvm_free_irq_source_id(kvm, pit->irq_source_id);
687 688 689
		kfree(pit);
		return NULL;
	}
690
	init_kthread_work(&pit->expired, pit_do_work);
S
Sheng Yang 已提交
691 692 693 694 695 696 697 698

	kvm->arch.vpit = pit;
	pit->kvm = kvm;

	pit_state = &pit->pit_state;
	pit_state->pit = pit;
	hrtimer_init(&pit_state->pit_timer.timer,
		     CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
699 700 701
	pit_state->irq_ack_notifier.gsi = 0;
	pit_state->irq_ack_notifier.irq_acked = kvm_pit_ack_irq;
	kvm_register_irq_ack_notifier(kvm, &pit_state->irq_ack_notifier);
702
	pit_state->pit_timer.reinject = true;
S
Sheng Yang 已提交
703 704
	mutex_unlock(&pit->pit_state.lock);

705
	kvm_pit_reset(pit);
S
Sheng Yang 已提交
706

707 708 709
	pit->mask_notifier.func = pit_mask_notifer;
	kvm_register_irq_mask_notifier(kvm, 0, &pit->mask_notifier);

710
	kvm_iodevice_init(&pit->dev, &pit_dev_ops);
711 712
	ret = kvm_io_bus_register_dev(kvm, KVM_PIO_BUS, KVM_PIT_BASE_ADDRESS,
				      KVM_PIT_MEM_LENGTH, &pit->dev);
713 714
	if (ret < 0)
		goto fail;
715 716 717

	if (flags & KVM_PIT_SPEAKER_DUMMY) {
		kvm_iodevice_init(&pit->speaker_dev, &speaker_dev_ops);
M
Marcelo Tosatti 已提交
718
		ret = kvm_io_bus_register_dev(kvm, KVM_PIO_BUS,
719 720
					      KVM_SPEAKER_BASE_ADDRESS, 4,
					      &pit->speaker_dev);
721 722
		if (ret < 0)
			goto fail_unregister;
723 724
	}

S
Sheng Yang 已提交
725
	return pit;
726 727

fail_unregister:
M
Marcelo Tosatti 已提交
728
	kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS, &pit->dev);
729 730

fail:
731 732 733
	kvm_unregister_irq_mask_notifier(kvm, 0, &pit->mask_notifier);
	kvm_unregister_irq_ack_notifier(kvm, &pit_state->irq_ack_notifier);
	kvm_free_irq_source_id(kvm, pit->irq_source_id);
734
	kthread_stop(pit->worker_task);
735 736
	kfree(pit);
	return NULL;
S
Sheng Yang 已提交
737 738 739 740 741 742 743
}

void kvm_free_pit(struct kvm *kvm)
{
	struct hrtimer *timer;

	if (kvm->arch.vpit) {
744 745 746
		kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS, &kvm->arch.vpit->dev);
		kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS,
					      &kvm->arch.vpit->speaker_dev);
747 748
		kvm_unregister_irq_mask_notifier(kvm, 0,
					       &kvm->arch.vpit->mask_notifier);
749 750
		kvm_unregister_irq_ack_notifier(kvm,
				&kvm->arch.vpit->pit_state.irq_ack_notifier);
S
Sheng Yang 已提交
751 752 753
		mutex_lock(&kvm->arch.vpit->pit_state.lock);
		timer = &kvm->arch.vpit->pit_state.pit_timer.timer;
		hrtimer_cancel(timer);
754 755
		flush_kthread_work(&kvm->arch.vpit->expired);
		kthread_stop(kvm->arch.vpit->worker_task);
756
		kvm_free_irq_source_id(kvm, kvm->arch.vpit->irq_source_id);
S
Sheng Yang 已提交
757 758 759 760
		mutex_unlock(&kvm->arch.vpit->pit_state.lock);
		kfree(kvm->arch.vpit);
	}
}