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>
37
#include <linux/workqueue.h>
S
Sheng Yang 已提交
38 39 40 41 42

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

#ifndef CONFIG_X86_64
R
Roman Zippel 已提交
43
#define mod_64(x, y) ((x) - (y) * div64_u64(x, y))
S
Sheng Yang 已提交
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
#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 已提交
68 69
	res.l.high = div64_u64(rh, c);
	res.l.low = div64_u64(((mod_64(rh, c) << 32) + (rl & 0xffffffff)), c);
S
Sheng Yang 已提交
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 98
	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;
}

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

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

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

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

115 116 117 118 119 120 121 122 123
	/*
	 * 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.
	 */
124
	remaining = hrtimer_get_remaining(&ps->pit_timer.timer);
125 126
	elapsed = ps->pit_timer.period - ktime_to_ns(remaining);
	elapsed = mod_64(elapsed, ps->pit_timer.period);
127 128 129 130 131 132 133 134 135 136 137 138 139

	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 已提交
140 141 142 143 144 145 146 147 148
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));

149
	t = kpit_elapsed(kvm, c, channel);
S
Sheng Yang 已提交
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 178
	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));

179
	t = kpit_elapsed(kvm, c, channel);
S
Sheng Yang 已提交
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 234
	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;
	}
}

235
static void kvm_pit_ack_irq(struct kvm_irq_ack_notifier *kian)
236 237 238
{
	struct kvm_kpit_state *ps = container_of(kian, struct kvm_kpit_state,
						 irq_ack_notifier);
239 240 241 242 243 244 245 246
	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
		 */
247
		atomic_inc(&ps->pit_timer.pending);
248 249 250 251 252
	else if (value > 0)
		/* in this case, we had multiple outstanding pit interrupts
		 * that we needed to inject.  Reinject
		 */
		queue_work(ps->pit->wq, &ps->pit->expired);
253
	ps->irq_ack = 1;
254
	spin_unlock(&ps->inject_lock);
255 256
}

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

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

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

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

276 277 278 279 280 281 282
static bool kpit_is_periodic(struct kvm_timer *ktimer)
{
	struct kvm_kpit_state *ps = container_of(ktimer, struct kvm_kpit_state,
						 pit_timer);
	return ps->is_periodic;
}

283
static struct kvm_timer_ops kpit_ops = {
284 285 286
	.is_periodic = kpit_is_periodic,
};

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 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
static void pit_do_work(struct work_struct *work)
{
	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)
{
	struct kvm_timer *ktimer = container_of(data, struct kvm_timer, timer);
	struct kvm_pit *pt = ktimer->kvm->arch.vpit;

	if (ktimer->reinject || !atomic_read(&ktimer->pending)) {
		atomic_inc(&ktimer->pending);
		queue_work(pt->wq, &pt->expired);
	}

	if (ktimer->t_ops->is_periodic(ktimer)) {
		hrtimer_add_expires_ns(&ktimer->timer, ktimer->period);
		return HRTIMER_RESTART;
	} else
		return HRTIMER_NORESTART;
}

341
static void create_pit_timer(struct kvm_kpit_state *ps, u32 val, int is_period)
S
Sheng Yang 已提交
342
{
343
	struct kvm_timer *pt = &ps->pit_timer;
S
Sheng Yang 已提交
344 345 346 347
	s64 interval;

	interval = muldiv64(val, NSEC_PER_SEC, KVM_PIT_FREQ);

J
Joe Perches 已提交
348
	pr_debug("create pit timer, interval is %llu nsec\n", interval);
S
Sheng Yang 已提交
349 350 351

	/* TODO The new value only affected after the retriggered */
	hrtimer_cancel(&pt->timer);
352
	cancel_work_sync(&ps->pit->expired);
353
	pt->period = interval;
354 355
	ps->is_periodic = is_period;

356
	pt->timer.function = pit_timer_fn;
357 358 359
	pt->t_ops = &kpit_ops;
	pt->kvm = ps->pit->kvm;

S
Sheng Yang 已提交
360
	atomic_set(&pt->pending, 0);
361
	ps->irq_ack = 1;
S
Sheng Yang 已提交
362 363 364 365 366 367 368 369 370 371 372

	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 已提交
373
	pr_debug("load_count val is %d, channel is %d\n", val, channel);
S
Sheng Yang 已提交
374 375

	/*
376 377
	 * The largest possible initial count is 0; this is equivalent
	 * to 216 for binary counting and 104 for BCD counting.
S
Sheng Yang 已提交
378 379 380 381 382 383
	 */
	if (val == 0)
		val = 0x10000;

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

384 385
	if (channel != 0) {
		ps->channels[channel].count_load_time = ktime_get();
S
Sheng Yang 已提交
386
		return;
387
	}
S
Sheng Yang 已提交
388 389 390 391

	/* Two types of timer
	 * mode 1 is one shot, mode 2 is period, otherwise del timer */
	switch (ps->channels[0].mode) {
392
	case 0:
S
Sheng Yang 已提交
393
	case 1:
M
Marcelo Tosatti 已提交
394 395
        /* FIXME: enhance mode 4 precision */
	case 4:
B
Beth Kon 已提交
396 397 398
		if (!(ps->flags & KVM_PIT_FLAGS_HPET_LEGACY)) {
			create_pit_timer(ps, val, 0);
		}
S
Sheng Yang 已提交
399 400
		break;
	case 2:
A
Aurelien Jarno 已提交
401
	case 3:
B
Beth Kon 已提交
402 403 404
		if (!(ps->flags & KVM_PIT_FLAGS_HPET_LEGACY)){
			create_pit_timer(ps, val, 1);
		}
S
Sheng Yang 已提交
405 406
		break;
	default:
407
		destroy_pit_timer(kvm->arch.vpit);
S
Sheng Yang 已提交
408 409 410
	}
}

B
Beth Kon 已提交
411
void kvm_pit_load_count(struct kvm *kvm, int channel, u32 val, int hpet_legacy_start)
412
{
B
Beth Kon 已提交
413 414 415 416 417 418 419 420 421 422
	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);
	}
423 424
}

G
Gregory Haskins 已提交
425 426 427 428 429 430 431 432 433 434
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);
}

435 436 437 438 439 440 441 442
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 已提交
443
{
G
Gregory Haskins 已提交
444
	struct kvm_pit *pit = dev_to_pit(this);
S
Sheng Yang 已提交
445 446 447 448 449
	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;
450 451
	if (!pit_in_range(addr))
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
452 453 454 455 456 457 458

	val  &= 0xff;
	addr &= KVM_PIT_CHANNEL_MASK;

	mutex_lock(&pit_state->lock);

	if (val != 0)
J
Joe Perches 已提交
459 460
		pr_debug("write addr is 0x%x, len is %d, val is 0x%x\n",
			 (unsigned int)addr, len, val);
S
Sheng Yang 已提交
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 501 502 503 504 505 506 507 508 509 510 511 512 513

	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);
514
	return 0;
S
Sheng Yang 已提交
515 516
}

517 518
static int pit_ioport_read(struct kvm_io_device *this,
			   gpa_t addr, int len, void *data)
S
Sheng Yang 已提交
519
{
G
Gregory Haskins 已提交
520
	struct kvm_pit *pit = dev_to_pit(this);
S
Sheng Yang 已提交
521 522 523 524
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	int ret, count;
	struct kvm_kpit_channel_state *s;
525 526
	if (!pit_in_range(addr))
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
527 528

	addr &= KVM_PIT_CHANNEL_MASK;
529 530 531
	if (addr == 3)
		return 0;

S
Sheng Yang 已提交
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 571 572 573 574 575 576 577 578 579 580 581 582 583
	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);
584
	return 0;
S
Sheng Yang 已提交
585 586
}

587 588
static int speaker_ioport_write(struct kvm_io_device *this,
				gpa_t addr, int len, const void *data)
S
Sheng Yang 已提交
589
{
G
Gregory Haskins 已提交
590
	struct kvm_pit *pit = speaker_to_pit(this);
S
Sheng Yang 已提交
591 592 593
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	u32 val = *(u32 *) data;
594 595
	if (addr != KVM_SPEAKER_BASE_ADDRESS)
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
596 597 598 599 600

	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);
601
	return 0;
S
Sheng Yang 已提交
602 603
}

604 605
static int speaker_ioport_read(struct kvm_io_device *this,
			       gpa_t addr, int len, void *data)
S
Sheng Yang 已提交
606
{
G
Gregory Haskins 已提交
607
	struct kvm_pit *pit = speaker_to_pit(this);
S
Sheng Yang 已提交
608 609 610 611
	struct kvm_kpit_state *pit_state = &pit->pit_state;
	struct kvm *kvm = pit->kvm;
	unsigned int refresh_clock;
	int ret;
612 613
	if (addr != KVM_SPEAKER_BASE_ADDRESS)
		return -EOPNOTSUPP;
S
Sheng Yang 已提交
614 615 616 617 618 619 620 621 622 623 624

	/* 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);
625
	return 0;
S
Sheng Yang 已提交
626 627
}

628
void kvm_pit_reset(struct kvm_pit *pit)
S
Sheng Yang 已提交
629 630
{
	int i;
631 632 633
	struct kvm_kpit_channel_state *c;

	mutex_lock(&pit->pit_state.lock);
B
Beth Kon 已提交
634
	pit->pit_state.flags = 0;
635 636 637 638 639 640 641 642 643
	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);
644
	pit->pit_state.irq_ack = 1;
645 646
}

647 648 649 650 651 652 653 654 655 656
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 已提交
657 658 659 660 661 662 663 664 665 666
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,
};

667
/* Caller must hold slots_lock */
668
struct kvm_pit *kvm_create_pit(struct kvm *kvm, u32 flags)
669
{
S
Sheng Yang 已提交
670 671
	struct kvm_pit *pit;
	struct kvm_kpit_state *pit_state;
672
	int ret;
S
Sheng Yang 已提交
673 674 675 676 677

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

678
	pit->irq_source_id = kvm_request_irq_source_id(kvm);
679 680
	if (pit->irq_source_id < 0) {
		kfree(pit);
681
		return NULL;
682
	}
683

S
Sheng Yang 已提交
684 685
	mutex_init(&pit->pit_state.lock);
	mutex_lock(&pit->pit_state.lock);
686 687 688 689
	spin_lock_init(&pit->pit_state.inject_lock);

	pit->wq = create_singlethread_workqueue("kvm-pit-wq");
	if (!pit->wq) {
690
		mutex_unlock(&pit->pit_state.lock);
691
		kvm_free_irq_source_id(kvm, pit->irq_source_id);
692 693 694 695
		kfree(pit);
		return NULL;
	}
	INIT_WORK(&pit->expired, pit_do_work);
S
Sheng Yang 已提交
696 697 698 699 700 701 702 703

	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);
704 705 706
	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);
707
	pit_state->pit_timer.reinject = true;
S
Sheng Yang 已提交
708 709
	mutex_unlock(&pit->pit_state.lock);

710
	kvm_pit_reset(pit);
S
Sheng Yang 已提交
711

712 713 714
	pit->mask_notifier.func = pit_mask_notifer;
	kvm_register_irq_mask_notifier(kvm, 0, &pit->mask_notifier);

715
	kvm_iodevice_init(&pit->dev, &pit_dev_ops);
M
Marcelo Tosatti 已提交
716
	ret = kvm_io_bus_register_dev(kvm, KVM_PIO_BUS, &pit->dev);
717 718
	if (ret < 0)
		goto fail;
719 720 721

	if (flags & KVM_PIT_SPEAKER_DUMMY) {
		kvm_iodevice_init(&pit->speaker_dev, &speaker_dev_ops);
M
Marcelo Tosatti 已提交
722
		ret = kvm_io_bus_register_dev(kvm, KVM_PIO_BUS,
723 724 725
						&pit->speaker_dev);
		if (ret < 0)
			goto fail_unregister;
726 727
	}

S
Sheng Yang 已提交
728
	return pit;
729 730

fail_unregister:
M
Marcelo Tosatti 已提交
731
	kvm_io_bus_unregister_dev(kvm, KVM_PIO_BUS, &pit->dev);
732 733

fail:
734 735 736
	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);
737
	destroy_workqueue(pit->wq);
738 739
	kfree(pit);
	return NULL;
S
Sheng Yang 已提交
740 741 742 743 744 745 746
}

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

	if (kvm->arch.vpit) {
747 748 749
		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);
750 751
		kvm_unregister_irq_mask_notifier(kvm, 0,
					       &kvm->arch.vpit->mask_notifier);
752 753
		kvm_unregister_irq_ack_notifier(kvm,
				&kvm->arch.vpit->pit_state.irq_ack_notifier);
S
Sheng Yang 已提交
754 755 756
		mutex_lock(&kvm->arch.vpit->pit_state.lock);
		timer = &kvm->arch.vpit->pit_state.pit_timer.timer;
		hrtimer_cancel(timer);
757
		cancel_work_sync(&kvm->arch.vpit->expired);
758
		kvm_free_irq_source_id(kvm, kvm->arch.vpit->irq_source_id);
S
Sheng Yang 已提交
759
		mutex_unlock(&kvm->arch.vpit->pit_state.lock);
760
		destroy_workqueue(kvm->arch.vpit->wq);
S
Sheng Yang 已提交
761 762 763
		kfree(kvm->arch.vpit);
	}
}